app_clean.py

from gtts import gTTS
import tempfile
import json
import datetime
import gradio as gr
import re
import os
from openai import OpenAI

# OpenAI Configuration
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "your-api-key-here")
client = OpenAI(api_key=OPENAI_API_KEY)

# System prompt for ChatGPT to act as a compassionate therapist
SYSTEM_PROMPT = """You are StrongMind Therapist - a compassionate, empathetic AI therapist designed to provide emotional support and mental health guidance. 

Your role is to:

- Listen actively and validate the person's feelings

- Provide thoughtful, compassionate responses

- Offer practical coping strategies and mental health tips when appropriate

- Be supportive and non-judgmental

- Encourage them to seek professional help if needed

- Keep responses conversational and natural, like talking to a caring human counselor

- Be brief but warm (2-3 sentences typically, unless they ask for more detail)"""

# User data
user_info = {"name": "", "age": "", "gender": "", "language": "english", "Guardian_info": ""}
chat_history = []
journal_entries = []

calm_tips = [
    "Take 3 deep breaths.", "Listen to nature.", "Stretch your body.",
    "Drink water.", "Think of one good thing today.", "Close your eyes for 1 minute.",
    "Write your feelings.", "Smile at yourself.", "Imagine a peaceful place.",
    "Say a positive affirmation."
]

study_tips = [
    "Use Pomodoro: 25min study, 5min break", "Make a daily to-do list",
    "Avoid multitasking", "Use color-coded notes", "Take 10-min exercise breaks",
    "Sleep 7-9 hrs daily", "Drink water during study", "Use active recall",
    "Study hardest topics first", "Test yourself often"
]

tip_index = {"calm": 0, "study": 0}

lang_codes = {
    "english": "en", "hindi": "hi", "marathi": "mr", "bengali": "bn",
    "tamil": "ta", "telugu": "te", "malayalam": "ml", "spanish": "es",
    "french": "fr", "german": "de"
}

harmful_keywords = [
    "abomination", "annihilate", "arson", "assault", "atrocity", "backstab",
    "barbaric", "beast", "belittle", "berserk", "betray", "betrayal", "bigot",
    "blacklist", "blood", "bloodbath", "bloody", "bomb", "brutal", "brute",
    "burn", "butcher", "chaos", "cheater", "cold-blooded", "conman", "corrupt",
    "cowardly", "crash", "crime", "cruel", "cruelty", "damage", "danger",
    "dangerous", "deadly", "deceive", "defame", "defeat", "degenerate", "demon",
    "despise", "destroy", "destruction", "devil", "dirty", "disaster", "dishonor",
    "drunk", "dumbass", "dungeon", "enemy", "enrage", "evil", "exploit",
    "exterminate", "fake", "fanatic", "fascist", "fatal", "fire", "fraud",
    "gang", "gangster", "ghastly", "ghoul", "gory", "grotesque", "gruesome",
    "hag", "halfwit", "harm", "hateful", "havoc", "hell", "hideous", "horrid",
    "horrific", "hostility", "illegal", "illicit", "immoral", "imposter",
    "injure", "insensitive", "intolerant", "irate", "jerky", "joker", "kidnap",
    "liar", "loath", "loathsome", "lowlife", "madness", "malicious", "malign",
    "manipulate", "menace", "monster", "monstrous", "murderous", "mutant",
    "mutiny", "nasty", "nefarious", "neglect", "negative", "offend", "offender",
    "ominous", "oppress", "outrage", "parasite", "penalty", "perish", "pest",
    "plague", "poison", "poisonous", "pollute", "psychopath", "punch", "punish",
    "punishment", "quarrel", "rage", "rat", "rebellion", "reckless", "repel",
    "repulsive", "revolt", "rioter", "rob", "robber", "rotten", "ruin", "savage",
    "scare", "scary", "schemer", "scold", "scorn", "scoundrel", "scream",
    "screwed", "serpent", "shady", "sham", "shameful", "shatter", "shocking",
    "sin", "sinister", "slimy", "smash", "smite", "snob", "spite", "spiteful",
    "stab", "stench", "stink", "strike", "subvert", "suspicious", "target",
    "tease", "terrible", "terrify", "thief", "threat", "threaten", "thug",
    "torment", "torture", "trap", "trick", "trigger", "trouble", "unfair",
    "unjust", "unkind", "usurper", "vandal", "vandalize", "venom", "venomous",
    "villain", "villainous", "vindictive", "violent", "viper", "war",
    "warmonger", "waste", "wicked", "wild", "wound", "wrath", "wreck", "wrong",
    "yell", "zombie", "suicide", "kill myself", "end my life", "harm myself",
    "cut myself", "want to die", "die", "jump off", "self-harm", "give up",
    "not worth it", "ending it all", "self-hate", "suicidal", "tragic", "trauma",
    "pain", "painful", "sorrow", "unworthy", "hurt", "grief", "agony", "doom",
    "doomed"
]

