Update app.py

app.py

@@ -66,31 +66,44 @@ def detect_depression_risk(text: str) -> dict:
     }
 
 # --- 4. AGENT REASONING (SambaNova + Nebius Kimi) ---
-def agent_reasoning(text, risk_data, provider="SambaNova"):
+# --- 4. AGENT REASONING (SambaNova + Nebius Kimi) ---
+def agent_reasoning(text, risk_data, provider="SambaNova (Llama 3.3)"):
     """
     Uses Sponsor APIs to generate the analysis report.
     """
     client = None
     model_id = None
     
-    # SYSTEM PROMPT
+    # IMPROVED SYSTEM PROMPT
     system_prompt = f"""
-    You are 'Dr. Longformer', an empathetic Clinical AI Research Agent.
-    
-    CLINICAL DATA:
-    - Risk Level: {risk_data['risk_level']} ({risk_data['probability']:.1%})
-    - Biomarker: {risk_data['biomarker']}
-    
-    USER TEXT SNIPPET: "{text[:800]}..."
-    
-    INSTRUCTIONS:
-    1. Acknowledge the user's situation based on the text.
-    2. Explain the risk level using thesis terms: 'Nocturnal Posting' (High), 'Supportive Responder' (Moderate), or 'Healthy External Focus' (Low).
-    3. Be compassionate. Max 100 words.
+    You are 'MindSight', a Clinical Research Agent specialized in linguistic biomarker analysis.
+    
+    CONTEXTUAL DATA:
+    - Detected Risk: {risk_data['risk_level']} ({risk_data['probability']:.1%})
+    - Key Biomarker: {risk_data['biomarker']}
+    - Dataset Context: Based on eRisk longitudinal analysis (2017-2022).
+    
+    USER TEXT SNIPPET: "{text[:1000]}..."
+    
+    YOUR MISSION:
+    Synthesize the clinical data with the user's personal narrative to provide a supportive, research-backed insight.
+    
+    GUIDELINES:
+    1. **Validate:** Start by acknowledging the specific struggles or sentiments expressed in the text (e.g., "I hear your exhaustion...").
+    2. **Connect:** Explain *why* the risk level was triggered using the Biomarker. 
+       - If High Risk: Link sleep/energy complaints to the "Nocturnal Posting" pattern (activity spikes 00:00-05:00 UTC).
+       - If Moderate Risk: Link their focus on others to the "Supportive Responder" effect found in online echo chambers.
+       - If Low Risk: Affirm their "Healthy External Focus" (hobbies, work, events).
+    3. **Bridge:** Offer a gentle, non-medical bridge to the resources below.
+    
+    CONSTRAINTS:
+    - Be warm but objective.
+    - NO medical diagnosis (use "suggests alignment with...").
+    - Max 80 words.
     """
 
     try:
-        # --- SPONSOR 1: NEBIUS (Kimi K2 Thinking) ---
+        # --- SPONSOR 1: NEBIUS (Kimi K2) ---
         if provider == "Nebius (Kimi K2)":
             api_key = os.getenv("NEBIUS_API_KEY")
             if not api_key: return "⚠️ Nebius API Key missing."
@@ -102,7 +115,7 @@ def agent_reasoning(text, risk_data, provider="SambaNova"):
             model_id = "moonshotai/Kimi-K2-Instruct" 
 
         # --- SPONSOR 2: SAMBANOVA (Llama 3.3) ---
-        elif provider == "SambaNova(Llama 3.3)":
+        else: # Default to SambaNova
             api_key = os.getenv("SAMBANOVA_API_KEY")
             if not api_key: return "⚠️ SambaNova API Key missing."
             
@@ -140,19 +153,18 @@ def full_analysis_pipeline(user_text, location, provider):
     # 3. Agent Reasoning
     explanation = agent_reasoning(user_text, risk_data, provider)
     
-    # 4. Color Logic
-    color = "green" if risk_data['probability'] < 0.4 else "orange" if risk_data['probability'] < 0.6 else "red"
+    # 4. Color Logic (Fixed)
+    color = risk_data['color']
     
     # 5. Build HTML/Markdown Report
-    # FIX: Added 'white-space: pre-wrap' to preserve newlines in display
     return f"""
     <div style="border-left: 6px solid {color}; padding-left: 15px; background-color: {color}10; border-radius: 5px; padding: 20px;">
         <h2 style="color:{color}; margin:0;">{risk_data['risk_level']}</h2>
         <h3 style="margin-top:5px;">Probability: {risk_data['probability']*100:.1f}%</h3>
     </div>
     
-    ### 🧠 Agent Analysis ({provider})
-    <div style="white-space: pre-wrap; font-family: sans-serif;">
+    ### 👁️ MindSight Analysis ({provider})
+    <div style="white-space: pre-wrap; font-family: sans-serif; font-size: 1.1em;">
     {explanation}
     </div>
     
@@ -204,7 +216,7 @@ with gr.Blocks(title="Depression Risk Agent") as demo:
             with gr.Row():
                 # RENAMED to be clearer
                 loc_dropdown = gr.Dropdown(["Global", "US", "Malaysia"], value="Malaysia", label="Crisis Resource Region")
-                provider_dropdown = gr.Dropdown(["SambaNova", "Nebius (Kimi K2)"], value="SambaNova", label="Reasoning Brain")
+                provider_dropdown = gr.Dropdown(["SambaNova (Llama 3.3)", "Nebius (Kimi K2)"], value="SambaNova (Llama 3.3)", label="Reasoning Brain")
             
             submit = gr.Button("🚀 Run Analysis Agent", variant="primary", size="lg")