Update app.py

app.py

@@ -1,47 +1,130 @@
 import gradio as gr
 import google.generativeai as genai
 import os
+import json
+from datetime import datetime
+import plotly.express as px
+import pandas as pd
+from langdetect import detect
+import random
 
 # Configure Gemini API (expects GEMINI_API_KEY in Hugging Face secrets)
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-# Gemini Prompt Template
-prompt_template = """
-You are a compassionate mental health support chatbot. Engage in a supportive conversation with the user based on their input: {user_input}. 
-- Provide empathetic, sensitive responses.
-- If signs of distress are detected, suggest coping strategies like deep breathing or mindfulness.
-- Recommend professional resources or helplines if appropriate, such as the National Suicide Prevention Lifeline (1-800-273-8255) or MentalHealth.gov.
-- Keep responses concise, warm, and encouraging.
-"""
+# Coping Strategies Library
+coping_strategies = {
+    "happy": [
+        "Keep the positivity flowing! Try writing down three things you’re grateful for today.",
+        "Share your joy! Call a friend or loved one to spread the good vibes."
+    ],
+    "sad": [
+        "It’s okay to feel this way. Try a gentle activity like listening to calming music or taking a short walk.",
+        "Write down your thoughts in a journal to process what’s on your mind."
+    ],
+    "anxious": [
+        "Take slow, deep breaths: inhale for 4 seconds, hold for 4, exhale for 4.",
+        "Try a grounding exercise: name 5 things you see, 4 you can touch, 3 you hear, 2 you smell, 1 you taste."
+    ],
+    "stressed": [
+        "Pause for a moment and stretch your body to release tension.",
+        "Break tasks into smaller steps and tackle one at a time."
+    ],
+    "other": [
+        "Reflect on what’s on your mind with a quick mindfulness moment: focus on your breath for 60 seconds.",
+        "Engage in a favorite hobby to lift your spirits."
+    ]
+}
+
+# Regional Resources
+regional_resources = {
+    "USA": [
+        "National Suicide Prevention Lifeline: 1-800-273-8255",
+        "Crisis Text Line: Text HOME to 741741",
+        "[MentalHealth.gov](https://www.mentalhealth.gov/)"
+    ],
+    "India": [
+        "Vandrevala Foundation: 1860-2662-345",
+        "AASRA Suicide Prevention: +91-9820466726",
+        "[iCall Helpline](https://icallhelpline.org/)"
+    ],
+    "UK": [
+        "Samaritans: 116 123",
+        "Shout Crisis Text Line: Text SHOUT to 85258",
+        "[Mind UK](https://www.mind.org.uk/)"
+    ],
+    "Global": [
+        "Befrienders Worldwide: [Find a helpline](https://befrienders.org/)",
+        "[WHO Mental Health Resources](https://www.who.int/health-topics/mental-health)"
+    ]
+}
 
 # Initialize Gemini model
 model = genai.GenerativeModel("gemini-1.5-flash", 
                              generation_config={"temperature": 0.7, "max_output_tokens": 200})
 
+# Mood journal storage (in-memory for simplicity, persists in session)
+if "mood_journal" not in gr.State():
+    gr.State()["mood_journal"] = []
+
+# Feedback storage (in-memory, could be saved to file)
+if "feedback" not in gr.State():
+    gr.State()["feedback"] = []
+
 # Chatbot function
-def chatbot_function(message, history, mood):
-    # Initialize history if None
+def chatbot_function(message, history, mood, conversation_mode, region):
     if history is None:
         history = []
     
+    # Multilingual Support: Detect input language
+    try:
+        lang = detect(message) if message.strip() else "en"
+    except:
+        lang = "en"
+    
     # Append mood to history if provided
     if mood and mood != "Select mood (optional)":
         history.append([f"I'm feeling {mood.lower()}.", None])
     
-    # Append user message to history
+    # Append user message
     history.append([message, None])
     
-    # Create prompt with user input
-    prompt = prompt_template.format(user_input=message)
+    # Themed Conversation Modes: Adjust tone
+    tone_instruction = {
+        "Calm": "Respond in a soothing, gentle tone to promote relaxation.",
+        "Motivational": "Use an uplifting, encouraging tone to inspire confidence.",
+        "Neutral": "Maintain a balanced, empathetic tone."
+    }.get(conversation_mode, "Maintain a balanced, empathetic tone.")
+    
+    # Prepare prompt
+    prompt = f"""
+    You are a compassionate mental health support chatbot. Engage in a supportive conversation with the user based on their input: {message}. 
+    - Provide empathetic, sensitive responses in the user's language (detected as {lang}).
+    - {tone_instruction}
+    - If signs of distress are detected, suggest coping strategies relevant to their mood or input.
+    - Recommend professional resources tailored to the user's region ({region}).
+    - Keep responses concise, warm, and encouraging.
+    """
     
     # Generate response
     try:
         response = model.generate_content(prompt)
         response_text = response.text
-    except Exception as e:
+    except Exception:
         response_text = "I'm here for you. Could you share a bit more so I can support you better?"
     
-    # Update history with bot response
+    # Add coping strategy if mood is provided
+    if mood and mood != "Select mood (optional)":
+        mood_key = mood.lower()
+        if mood_key in coping_strategies:
+            strategy = random.choice(coping_strategies[mood_key])
+            response_text += f"\n\n**Coping Strategy**: {strategy}"
+    
+    # Add regional resources
+    region_key = region if region in regional_resources else "Global"
+    resources = "\n\n**Recommended Resources**:\n" + "\n".join(regional_resources[region_key])
+    response_text += resources
+    
+    # Update history
     history[-1][1] = response_text
     
     # Format history for display
