Update app.py

app.py

@@ -6,7 +6,6 @@ import pandas as pd
 import requests
 from io import StringIO
 import speech_recognition as sr
-import json
 import random
 
 # --- Your Dummy Model and Helpers (same as before) ---
@@ -45,7 +44,21 @@ file_id = "1yVJh_NVL4Y4qEXGym47UCK5ZNZgVZYv"
 url = f"https://drive.google.com/uc?export=download&id={file_id}"
 response = requests.get(url)
 csv_text = response.text
+
+# Read CSV and clean column names
 solutions_df = pd.read_csv(StringIO(csv_text), header=0, on_bad_lines='skip')
+solutions_df.columns = solutions_df.columns.str.strip().str.lower()  # clean columns
+
+# Show debug info in Streamlit UI (remove later if you want)
+st.write("Columns found in CSV:", list(solutions_df.columns))
+st.write("First few rows of CSV:")
+st.write(solutions_df.head())
+
+# Stop if 'emotion' column missing
+if 'emotion' not in solutions_df.columns:
+    st.error("CSV is missing the 'emotion' column. Please check your file or rename the column.")
+    st.stop()
+
 used_solutions = {emotion: set() for emotion in solutions_df['emotion'].unique()}
 
 negative_words = [
@@ -186,8 +199,7 @@ if st.button("Send"):
                 "feedback": ""
             })
 
-# Show conversation history
+# Show conversation history (last 5)
 for chat in st.session_state.history[-5:]:
     st.markdown(f"**You:** {chat['user']}")
     st.markdown(f"**EmotiBot:** {chat['bot']}")
-