Update app.py

app.py

@@ -10,7 +10,7 @@ import gradio as gr
 import speech_recognition as sr
 import json
 
-# --- Dummy vocab and label encoder ---
+# Dummy vocab and label encoder
 vocab = {'<PAD>': 0, '<UNK>': 1, 'i': 2, 'am': 3, 'feeling': 4, 'sad': 5, 'happy': 6,
          'angry': 7, 'love': 8, 'stressed': 9, 'anxious': 10}
 MAX_LEN = 16
@@ -42,50 +42,36 @@ def preprocess_input(text):
     padded = encoded[:MAX_LEN] + [vocab['<PAD>']] * max(0, MAX_LEN - len(encoded))
     return torch.tensor([padded], dtype=torch.long)
 
-# --- Load solutions CSV from Google Drive ---
-file_id = "1yVJh_NVL4Y4YqEXGym47UCK5ZNZgVZYv"  # Replace with your file ID
+# Load solutions CSV from Google Drive
+file_id = "1yVJh_NVL4Y4YqEXGym47UCK5ZNZgVZYv"  # replace with your CSV file ID
 url = f"https://drive.google.com/uc?export=download&id={file_id}"
 response = requests.get(url)
 csv_text = response.text
-
 if csv_text.strip().startswith('<'):
-    raise Exception("ERROR: Google Drive link is not returning CSV! Check your sharing settings.")
-
+    raise Exception("ERROR: Google Drive link is not returning CSV! Check sharing settings.")
 solutions_df = pd.read_csv(StringIO(csv_text), header=0, on_bad_lines='skip')
 used_solutions = {emotion: set() for emotion in solutions_df['emotion'].unique()}
 
-negative_words = [
-    "not", "bad", "sad", "anxious", "anxiety", "depressed", "upset", "shit", "stress",
-    "worried", "unwell", "struggling", "low", "down", "terrible", "awful",
-    "nervous", "panic", "afraid", "scared", "tense", "overwhelmed", "fear", "uneasy"
-]
+negative_words = ["not", "bad", "sad", "anxious", "anxiety", "depressed", "upset", "shit", "stress",
+                 "worried", "unwell", "struggling", "low", "down", "terrible", "awful",
+                 "nervous", "panic", "afraid", "scared", "tense", "overwhelmed", "fear", "uneasy"]
 
 responses = {
-    "sadness": [
-        "It’s okay to feel down sometimes. I’m here to support you.",
-        "I'm really sorry you're going through this. Want to talk more about it?",
-        "You're not alone — I’m here for you."
-    ],
-    "anger": [
-        "That must have been frustrating. Want to vent about it?",
-        "It's okay to feel this way. I'm listening.",
-        "Would it help to talk through it?"
-    ],
-    "love": [
-        "That’s beautiful to hear! What made you feel that way?",
-        "It’s amazing to experience moments like that.",
-        "Sounds like something truly meaningful."
-    ],
-    "happiness": [
-        "That's awesome! What’s bringing you joy today?",
-        "I love hearing good news. 😊",
-        "Yay! Want to share more about it?"
-    ],
-    "neutral": [
-        "Got it. I’m here if you want to dive deeper.",
-        "Thanks for sharing that. Tell me more if you’d like.",
-        "I’m listening. How else can I support you?"
-    ]
+    "sadness": ["It’s okay to feel down sometimes. I’m here to support you.",
+                "I'm really sorry you're going through this. Want to talk more about it?",
+                "You're not alone — I’m here for you."],
+    "anger": ["That must have been frustrating. Want to vent about it?",
+              "It's okay to feel this way. I'm listening.",
+              "Would it help to talk through it?"],
+    "love": ["That’s beautiful to hear! What made you feel that way?",
+             "It’s amazing to experience moments like that.",
+             "Sounds like something truly meaningful."],
+    "happiness": ["That's awesome! What’s bringing you joy today?",
+                  "I love hearing good news. 😊",
+                  "Yay! Want to share more about it?"],
+    "neutral": ["Got it. I’m here if you want to dive deeper.",
+                "Thanks for sharing that. Tell me more if you’d like.",
+                "I’m listening. How else can I support you?"]
 }
 
 def get_unique_solution(emotion):
@@ -125,8 +111,7 @@ def audio_to_text(audio_file):
     with sr.AudioFile(audio_file) as source:
         audio = recog.record(source)
     try:
-        text = recog.recognize_google(audio)
-        return text
+        return recog.recognize_google(audio)
     except Exception:
         return ""
 
@@ -134,7 +119,6 @@ GLOBAL_CONVO_HISTORY = []
 USER_FEEDBACK_STATE = {}
 
 def emoti_chat(audio, text, history_json=""):
-    # Get user input from voice or text
     if text and text.strip():
         user_input = text
     elif audio is not None:
@@ -147,11 +131,9 @@ def emoti_chat(audio, text, history_json=""):
 
     user_input = correct_spelling(user_input)
 
-    # Exit phrases
     if user_input.lower().strip() in ["exit", "quit", "goodbye", "bye", "close"]:
         return "Take care! I’m here whenever you want to talk. 👋", json.dumps(GLOBAL_CONVO_HISTORY[-5:], indent=2), gr.update(visible=False)
 
-    # Feedback handling
     user_id = "default_user"
     state = USER_FEEDBACK_STATE.get(user_id, {"emotion": None, "pending": False})
 
@@ -167,7 +149,6 @@ def emoti_chat(audio, text, history_json=""):
             USER_FEEDBACK_STATE[user_id] = {"emotion": None, "pending": False}
             return "How can I help you further?", json.dumps(GLOBAL_CONVO_HISTORY[-5:], indent=2), ""
 
-    # Normal user message
     pred_emotion = get_emotion(user_input)
     support = random.choice(responses.get(pred_emotion, responses["neutral"]))
     try:
@@ -187,21 +168,19 @@ def emoti_chat(audio, text, history_json=""):
     USER_FEEDBACK_STATE[user_id] = {"emotion": pred_emotion, "pending": True}
     return reply, json.dumps(GLOBAL_CONVO_HISTORY[-5:], indent=2), ""
 
-import gradio as gr
-
 iface = gr.Interface(
     fn=emoti_chat,
     inputs=[
         gr.Audio(type="filepath", label="🎤 Speak your message"),
         gr.Textbox(lines=2, placeholder="Or type your message here...", label="💬 Type message"),
-        gr.Textbox(lines=1, value="", visible=False)  # Hidden, conversation history
+        gr.Textbox(lines=1, value="", visible=False)
     ],
     outputs=[
         gr.Textbox(label="EmotiBot Reply"),
         gr.Textbox(label="Conversation History (JSON)", visible=False)
     ],
     title="EmotiBot Connect",
-    description="Talk to EmotiBot using your voice or by typing. Detects your emotion, gives dynamic suggestions, and keeps conversation history!"
+    description="Talk to EmotiBot using your voice or by typing. Detects your emotion, gives suggestions, and keeps history!"
 )
 
 iface.launch()