@@ -54,7 +137,30 @@ def chatbot_function(message, history, mood):
     
     return chat_display, history
 
-# Emergency resources function
+# Mood journal function
+def log_mood(mood):
+    if mood and mood != "Select mood (optional)":
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        gr.State()["mood_journal"].append({"timestamp": timestamp, "mood": mood.lower()})
+        return "Mood logged successfully!"
+    return "Please select a mood to log."
+
+# Mood trend visualization
+def show_mood_trends():
+    if not gr.State()["mood_journal"]:
+        return "No moods logged yet."
+    df = pd.DataFrame(gr.State()["mood_journal"])
+    fig = px.line(df, x="timestamp", y="mood", title="Mood Trends Over Time", markers=True)
+    return fig
+
+# Feedback function
+def submit_feedback(feedback_text, rating):
+    if feedback_text or rating:
+        gr.State()["feedback"].append({"text": feedback_text, "rating": rating, "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")})
+        return "Thank you for your feedback!"
+    return "Please provide feedback or a rating."
+
+# Emergency resources
 def show_emergency_resources():
     return """
     **Crisis Support:**
@@ -63,8 +169,23 @@ def show_emergency_resources():
     - [MentalHealth.gov](https://www.mentalhealth.gov/)
     """
 
-# Gradio Interface
-with gr.Blocks(title="Mental Health Support Chatbot") as demo:
+# Gradio Interface with Dark/Light Mode
+css = """
+body { transition: background-color 0.3s, color 0.3s; }
+.light-mode { background-color: #ffffff; color: #000000; }
+.dark-mode { background-color: #1a1a1a; color: #ffffff; }
+button, input, select, textarea { border-radius: 8px; margin: 5px; }
+"""
+with gr.Blocks(css=css, title="Mental Health Support Chatbot") as demo:
+    # Theme Toggle
+    theme_state = gr.State(value="light")
+    def toggle_theme(theme):
+        new_theme = "dark" if theme == "light" else "light"
+        return new_theme, f".{new_theme}-mode"
+    theme_btn = gr.Button("Toggle Dark/Light Mode")
+    theme_output = gr.HTML(value=".light-mode")
+    theme_btn.click(toggle_theme, inputs=theme_state, outputs=[theme_state, theme_output])
+    
     gr.Markdown("# Mental Health Support Chatbot")
     gr.Markdown("I'm here to listen and support you. Feel free to share how you're feeling.")
     
@@ -74,6 +195,20 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         label="How are you feeling today? (Optional)"
     )
     
+    # Conversation Mode
+    conversation_mode = gr.Radio(
+        choices=["Neutral", "Calm", "Motivational"],
+        label="Conversation Style",
+        value="Neutral"
+    )
+    
+    # Region Selector
+    region = gr.Dropdown(
+        choices=["USA", "India", "UK", "Global"],
+        label="Select your region for tailored resources",
+        value="Global"
+    )
+    
     # Chat Interface
     chatbot = gr.Textbox(
         label="Conversation",
@@ -95,18 +230,32 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     emergency_btn = gr.Button("Emergency Resources")
     emergency_output = gr.Markdown("")
     
-    # State to store chat history
+    # Mood Journal
+    with gr.Accordion("Mood Journal"):
+        log_mood_btn = gr.Button("Log Mood")
+        mood_log_output = gr.Textbox(label="Mood Log Status", interactive=False)
+        mood_trend_btn = gr.Button("Show Mood Trends")
+        mood_trend_output = gr.Plot()
+    
+    # Feedback
+    with gr.Accordion("Provide Feedback"):
+        feedback_text = gr.Textbox(label="Your Feedback (Optional)", placeholder="How can we improve?")
+        feedback_rating = gr.Slider(minimum=1, maximum=5, step=1, label="Rate this interaction")
+        feedback_btn = gr.Button("Submit Feedback")
+        feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
+    
+    # State
     history = gr.State()
     
     # Event Handlers
     submit_btn.click(
         fn=chatbot_function,
-        inputs=[user_input, history, mood],
+        inputs=[user_input, history, mood, conversation_mode, region],
         outputs=[chatbot, history]
     )
     user_input.submit(
         fn=chatbot_function,
-        inputs=[user_input, history, mood],
+        inputs=[user_input, history, mood, conversation_mode, region],
         outputs=[chatbot, history]
     )
     clear_btn.click(
@@ -119,6 +268,21 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         inputs=None,
         outputs=emergency_output
     )
+    log_mood_btn.click(
+        fn=log_mood,
+        inputs=mood,
+        outputs=mood_log_output
+    )
+    mood_trend_btn.click(
+        fn=show_mood_trends,
+        inputs=None,
+        outputs=mood_trend_output
+    )
+    feedback_btn.click(
+        fn=submit_feedback,
+        inputs=[feedback_text, feedback_rating],
+        outputs=feedback_output
+    )
 
     # Resource Links
     gr.Markdown("""
@@ -129,6 +293,6 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     - [Crisis Text Line](https://www.crisistextline.org/)
     """)
 
-# Launch (not needed for Hugging Face, included for local testing)
+# Launch (for local testing)
 if __name__ == "__main__":
     demo.launch()