Upload 3 files

app.py

@@ -2,139 +2,148 @@ import gradio as gr
 from gtts import gTTS
 import tempfile
 import datetime
+import openai
 
+# Replace with your OpenAI GPT-4o key
+openai.api_key = "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"
+
+# User data
 user_info = {"name": "", "age": "", "gender": "", "language": "english"}
 chat_history = []
 journal_entries = []
 
-def save_user_info(name, age, gender, language):
-    user_info.update({"name": name, "age": age, "gender": gender, "language": language})
-    return f"Welcome, {name}!"
-
-def generate_response(input_text):
-    # Simulating GPT-based response (freeform, not word-based)
-    if "?" in input_text:
-        response = "That's a thoughtful question. Let's explore it together."
-    elif len(input_text.split()) <= 3:
-        response = "Could you share a little more?"
-    else:
-        response = "I'm here for you. That sounds like a lot — want to talk more about it?"
-    chat_history.append({"user": input_text, "bot": response})
-    return response
+# Calm tips
+calm_tips = [
+    "Take 3 deep breaths.", "Listen to nature.", "Stretch your body.",
+    "Drink water.", "Think of one happy moment.",
+    "Close your eyes and count to 10.", "Smile, even if just a little.",
+    "Look outside and notice one peaceful thing.", "Take a short walk.",
+    "Say something kind to yourself."
+]
+
+# Study tips
+study_tips = [
+    "Break study time into chunks.", "Use mind maps for better recall.",
+    "Teach someone what you learned.", "Take short breaks.",
+    "Use the Pomodoro technique.", "Keep your study space clean.",
+    "Avoid multitasking.", "Revise regularly.", "Set clear goals.",
+    "Reward yourself after studying."
+]
+
+# Chatbot
+def respond(message):
+    chat_history.append(("You", message))
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",
+            messages=[{"role": "system", "content": "You are a kind AI therapist."}] +
+                     [{"role": "user", "content": m[1]} for m in chat_history if m[0] == "You"],
+            max_tokens=150
+        )
+        reply = response.choices[0].message.content.strip()
+    except Exception as e:
+        reply = "Sorry, there was an error connecting to the AI."
+    chat_history.append(("Therapist", reply))
+    return reply
 
 def text_to_speech(text):
-    tts = gTTS(text)
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
+    tts = gTTS(text=text, lang='en')
+    with tempfile.NamedTemporaryFile(delete=True, suffix=".mp3") as fp:
         tts.save(fp.name)
         return fp.name
 
-def chat_function(user_input):
-    response = generate_response(user_input)
-    audio_path = text_to_speech(response)
-    return response, audio_path
-
-def view_chat_history():
-    return "\n".join([f"You: {c['user']}\nAI: {c['bot']}" for c in chat_history])
+def chat_interface(user_input):
+    reply = respond(user_input)
+    audio_path = text_to_speech(reply)
+    return reply, audio_path
 
+# Journal
 def save_journal(entry):
-    now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
-    journal_entries.append(f"{now} - {entry}")
-    return "Entry saved."
-
-def view_journal():
-    return "\n\n".join(journal_entries)
-
-calm_tips = [
-    "Take a deep breath in... and out.",
-    "Look around you and find something beautiful.",
-    "Gently stretch your arms and shoulders.",
-    "Drink a glass of water slowly.",
-    "Think of one thing you’re grateful for.",
-    "Close your eyes for 10 seconds.",
-    "Smile — even a little one.",
-    "Sit back and relax your neck.",
-    "Imagine a peaceful place.",
-    "You're doing better than you think."
-]
-
-def get_calm_tip(index=[0]):
-    tip = calm_tips[index[0] % len(calm_tips)]
-    index[0] += 1
+    now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    journal_entries.append((now, entry))
+    return f"Saved at {now}"
+
+def show_journal_history():
+    return "\n\n".join([f"{t}\n{e}" for t, e in journal_entries]) or "No entries yet."
+
+# Calm Tips
+calm_index = {"value": 0}
+def next_calm_tip():
+    tip = calm_tips[calm_index["value"] % len(calm_tips)]
+    calm_index["value"] += 1
     return tip
 
-study_tips = [
-    "Use the Pomodoro technique: 25 mins study, 5 mins break.",
-    "Keep a distraction list nearby.",
-    "Teach what you learn.",
-    "Stay hydrated and take eye breaks.",
-    "Review your notes before sleeping.",
-    "Use flashcards for quick revision.",
-    "Take handwritten notes.",
-    "Use colored pens or highlighters.",
-    "Keep your phone on silent while studying.",
-    "Revise daily instead of cramming."
-]
+# Personal Info
+def save_personal_info(name, age, gender, language):
+    user_info.update({"name": name, "age": age, "gender": gender, "language": language})
+    return f"Info saved: {name}, {age}, {gender}, {language}"
 
-def get_study_tip(index=[0]):
-    tip = study_tips[index[0] % len(study_tips)]
-    index[0] += 1
-    return tip
+# Game iframe
+def game_iframe():
+    return '<iframe src="https://onlinegames.io/" width="100%" height="600px" frameborder="0"></iframe>'
 
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("## 🧠 StrongMind Therapist 2.0")
-    gr.Markdown("Welcome to your peaceful space to talk, journal, and focus.")
+# Pomodoro Timer
+def pomodoro_timer():
+    return "Use an external timer or focus for 25 minutes, then rest for 5 minutes. Repeat."
 
+# TABS
+with gr.Blocks(theme=gr.themes.Soft(), css="body {background-color: black; color: white;} .gr-button {background-color: pink;}") as main_app:
     with gr.Tabs():
