Upload 4 files

app.py

@@ -1,29 +1,37 @@
-
 import gradio as gr
 from gtts import gTTS
 import tempfile
 import json
 import datetime
 
+# 🌐 Language support dictionary
 lang_codes = {
     "english": "en", "hindi": "hi", "marathi": "mr", "bengali": "bn",
     "tamil": "ta", "telugu": "te", "malayalam": "ml", "spanish": "es",
     "french": "fr", "german": "de"
 }
 
+# 🧠 Store user info and chat history
 user_info = {"name": "", "age": "", "language": "english", "mood": ""}
 chat_history = []
 
+# 🌸 Welcome screen message
 def welcome_screen():
-    return "🧠 Welcome to StrongMind Therapist!\nYour safe space to talk, heal, and grow."
+    return "🧠 Welcome to StrongMind Therapist!\nYour peaceful space to talk, heal, and grow 💖"
+
+# ✅ Set user preferences
 def set_preferences(name, age, language, mood):
     user_info["name"] = name
     user_info["age"] = age
     user_info["language"] = language.lower()
     user_info["mood"] = mood
-    greeting = f"Welcome {name}! Let's begin your therapy session in {language}. You said you're feeling {mood}. I'm here to listen."
+    greeting = (
+        f"Welcome {name}! Let's begin your therapy session in {language.title()}. "
+        f"You said you're feeling {mood}. I’m here to support you 🌱"
+    )
     return greeting
 
+# 💬 Therapist reply generator
 def get_therapist_reply(audio, text_input):
     try:
         if audio:
@@ -34,10 +42,11 @@ def get_therapist_reply(audio, text_input):
         elif text_input:
             user_text = text_input
         else:
-            return "Please say or type something to begin.", None
+            return "Please speak or type something to begin.", None
 
         reply = generate_reply(user_text)
 
+        # 💾 Save chat
         chat_entry = {
             "user": user_text,
             "reply": reply,
@@ -49,6 +58,7 @@ def get_therapist_reply(audio, text_input):
         with open("chat_history.json", "w") as f:
             json.dump(chat_history, f, indent=4)
 
+        # 🎧 Convert reply to voice
         lang_code = lang_codes.get(user_info["language"], "en")
         tts = gTTS(text=reply, lang=lang_code)
         audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
@@ -58,58 +68,69 @@ def get_therapist_reply(audio, text_input):
     except Exception as e:
         return f"⚠️ Error: {str(e)}", None
 
+# 🤖 Smart response generator
 def generate_reply(user_input):
     user_input = user_input.lower()
+
     if any(word in user_input for word in ["suicide", "kill", "die", "harm", "cut"]):
         return (
-            "💔 I believe you're very hurt, but I can't provide help in this situation.
-"
-            "Please contact a professional therapist:
-"
-            "📞 Dr. Meera Kapoor – +91 9876543210
-"
-            "📞 Dr. Arjun Das – +91 9123456780
-You are not alone 💚"
+            "💔 I believe you're very hurt, but I can't provide help in this situation.\n"
+            "Please contact a professional therapist:\n"
+            "📞 Dr. Meera Kapoor – +91 9876543210\n"
+            "📞 Dr. Arjun Das – +91 9123456780\nYou are not alone 💚"
         )
+
     if any(word in user_input for word in ["sad", "anxiety", "angry", "unwell", "hate", "stress"]):
-        return "Hey, no need to worry. I'm here for you. Just tell me more — we’ll figure it out together. 🧡"
+        return "🌈 I hear you. Let's take a deep breath together. You’re not alone — tell me more, I'm here for you."
+
     if any(word in user_input for word in ["what should i do", "advice", "guide", "suggest"]):
         return (
-            "🧠 When you're confused, take 3 deep breaths and write down small steps you can take. "
-            "Would you like help listing some ideas?"
+            "🧭 When you're unsure, take a moment to pause and reflect. Try breaking your problem into small steps. "
+            "Would you like help listing ideas?"
         )
-    if "joke" in user_input or "make me happy" in user_input:
-        return "😄 Why don't scientists trust atoms? Because they make up everything!"
-    return "Thank you for sharing. I'm here with you. You can keep talking — I’m listening."
 
-with gr.Blocks(css="body { background-color: #ffe6f0; }") as app:
-    gr.Markdown("## 🌸 StrongMind Therapist with Nature Sounds")
-    gr.Audio(label="🎵 Calm Nature Music", value="calm_nature.mp3", interactive=False)
+    if any(word in user_input for word in ["joke", "make me happy", "funny"]):
+        return "😄 Why don’t eggs tell each other secrets? Because they might crack up!"
+
+    return "🪷 Thank you for sharing. You are brave. Keep going, I'm right here listening."
 
+# 🎨 Build the Gradio interface
+with gr.Blocks(theme=gr.themes.Base(primary_hue="pink", secondary_hue="pink")) as app:
+    gr.Markdown("## 🧠 StrongMind Therapist")
+    gr.Markdown(welcome_screen())
+
+    # 🎚️ Preference tab
     with gr.Tab("🛠️ Start"):
-        name = gr.Textbox(label="Your Name")
-        age = gr.Textbox(label="Your Age")
-        lang = gr.Dropdown(choices=list(lang_codes.keys()), label="Preferred Language")
-        mood = gr.Textbox(label="How are you feeling today?")
+        name = gr.Textbox(label="👤 Your Name")
+        age = gr.Textbox(label="🎂 Your Age")
+        lang = gr.Dropdown(choices=list(lang_codes.keys()), label="🌐 Preferred Language")
+        mood = gr.Textbox(label="🧘 How are you feeling today?")
         start = gr.Button("✅ Start Session")
-        welcome_msg = gr.Textbox(label="System Message")
+        welcome_msg = gr.Textbox(label="📝 System Message")
         start.click(set_preferences, inputs=[name, age, lang, mood], outputs=welcome_msg)
 
+    # 🎙️ Chat tab
     with gr.Tab("💬 Chat"):
         audio = gr.Audio(type="filepath", label="🎙️ Speak")
-        text_input = gr.Textbox(label="⌨️ Or type here")
+        text_input = gr.Textbox(label="⌨️ Or type your feelings")
         btn = gr.Button("🧘 Get Response")
         response = gr.Textbox(label="💬 Therapist's Reply")
         voice_out = gr.Audio(label="🔊 Voice Reply")
         btn.click(get_therapist_reply, inputs=[audio, text_input], outputs=[response, voice_out])
 
+    # 🗂 Chat history
     with gr.Tab("📜 Chat History"):
         def show_history():
             if not chat_history:
                 return "No chats yet."
-            return "\n".join([f"🕒 {c['time']}\nYou: {c['user']}\nBot: {c['reply']}\n" for c in chat_history])
+            return "\n".join([
+                f"🕒 {c['time']}\nYou: {c['user']}\nBot: {c['reply']}\n" for c in chat_history
+            ])
         history_btn = gr.Button("📂 Load Past Chats")
-        history_output = gr.Textbox(label="🗂️ Your Past Conversations", lines=20)
+        history_output = gr.Textbox(label="📄 Your Past Conversations", lines=20)
         history_btn.click(show_history, outputs=history_output)
 
+    # 🌿 Background music (optional hidden audio player)
+    gr.Audio(value="calm_nature.mp3", autoplay=True, visible=False)
+
 app.launch()