Upload 4 files

app.py

@@ -1,37 +1,64 @@
+
 import gradio as gr
 from gtts import gTTS
 import tempfile
 import json
 import datetime
+import random
+import os
 
-# 🌐 Language support dictionary
+# Supported languages and codes
 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 = []
+journal_entries = []
 
-# 🌸 Welcome screen message
 def welcome_screen():
-    return "🧠 Welcome to StrongMind Therapist!\nYour peaceful space to talk, heal, and grow 💖"
+    return "🧠 Welcome to StrongMind Therapist!
+Your safe 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.title()}. "
-        f"You said you're feeling {mood}. I’m here to support you 🌱"
-    )
+    greeting = f"Welcome {name}! Let's begin your therapy session in {language}. You said you're feeling {mood}. I'm here to listen."
     return greeting
 
-# 💬 Therapist reply 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.\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"]):
+        responses = [
+            "Hey, it’s okay to feel this way. I’m right here. Let's talk. 🌻",
+            "Don’t worry, we’ll get through this together. Tell me what’s bothering you. 💛",
+            "I'm here for you. You can tell me anything. 💬"
+        ]
+        return random.choice(responses)
+
+    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?")
+
+    if "joke" in user_input or "make me happy" in user_input:
+        return random.choice([
+            "😄 Why don’t scientists trust atoms? Because they make up everything!",
+            "😂 I told my computer I needed a break, and it said: ‘Why? I’m not the one overloaded!’",
+            "😹 Why did the student eat his homework? Because the teacher said it was a piece of cake!"
+        ])
+
+    return "Thanks for sharing. I'm listening, tell me more. 🌈"
+
 def get_therapist_reply(audio, text_input):
     try:
         if audio:
@@ -42,11 +69,10 @@ def get_therapist_reply(audio, text_input):
         elif text_input:
             user_text = text_input
         else:
-            return "Please speak or type something to begin.", None
+            return "Please say or type something to begin.", None
 
         reply = generate_reply(user_text)
 
-        # 💾 Save chat
         chat_entry = {
             "user": user_text,
             "reply": reply,
@@ -58,7 +84,6 @@ 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")
@@ -68,69 +93,65 @@ 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.\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 "🌈 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 unsure, take a moment to pause and reflect. Try breaking your problem into small steps. "
-            "Would you like help listing ideas?"
-        )
-
-    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")
+def save_journal(text):
+    journal_entries.append({
+        "entry": text,
+        "time": datetime.datetime.now().isoformat()
+    })
+    with open("journal.json", "w") as f:
+        json.dump(journal_entries, f, indent=2)
+    return "📝 Journal saved!"
+
+def study_tips():
+    return "\n".join([
+        "📌 Break tasks into small parts",
+        "⏰ Take a 5-min break every 25 mins (Pomodoro)",
+        "🧘 Do breathing before starting work",
+        "📴 Keep phone away while studying",
+        "✅ Reward yourself after completing tasks"
+    ])
+
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Soft(), css="body { background-color: #ffe6f0 }") as app:
+    gr.Markdown("## 🌸 StrongMind Therapist 2.0 Pro")
     gr.Markdown(welcome_screen())
+    gr.Audio(value="calm.mp3", autoplay=True, visible=False)
 
-    # 🎚️ 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"):
+    with gr.Tab("💬 Therapist Chat"):
         audio = gr.Audio(type="filepath", label="🎙️ Speak")
-        text_input = gr.Textbox(label="⌨️ Or type your feelings")
+        text_input = gr.Textbox(label="⌨️ Or type here")
         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("📔 Journaling"):
+        journal_text = gr.Textbox(label="Write your thoughts", lines=5)
+        journal_btn = gr.Button("Save Journal")
+        journal_status = gr.Textbox(label="Status")
+        journal_btn.click(save_journal, inputs=journal_text, outputs=journal_status)
+
+    with gr.Tab("📚 Study Tips"):
+        tip_btn = gr.Button("Get Tips")
+        tip_output = gr.Textbox(label="Focus & Study Tips")
+        tip_btn.click(fn=study_tips, outputs=tip_output)
+
     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\n".join([f"🕒 {c['time']}\nYou: {c['user']}\nBot: {c['reply']}" 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()