Upload 5 files

app.py

@@ -1,13 +1,12 @@
-
 import gradio as gr
 from gtts import gTTS
 import tempfile
 import json
 import datetime
-import random
 import os
+import base64
 
-# Supported languages and codes
+# Language codes for voice support
 lang_codes = {
     "english": "en", "hindi": "hi", "marathi": "mr", "bengali": "bn",
     "tamil": "ta", "telugu": "te", "malayalam": "ml", "spanish": "es",
@@ -16,48 +15,35 @@ lang_codes = {
 
 user_info = {"name": "", "age": "", "language": "english", "mood": ""}
 chat_history = []
-journal_entries = []
 
 def welcome_screen():
-    return "🧠 Welcome to StrongMind Therapist!
-Your safe space to talk, heal, and grow."
+    return "🧠 Welcome to StrongMind Therapist!\nYour safe space to talk, heal, and grow."
 
 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."
-    return greeting
+    return f"Welcome {name}! Let's begin your therapy session in {language}. You said you're feeling {mood}. I'm here to listen."
 
 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. 🌈"
+        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 💚"
+        )
+    elif 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. 🧡"
+    elif 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 ideas?"
+    elif "joke" in user_input or "make me happy" in user_input:
+        return "😄 Why don't scientists trust atoms? Because they make up everything!"
+    else:
+        return "Thanks for sharing. I'm here with you. You can keep talking — I’m listening."
 
 def get_therapist_reply(audio, text_input):
     try:
@@ -93,40 +79,34 @@ def get_therapist_reply(audio, text_input):
     except Exception as e:
         return f"⚠️ Error: {str(e)}", None
 
-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)
+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])
+
+with gr.Blocks(css="body {background-color: #ffe6f0;}") as app:
+    with open("calm.mp3", "rb") as f:
+        audio_bytes = f.read()
+        encoded = base64.b64encode(audio_bytes).decode()
+        gr.HTML(f"""
+        <audio autoplay loop hidden>
+            <source src="data:audio/mp3;base64,{encoded}" type="audio/mp3">
+        </audio>
+        """)
+
+    gr.Markdown("## 🌿 StrongMind Therapist 2.0")
 
     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?")
+        gr.Markdown(welcome_screen())
+        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")
         start.click(set_preferences, inputs=[name, age, lang, mood], outputs=welcome_msg)
 
-    with gr.Tab("💬 Therapist Chat"):
+    with gr.Tab("💬 Chat"):
         audio = gr.Audio(type="filepath", label="🎙️ Speak")
         text_input = gr.Textbox(label="⌨️ Or type here")
         btn = gr.Button("🧘 Get Response")
@@ -134,22 +114,29 @@ with gr.Blocks(theme=gr.themes.Soft(), css="body { background-color: #ffe6f0 }")
         voice_out = gr.Audio(label="🔊 Voice Reply")
         btn.click(get_therapist_reply, inputs=[audio, text_input], outputs=[response, voice_out])
 
-    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("📓 Journal"):
+        journal = gr.Textbox(label="✍️ Write your thoughts", lines=8)
+        save_journal = gr.Button("💾 Save Entry")
+        status = gr.Textbox(label="🗂️ Status")
+        def save_entry(text):
+            with open("journal.txt", "a", encoding="utf-8") as f:
+                f.write(f"{datetime.datetime.now()}\n{text}\n\n")
+            return "Saved to journal.txt ✅"
+        save_journal.click(save_entry, inputs=journal, outputs=status)
+
+    with gr.Tab("📘 Study Tips"):
+        def tips():
+            return (
+                "🧠 Study Tips:\n"
+                "1. Pomodoro method: 25 min study, 5 min break\n"
+                "2. Eliminate distractions (put phone away)\n"
+                "3. Summarize what you learned out loud\n"
+                "4. Stay hydrated and eat light snacks\n"
+                "5. Sleep well!"
+            )
+        gr.Button("📚 Show Tips").click(tips, outputs=gr.Textbox(lines=8, label="Tips"))
 
     with gr.Tab("📜 Chat History"):
-        def show_history():
-            if not chat_history:
-                return "No chats yet."
-            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_btn.click(show_history, outputs=history_output)