-        with gr.Tab("Login"):
+        with gr.Tab("Personal Info"):
             name = gr.Textbox(label="Name")
             age = gr.Textbox(label="Age")
-            gender = gr.Dropdown(["Male", "Female", "Other"], label="Gender")
-            language = gr.Dropdown(["english", "hindi"], label="Preferred Language")
-            login_btn = gr.Button("Login")
-            welcome_msg = gr.Textbox(label="Welcome Message", interactive=False)
-            login_btn.click(save_user_info, inputs=[name, age, gender, language], outputs=welcome_msg)
-
-        with gr.Tab("Personal Info"):
-            def display_info():
-                now = datetime.datetime.now().strftime("%A, %d %B %Y")
-                return f"{now}\nName: {user_info['name']}\nAge: {user_info['age']}\nGender: {user_info['gender']}\nLanguage: {user_info['language']}"
-            info_display = gr.Textbox(label="User Info", interactive=False)
-            gr.Button("Show Info").click(display_info, outputs=info_display)
+            gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
+            language = gr.Dropdown(["English", "Hindi"], label="Language", value="English")
+            save_btn = gr.Button("Save Info")
+            output = gr.Textbox(label="Saved Info")
+            save_btn.click(save_personal_info, inputs=[name, age, gender, language], outputs=output)
 
         with gr.Tab("Chat"):
-            gr.Markdown("🗨️ Talk to me. How was your day?")
-            user_input = gr.Textbox(label="You")
-            bot_response = gr.Textbox(label="Therapist", interactive=False)
-            bot_audio = gr.Audio(label="Voice", interactive=False, type="filepath")
-            chat_btn = gr.Button("Send")
-            chat_btn.click(chat_function, inputs=user_input, outputs=[bot_response, bot_audio])
-
-        with gr.Tab("Chat History"):
-            history_output = gr.Textbox(label="Past Chats", lines=10)
-            gr.Button("View").click(view_chat_history, outputs=history_output)
+            chat_input = gr.Textbox(label="How do you feel today?")
+            chat_output = gr.Textbox(label="Therapist says:")
+            audio_output = gr.Audio(label="Voice Response", autoplay=True)
+            send_btn = gr.Button("Send")
+            send_btn.click(chat_interface, inputs=chat_input, outputs=[chat_output, audio_output])
 
         with gr.Tab("Journal"):
-            journal_input = gr.Textbox(label="Your Entry", lines=5)
-            journal_output = gr.Textbox(label="Save Status")
-            gr.Button("Save Entry").click(save_journal, inputs=journal_input, outputs=journal_output)
+            journal_input = gr.Textbox(label="Write your thoughts here...", lines=5)
+            save_journal_btn = gr.Button("Save Entry")
+            save_status = gr.Textbox(label="Status")
+            save_journal_btn.click(save_journal, inputs=journal_input, outputs=save_status)
 
         with gr.Tab("Journal History"):
-            journal_view = gr.Textbox(label="All Entries", lines=10)
-            gr.Button("View").click(view_journal, outputs=journal_view)
+            history_output = gr.Textbox(label="Your Journal Entries", lines=10)
+            show_btn = gr.Button("Show History")
+            show_btn.click(show_journal_history, outputs=history_output)
 
-        with gr.Tab("Pomodoro Timer"):
-            gr.Markdown("⏲️ 25 min focus + 5 min break. Set your timer!")
-
-        with gr.Tab("Calm Space"):
-            calm_output = gr.Textbox(label="Calm Tip", interactive=False)
-            gr.Button("Show Calm Tip").click(get_calm_tip, outputs=calm_output)
+        with gr.Tab("Calm Tips"):
+            calm_output = gr.Textbox(label="Try this...")
+            next_btn = gr.Button("Next Tip")
+            next_btn.click(next_calm_tip, outputs=calm_output)
 
         with gr.Tab("Study Tips"):
-            tip_output = gr.Textbox(label="Tip", interactive=False)
-            gr.Button("Study Tip").click(get_study_tip, outputs=tip_output)
+            tip_output = gr.Textbox(label="Study Tip")
+            tip_btn = gr.Button("Get Tip")
+            tip_btn.click(lambda: study_tips[datetime.datetime.now().second % len(study_tips)], outputs=tip_output)
+
+        with gr.Tab("Pomodoro Timer"):
+            pomo_output = gr.Textbox(label="Pomodoro Info")
+            pomo_btn = gr.Button("Show Timer Tip")
+            pomo_btn.click(pomodoro_timer, outputs=pomo_output)
 
         with gr.Tab("Games"):
-            gr.Markdown("🎮 Memory | Dress Up | Driving | Relaxing")
-            gr.HTML("""<iframe src="https://www.crazygames.com" width="100%" height="500px"></iframe>""")
+            gr.HTML(game_iframe())
+
+# WELCOME SCREEN
+with gr.Blocks(theme=gr.themes.Soft(), css="body {background-color: black; color: white;} .gr-button {background-color: pink;}") as welcome_screen:
+    gr.Markdown("<h1 style='text-align: center; color: white;'>StrongMind Therapist 2.0</h1>")
+    gr.Markdown("<h3 style='text-align: center; color: white;'>🧠 Welcome to StrongMind Therapist! Your peaceful space to talk, journal, and focus.</h3>")
+    get_started_btn = gr.Button("Get Started", elem_id="get-started", scale=2)
+
+    def show_main_app():
+        main_app.launch(share=False)
+
+    get_started_btn.click(fn=None, inputs=None, outputs=None, _js="() => { document.body.innerHTML = ''; }", queue=False).then(fn=show_main_app)
 
-app.launch()
+# RUN APP
+welcome_screen.launch()