Update app.py

app.py

@@ -50,6 +50,12 @@ def detect_anomaly(csv_file):
         return "✅ No significant anomalies detected."
     return "⚠️ Anomalies found:\\n" + anomalies.to_string(index=False)
 
+# module 6: chat assistant
+def chat_assistant(question):
+    prompt = f"Answer this smart city sustainability question:\n\nQ: {question}\nA:"
+    result = llm(prompt, max_new_tokens=200, temperature=0.7)[0]["generated_text"]
+    return result.replace(prompt, "").strip()
+
 
 # Gradio Interface
 app = gr.TabbedInterface(
@@ -58,9 +64,10 @@ app = gr.TabbedInterface(
         gr.Interface(fn=citizen_feedback, inputs=gr.Textbox(lines=2, label="Describe the Issue"), outputs="text", title="📣 Citizen Feedback"),
         gr.Interface(fn=kpi_forecasting, inputs=gr.File(label="Upload CSV with Year and Value columns"), outputs="text", title="📊 KPI Forecasting"),
         gr.Interface(fn=eco_tips, inputs=gr.Textbox(lines=2, label="Keyword (e.g. Plastic, Solar)"), outputs="text", title="🌱 Eco Tips Generator"),
-        gr.Interface(fn=detect_anomaly, inputs=gr.File(label="Upload CSV with 'value' column"), outputs="text", title="🚨 Anomaly Detection")
+        gr.Interface(fn=detect_anomaly, inputs=gr.File(label="Upload CSV with 'value' column"), outputs="text", title="🚨 Anomaly Detection"),
+        gr.Interface(fn=chat_assistant, inputs=gr.Textbox(lines=2, label="Ask your question"), outputs="text", title="💬 Smart City Chat")
     ],
-    tab_names=["Summarize", "Feedback", "Forecast", "Eco Tips", "Anomalies"],
+    tab_names=["Summarize", "Feedback", "Forecast", "Eco Tips", "Anomalies", "Chat"],
     title="🌆 Sustainable Smart City Assistant"
 )