Update app.py

app.py

@@ -1,22 +1,25 @@
 import gradio as gr
-import json
+import pandas as pd
 import os
 
-# Load fake survey dataset safely
-survey_data = []
+# Load survey dataset from CSV
+survey_data = None
 try:
-    if os.path.exists("survey.json"):
-        with open("survey.json", "r") as f:
-            survey_data = json.load(f)
+    if os.path.exists("survey.csv"):
+        survey_data = pd.read_csv("survey.csv")
     else:
-        survey_data = [
-            {"employee": "Aiman", "mood": "Positive", "recommendation": "Keep up the good work!"},
-            {"employee": "Ali", "mood": "Negative", "recommendation": "Take a short break to reduce stress."},
-            {"employee": "Sara", "mood": "Neutral", "recommendation": "Encourage more collaboration."}
-        ]
+        # fallback fake dataset
+        survey_data = pd.DataFrame({
+            "employee": ["Aiman", "Ali", "Sara"],
+            "mood": ["Positive", "Negative", "Neutral"],
+            "recommendation": [
+                "Keep up the good work!",
+                "Take a short break to reduce stress.",
+                "Encourage more collaboration."
+            ]
+        })
 except Exception as e:
-    survey_data = []
-    print("Error loading survey.json:", str(e))
+    print("Error loading CSV:", str(e))
 
 
 def chatbot(query):
@@ -26,16 +29,16 @@ def chatbot(query):
 
         query_lower = query.lower()
 
-        # Check each employee in dataset
-        for entry in survey_data:
-            if entry["employee"].lower() in query_lower:
-                return f"✅ {entry['employee']} is feeling **{entry['mood']}**.\n💡 Recommendation: {entry['recommendation']}"
+        # Match employee name
+        for i, row in survey_data.iterrows():
+            if row["employee"].lower() in query_lower:
+                return f"✅ {row['employee']} is feeling **{row['mood']}**.\n💡 Recommendation: {row['recommendation']}"
 
-        # If employee not found
+        # No match found
         return "ℹ️ I don’t have data for that person. Try asking about Aiman, Ali, or Sara."
-
+    
     except Exception as e:
-        return f"🚨 Internal Error: {str(e)}"
+        return f"🚨 Error: {str(e)}"
 
 
 # Gradio Chat Interface