Update app.py

app.py

@@ -1,24 +1,23 @@
 import gradio as gr
-import requests
+from huggingface_hub import InferenceClient
 import os
 
-# --- 1. SETUP API CONNECTION ---
-API_KEY = os.getenv("HF_API_KEY") 
-# 🔴 FIX: We use the NEW 'router' URL + Zephyr Model (Reliable)
-API_URL = "https://router.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
-headers = {"Authorization": f"Bearer {API_KEY}"}
+# --- 1. SETUP API CONNECTION (The Official Way) ---
+API_KEY = os.getenv("HF_API_KEY")
+
+# We use the official client. It handles the URLs automatically.
+# We are using "Mistral-7B" because it is the most stable on the free tier.
+client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2", token=API_KEY)
 
 # --- 2. DEFINE THE LOGIC ---
 def analyze_biodiversity(species, habitat, region, activity, noise, flow, pollutants, breeding, disturbance):
+    
     if not API_KEY:
         return "⚠️ Error: API Key not found. Please add HF_API_KEY in Settings > Secrets."
 
-    prompt = f"""
-    <|system|>
-    You are BioVigilus, an expert AI Ecologist. 
-    Analyze the environmental impact of the scenario below.
-    </s>
-    <|user|>
+    # Prompt Engineering
+    prompt = f"""[INST] You are BioVigilus, an expert AI Ecologist.
+    
     SCENARIO:
     - Species: {species}
     - Habitat: {habitat} ({region})
@@ -28,40 +27,31 @@ def analyze_biodiversity(species, habitat, region, activity, noise, flow, pollut
     - Breeding Season: {breeding}
 
     TASK:
-    Generate a professional Impact Assessment Report.
-    Use clear headings and emojis.
-    Include:
+    Generate a structured Environmental Impact Assessment (EIA).
+    Use clear headings, bullet points, and emojis.
+    
+    Sections required:
     1. 🛑 **Risk Level** (Critical/High/Medium/Low)
-    2. 🌡️ **Sensitivity Score** (0-100)
+    2. 🌡️ **Ecological Sensitivity Score** (0-100)
     3. 💀 **Predicted Consequences**
-    4. 🛡️ **Mitigation Measures** (Bullet points)
+    4. 🛡️ **Mitigation Measures** (List 3 specific actions)
     5. ⚖️ **Final Verdict** (Proceed/Halt/Modify)
-    </s>
-    <|assistant|>
+    [/INST]
     """
 
     try:
-        # Lower max_new_tokens for speed
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt, "parameters": {"max_new_tokens": 500}})
-        
-        # --- ERROR HANDLING ---
-        if response.status_code == 503:
-             return "⏳ AI is waking up. Please wait 30 seconds and click 'Analyze' again."
-        
-        if response.status_code != 200:
-            return f"⚠️ API Error {response.status_code}: {response.text}"
-
-        result = response.json()
-        
-        if isinstance(result, list) and "generated_text" in result[0]:
-            # Clean up the output to remove the prompt text
-            raw_text = result[0]["generated_text"]
-            return raw_text.replace(prompt, "").strip()
-        else:
-            return str(result)
+        # The stream=True makes it feel faster, but we will wait for full response
+        response = client.text_generation(
+            prompt, 
+            max_new_tokens=600, 
+            temperature=0.7,
+            details=False
+        )
+        return response
             
     except Exception as e:
-        return f"⚠️ Connection Failed: {str(e)}"
+        # Detailed error message
+        return f"⚠️ Analysis Failed: {str(e)}"
 
 # --- 3. CUSTOM STYLE (Trojan Horse Method) ---
 custom_style = """
@@ -79,13 +69,14 @@ custom_style = """
     }
     button.primary:hover { transform: scale(1.02); }
 
-    .gr-box, .gr-input, input { 
+    .gr-box, .gr-input, input, textarea { 
         background-color: #1e293b !important; 
         border: 1px solid #334155 !important; 
+        color: white !important;
     }
     label span { color: #64748b !important; font-weight: bold; }
     
-    .output-text { font-family: monospace; background-color: #1e293b; padding: 15px; border-radius: 8px; }
+    .output-text { font-family: monospace; background-color: #1e293b; padding: 15px; border-radius: 8px; border: 1px solid #475569; }
 </style>
 """
 
@@ -117,7 +108,7 @@ with gr.Blocks() as app:
     btn = gr.Button("🔍 Analyze Ecological Impact", variant="primary")
     
     gr.Markdown("### 📋 Scientific Report")
-    output = gr.Textbox(label="Assessment Results", lines=12, interactive=False, elem_classes=["output-text"])
+    output = gr.Markdown("Results will appear here...", elem_classes=["output-text"])
 
     gr.Examples(
         examples=[