Update app.py

app.py

@@ -15,39 +15,40 @@ from groq import Groq
 import google.generativeai as genai
 
 # -------------------- ENVIRONMENT VARIABLES --------------------
-# We try to get keys, but we don't crash if they are missing
 HF_API_KEY = os.getenv("HF_API_KEY")
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 SENTINEL_CLIENT_ID = os.getenv("SENTINEL_CLIENT_ID")
 SENTINEL_CLIENT_SECRET = os.getenv("SENTINEL_CLIENT_SECRET")
 
-# -------------------- SENTINEL HUB CONFIG --------------------
+# -------------------- SENTINEL CONFIG --------------------
 config = SHConfig()
 if SENTINEL_CLIENT_ID and SENTINEL_CLIENT_SECRET:
     config.client_id = SENTINEL_CLIENT_ID
     config.client_secret = SENTINEL_CLIENT_SECRET
 
-# -------------------- AI SUMMARY FUNCTIONS --------------------
+# -------------------- AI FUNCTIONS (UPDATED MODELS) --------------------
 
 def gemini_summary(text):
-    """Backup: Google Gemini (High Reliability)"""
+    """Backup: Google Gemini 1.5 Flash"""
     try:
         if not GEMINI_API_KEY: return None, "Missing Key"
         genai.configure(api_key=GEMINI_API_KEY)
-        model = genai.GenerativeModel('gemini-pro')
+        # UPDATED MODEL NAME
+        model = genai.GenerativeModel('gemini-1.5-flash') 
         response = model.generate_content(text)
         return response.text, None
     except Exception as e:
         return None, str(e)
 
 def groq_summary(text):
-    """Primary: Groq (Fastest)"""
+    """Primary: Groq (Llama 3.3)"""
     try:
         if not GROQ_API_KEY: return None, "Missing Key"
         client = Groq(api_key=GROQ_API_KEY)
         completion = client.chat.completions.create(
-            model="mixtral-8x7b-32768",
+            # UPDATED MODEL NAME - The old Mixtral one is retired
+            model="llama-3.3-70b-versatile",
             messages=[{"role": "user", "content": text}]
         )
         return completion.choices[0].message.content, None
@@ -55,18 +56,18 @@ def groq_summary(text):
         return None, str(e)
 
 def hf_summary(text):
-    """Fallback: Hugging Face Free Inference"""
+    """Fallback: Hugging Face (Zephyr)"""
     try:
-        # Switching to a more stable model for free tier
-        url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
+        # UPDATED URL AND MODEL
+        url = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
         headers = {"Authorization": f"Bearer {HF_API_KEY}"}
         payload = {
-            "inputs": f"[INST] {text} [/INST]",
+            "inputs": f"<|system|>You are an environmental scientist.</s><|user|>{text}</s><|assistant|>",
             "parameters": {"max_new_tokens": 500}
         }
         r = requests.post(url, headers=headers, json=payload, timeout=25)
         if r.status_code == 200:
-            return r.json()[0]["generated_text"].replace(payload["inputs"], ""), None
+            return r.json()[0]["generated_text"].split("<|assistant|>")[-1], None
         else:
             return None, f"Status {r.status_code}: {r.text}"
     except Exception as e:
@@ -75,26 +76,25 @@ def hf_summary(text):
 def smart_summary(text):
     errors = []
     
-    # 1. Try Groq
+    # 1. Try Groq (Fastest)
     out, err = groq_summary(text)
     if out: return out
     errors.append(f"Groq Error: {err}")
     
-    # 2. Try Gemini (Google)
+    # 2. Try Gemini (Most Reliable)
     out, err = gemini_summary(text)
     if out: return out
     errors.append(f"Gemini Error: {err}")
     
-    # 3. Try Hugging Face
+    # 3. Try Hugging Face (Backup)
     if HF_API_KEY:
         out, err = hf_summary(text)
         if out: return out
         errors.append(f"HF Error: {err}")
     
-    # If all fail, return the error log so the user sees WHY
     return "⚠ ALL AI MODELS FAILED. DEBUG LOG:\n" + "\n".join(errors)
 
-# -------------------- CALCULATION LOGIC --------------------
+# -------------------- MATH & LOGIC --------------------
 def calculate_wqi(pH, do, nutrients):
     wqi = (7 - abs(7 - pH)) * 0.2 + (do/14) * 0.5 + (10 - nutrients) * 0.3
     wqi_score = max(0, min(100, int(wqi*10)))
@@ -122,7 +122,7 @@ def analyze_satellite_image(img):
     turbidity_score = int(np.mean(img_array)/2.55)
     return turbidity_score
 
-# -------------------- VISUALS & REPORTS --------------------
+# -------------------- VISUALS --------------------
 def create_plots(wqi, hsi, erosion, turbidity):
     fig = go.Figure()
     colors = ['#1E90FF', '#32CD32', '#FF4500', '#FFA500']
@@ -157,7 +157,6 @@ def generate_pdf(wqi, hsi, erosion, turbidity, summary_text):
     pdf.cell(0, 10, "Comprehensive Analysis", ln=True)
     pdf.set_font("Arial", "", 11)
     
-    # Handle Text Encoding (force cleanup of special characters)
     safe_summary = summary_text.encode('latin-1', 'replace').decode('latin-1')
     pdf.multi_cell(0, 6, safe_summary)
     
@@ -180,7 +179,6 @@ def generate_pdf(wqi, hsi, erosion, turbidity, summary_text):
 # -------------------- MAIN PROCESSOR --------------------
 def process_data(flow_rate, water_temp, sediment, construction, pH, do, nutrients, sat_img):
     try:
-        # 1. Calculate Scores
         wqi = calculate_wqi(pH, do, nutrients)
         hsi = calculate_hsi(flow_rate, water_temp, sediment)
         erosion = calculate_erosion(sediment, construction)
@@ -188,7 +186,6 @@ def process_data(flow_rate, water_temp, sediment, construction, pH, do, nutrient
         stability = river_stability(wqi, hsi, erosion)
         potability = potability_status(wqi)
         
-        # 2. Prompt
         prompt = f"""
         Act as an Environmental Scientist. Analyze this river data:
         - Water Quality (WQI): {wqi}/100 ({potability})
@@ -205,7 +202,6 @@ def process_data(flow_rate, water_temp, sediment, construction, pH, do, nutrient
         
         summary = smart_summary(prompt)
         
-        # 3. Generate Outputs
         fig = create_plots(wqi, hsi, erosion, turbidity)
         pdf_bytes = generate_pdf(wqi, hsi, erosion, turbidity, summary)
         
@@ -263,7 +259,6 @@ with gr.Blocks(title="FlumenIntel") as demo:
         )
 
     with gr.Row():
-        # LEFT COLUMN
         with gr.Column(scale=1):
             with gr.Group():
                 gr.Markdown("### 1. Hydrological Data")
@@ -284,7 +279,6 @@ with gr.Blocks(title="FlumenIntel") as demo:
             
             analyze_btn = gr.Button("🚀 Run Analysis", variant="primary", size="lg")
 
-        # RIGHT COLUMN
         with gr.Column(scale=2):
             status_box = gr.Textbox(label="Quick Status", interactive=False)