harmful_response = (
    "⚠️ It sounds like you're going through a really tough time.\n\n"
    "Please reach out to a professional:\n\n"
    "🇮🇳 **India Helplines**\n"
    "🧠 *Dr. Rachna Khanna Singh* – +91 99103 90559\n"
    "📞 *iCall Helpline* – +91 9152987821\n"
    "📞 *Vandrevala Foundation* – 1860 266 2345 or 1800 233 3330\n\n"
    "🌍 **International Helplines**\n"
    "📞 *Lifeline (USA)* – 988\n"
    "📞 *Samaritans (UK)* – 116 123\n"
    "📞 *Lifeline (Australia)* – 13 11 14\n\n"
    "You are not alone. There are people who care and want to help 💚"
)

def set_personal_info(name, age, gender, language, Guardian_info):
    user_info.update({"name": name, "age": age, "gender": gender, "language": language, "Guardian_info": Guardian_info})
    return gr.update(visible=True), f"✅ Welcome {name}! Preferences saved."

def show_personal_data():
    today = datetime.date.today().strftime("%Y-%m-%d (%A)")
    return f"📅 {today}\n👤 Name: {user_info['name']}\n🎂 Age: {user_info['age']}\n♀ Gender: {user_info['gender']}\n🌐 Language: {user_info['language']}\n Guardian: {user_info['Guardian_info']}"

def generate_reply(input_text):
    text = input_text.lower()
    for word in harmful_keywords:
        pattern = r'\b' + re.escape(word) + r'\b'
        if re.search(pattern, text):
            return harmful_response
    
    try:
        response = client.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user", "content": input_text}
            ],
            temperature=0.7,
            max_tokens=200
        )
        return response.choices[0].message.content
    except Exception as e:
        return "I'm having trouble connecting right now, but I'm here to listen. Can you tell me more about what you're feeling?"

def chat_function(audio_input, text_input):
    user_text = text_input.strip()
    if not user_text:
        return "Please type something or try again.", None
    try:
        reply = generate_reply(user_text)
        chat_history.append({"user": user_text, "bot": reply})
        lang_code = lang_codes.get(user_info["language"], "en")
        tts = gTTS(reply, lang=lang_code)
        audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
        tts.save(audio_file.name)
        return reply, audio_file.name
    except Exception as e:
        return f"Something went wrong: {str(e)}", None

def get_chat_history():
    if not chat_history:
        return "No conversations yet."
    return "\n\n".join([f"You: {c['user']}\nBot: {c['bot']}" for c in chat_history])
    
def save_journal(entry):
    if not entry.strip():
        return "⚠️ Please write something before saving."
    try:
        journal_entries.append(entry)
        with open("journal.json", "w") as f:
            json.dump(journal_entries, f, indent=2)
        return "✅ Journal entry saved successfully!"
    except Exception as e:
        return f"❌ Error saving journal: {str(e)}"

def show_journal_history():
    return "\n---\n".join(journal_entries) if journal_entries else "No journal entries yet."

def next_calm_tip():
    tip = calm_tips[tip_index["calm"] % len(calm_tips)]
    tip_index["calm"] += 1
    return tip

def next_study_tip():
    tip = study_tips[tip_index["study"] % len(study_tips)]
    tip_index["study"] += 1
    return tip

