Update app.py

app.py

@@ -12,11 +12,12 @@ API_URL = "https://api.groq.com/openai/v1/chat/completions"
 def get_risk_color(risk_level):
     """Return color based on risk level"""
     colors = {
-        "HIGH": "#ff4d4d",  # Red
-        "MEDIUM": "#ffa64d", # Orange
-        "LOW": "#4dff4d",   # Green
+        "HIGH": "red",
+        "MEDIUM": "orange",
+        "LOW": "green",
+        "ERROR": "gray"
     }
-    return colors.get(risk_level.upper(), "#666666")
+    return colors.get(risk_level.upper(), "gray")
 
 def analyze_flood(location, water_level, rainfall, historical_data):
     """Enhanced flood analysis with structured output"""
@@ -61,31 +62,26 @@ def analyze_flood(location, water_level, rainfall, historical_data):
         response.raise_for_status()
         result = json.loads(response.json()["choices"][0]["message"]["content"])
         
-        return format_output(result)
+        # Format results
+        risk_level = result.get("risk_level", "UNKNOWN")
+        color = get_risk_color(risk_level)
+        
+        return (
+            f"<span style='color: {color}; font-weight: bold'>{risk_level}</span>",
+            result.get("summary", "No summary available"),
+            "\n".join([f"• {point}" for point in result.get("detailed_analysis", [])]),
+            "\n".join([f"• {action}" for action in result.get("recommended_actions", [])]),
+            f"{result.get('confidence', 'N/A')}%"
+        )
         
     except Exception as e:
-        return format_output({"error": str(e)})
-
-def format_output(data):
-    """Format the API response for Gradio display"""
-    if "error" in data:
-        return {
-            "Risk Level": "ERROR",
-            "Summary": data["error"],
-            "Details": "Failed to get analysis",
-            "Actions": "Please try again later"
-        }
-    
-    risk_level = data.get("risk_level", "UNKNOWN")
-    color = get_risk_color(risk_level)
-    
-    return {
-        "Risk Level": f"<span style='color: {color}; font-weight: bold'>{risk_level}</span>",
-        "Summary": data.get("summary", "No summary available"),
-        "Detailed Analysis": "\n".join([f"• {point}" for point in data.get("detailed_analysis", [])]),
-        "Recommended Actions": "\n".join([f"• {action}" for action in data.get("recommended_actions", [])]),
-        "Confidence": f"{data.get('confidence', 'N/A')}%"
-    }
+        return (
+            "<span style='color: gray; font-weight: bold'>ERROR</span>",
+            f"Analysis failed: {str(e)}",
+            "Please try again later",
+            "Check your API key and connection",
+            "0%"
+        )
 
 # Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft(), title="Flood Risk Analyzer") as app:
@@ -141,17 +137,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Flood Risk Analyzer") as app:
     </div>
     """)
     
-    # Event Handling
+    # Event Handling - Now returning a tuple instead of dict
     submit_btn.click(
         fn=analyze_flood,
         inputs=[location, water_level, rainfall, historical],
-        outputs={
-            "Risk Level": risk_level,
-            "Summary": summary,
-            "Detailed Analysis": details,
-            "Recommended Actions": actions,
-            "Confidence": confidence
-        }
+        outputs=[risk_level, summary, details, actions, confidence]
     )
 
 app.launch()