Upload bios_controller.py with huggingface_hub

bios_controller.py

@@ -1173,31 +1173,69 @@ class NeonDBWriter:
         # 2. Save to results table
         sql_result = """
             INSERT INTO results (
-                id, diagnosis_id, overall_health_score, health_label,
-                dimension_scores, weaknesses, opportunities, action_items,
-                ai_narrative, model_version, meta_data, created_at, updated_at
+                id, diagnosis_id, user_id,
+                health_score, health_label, health_dimensions,
+                weaknesses, opportunities, action_items,
+                ai_narrative, model_used, generation_time_ms, pipeline,
+                created_at
             ) VALUES (
+                %s, %s, %s,
+                %s, %s, %s,
+                %s, %s, %s,
                 %s, %s, %s, %s,
-                %s, %s, %s, %s,
-                %s, %s, %s, NOW(), NOW()
+                NOW()
             )
         """
-
         params_result = (
-            str(uuid.uuid4()), diagnosis_id, d["healthScore"]["total"], d["healthScore"]["label"],
-            json.dumps(d["healthScore"]["dimensions"]), json.dumps(d["weaknesses"]),
-            json.dumps(d["opportunities"]), json.dumps(d["actionItems"]),
-            self.security.encrypt(d["narrative"]), d["modelUsed"], json.dumps(d["benchmarking"])
+            str(uuid.uuid4()), diagnosis_id, user_id,
+            d["healthScore"]["total"], d["healthScore"]["label"], json.dumps(d["healthScore"]["dimensions"]),
+            json.dumps(d["weaknesses"]), json.dumps(d["opportunities"]), json.dumps(d["actionItems"]),
+            d["narrative"], d["modelUsed"], d["generationTimeMs"], d["pipeline"]
         )
 
-        with psycopg.connect(self.database_url) as conn:
-            with conn.cursor() as cur:
-                cur.execute(sql_diagnosis, params_diagnosis)
-                cur.execute(sql_result, params_result)
-            conn.commit()
+        try:
+            with psycopg.connect(self.database_url, row_factory=psycopg_dict_row) as conn:
+                with conn.cursor() as cur:
+                    cur.execute(sql_diagnosis, params_diagnosis)
+                    cur.execute(sql_result, params_result)
+                    conn.commit()
+            return session_id
+        except Exception as e:
+            log.error("Failed to save diagnosis to Neon: %s", e)
+            return session_id
 
-        log.info(f"✅ Report saved to NeonDB | session_id={session_id} | score={d['health_score']}")
-        return session_id
+    def save_contact_lead(self, lead_data: dict) -> bool:
+        """
+        Saves a sales contact lead to the contact_leads table.
+        """
+        sql = """
+            INSERT INTO contact_leads (
+                first_name, last_name, email, company_name, job_title, 
+                country, project_budget, interests
+            ) VALUES (
+                %s, %s, %s, %s, %s, %s, %s, %s
+            )
+        """
+        params = (
+            lead_data["firstName"],
+            lead_data["lastName"],
+            lead_data["email"],
+            lead_data["companyName"],
+            lead_data["jobTitle"],
+            lead_data["country"],
+            lead_data["projectBudget"],
+            json.dumps(lead_data["interests"])
+        )
+
+        try:
+            with psycopg.connect(self.database_url) as conn:
+                with conn.cursor() as cur:
+                    cur.execute(sql, params)
+                    conn.commit()
+            return True
+        except Exception as e:
+            log.error("Failed to save contact lead to Neon: %s", e)
+            return False
 
     def fetch_report(self, session_id: str) -> Optional[dict]:
         """Fetch a previously saved report by session_id and decrypt sensitive fields."""
@@ -1240,7 +1278,6 @@ class NeonDBWriter:
             results.append(data)
         return results
 
-
 # ═══════════════════════════════════════════════════════════════════════════════
 # BIOS CONTROLLER  — the main entry point
 # ═══════════════════════════════════════════════════════════════════════════════