# Gradio App
with gr.Blocks() as app:
    welcome_screen = gr.Column(visible=True)
    full_app = gr.Tabs(visible=False)

    with welcome_screen:
        gr.HTML("""

            <style>

                #main-title { text-align: center; font-size: 2.5em; }

                #subtitle { text-align: center; font-size: 1.2em; color: #ccc; margin-bottom: 30px; }

                #start-btn { background-color: #ff69b4; color: white; font-weight: bold; border: none; border-radius: 10px; padding: 14px 45px; font-size: 1.1em; cursor: pointer; display: block; margin: 0 auto; }

                #start-btn:hover { background-color: #ff85c1; }

            </style>

        """)
        gr.Markdown("<h1 id='main-title'>StrongMind Therapist 2.0</h1>")
        gr.Markdown("<h3 id='subtitle'>Your peaceful space to talk, journal, and focus.</h3>")
        start_button = gr.Button("🌸 Get Started", elem_id="start-btn")

    def start_app():
        return gr.update(visible=False), gr.update(visible=True)

    start_button.click(start_app, outputs=[welcome_screen, full_app])

    with full_app:
        with gr.Tab("1️⃣ Personal Info"):
            name = gr.Textbox(label="Name")
            age = gr.Textbox(label="Age")
            gender = gr.Dropdown(["Male", "Female", "Other"], label="Gender")
            language = gr.Dropdown(list(lang_codes.keys()), label="Preferred Language")
            Guardian_info = gr.Textbox(label="Guardian information")
            btn = gr.Button("Save Info")
            popup = gr.Markdown(visible=False, elem_classes="alert-box")
            btn.click(set_personal_info, [name, age, gender, language, Guardian_info], [popup, popup])

        with gr.Tab("2️⃣ Personal Info Data"):
            show = gr.Button("Show My Info")
            info_display = gr.Textbox(lines=6)
            show.click(show_personal_data, outputs=info_display)

        with gr.Tab("3️⃣ Chat"):
            gr.Markdown("🗣️ Describe your day in one word.")
            audio_input = gr.Audio(type="filepath", label="🎧 Say something")
            text_input = gr.Textbox(label="⌨️ Or type here")
            send = gr.Button("Send")
            bot_reply = gr.Textbox(label="🧠 Therapist")
            voice = gr.Audio(label="🔊 Voice Reply")
            send.click(chat_function, [audio_input, text_input], [bot_reply, voice])

        with gr.Tab("4️⃣ Chat History"):
            show_history = gr.Button("📜 Show Chats")
            chat_out = gr.Textbox(lines=20, label="History")
            show_history.click(get_chat_history, outputs=chat_out)

        with gr.Tab("5️⃣ Journal"):
            journal_input = gr.Textbox(lines=6, label="Write your thoughts")
            save = gr.Button("Save")
            journal_status = gr.Textbox()
            save.click(save_journal, journal_input, journal_status)

        with gr.Tab("6️⃣ Journal History"):
            view = gr.Button("View Past Entries")
            past = gr.Textbox(lines=15, label="Previous Journals")
            view.click(show_journal_history, outputs=past)

        with gr.Tab("7️⃣ Calm Space"):
            tip_btn = gr.Button("🌿 Give Me a Calm Tip")
            calm_text = gr.Textbox()
            tip_btn.click(next_calm_tip, outputs=calm_text)

        with gr.Tab("8️⃣ Study Tips"):
            tip_btn2 = gr.Button("📚 Study Tip")
            study_text = gr.Textbox()
            tip_btn2.click(next_study_tip, outputs=study_text)

        with gr.Tab("9️⃣ Pomodoro"):
            gr.Markdown("⏱️ Use 25 min study + 5 min break cycles.\n(Use Android timers or JS for real timer)")
            
        with gr.Tab("💠 Games"):
            gr.Markdown("🎮 Play from the game portal below:")
            gr.HTML('<iframe src="https://www.onlinegames.io/embed/portal/" width="100%" height="600" frameborder="0" allowfullscreen></iframe>')

if __name__ == "__main__":
    app.launch()