Update app.py

app.py

@@ -14,10 +14,10 @@ from email.mime.text import MIMEText
 import re
 import json
 
-# Configure Gemini API (expects GEMINI_API_KEY in Hugging Face secrets)
+# Configure Gemini API
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-# Gmail API configuration (expects GMAIL_ADDRESS, GMAIL_TOKEN_JSON in secrets)
+# Gmail API configuration
 GMAIL_ADDRESS = os.getenv("GMAIL_ADDRESS")
 GMAIL_TOKEN_JSON = os.getenv("GMAIL_TOKEN_JSON")
 
@@ -25,19 +25,19 @@ GMAIL_TOKEN_JSON = os.getenv("GMAIL_TOKEN_JSON")
 def get_gmail_service():
     try:
         creds = Credentials.from_authorized_user_info(json.loads(GMAIL_TOKEN_JSON))
-        return build("gmail", "v1", credentials=creds)
+        return build("gmail", "v1", credentials=creds), None
     except Exception as e:
-        raise Exception(f"Failed to initialize Gmail API: {str(e)}")
+        return None, f"Failed to initialize Gmail API: {str(e)}"
 
 # Coping Strategies Library
 coping_strategies = {
     "happy": [
-        "Keep the positivity flowing! Try writing down three things you’re grateful for today.",
+        "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."
+        "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.",
@@ -47,9 +47,25 @@ coping_strategies = {
         "Pause for a moment and stretch your body to release tension.",
         "Break tasks into smaller steps and tackle one at a time."
     ],
+    "angry": [
+        "Take a moment to breathe deeply and count to ten to cool down.",
+        "Channel your energy into a physical activity like a brisk walk."
+    ],
+    "overwhelmed": [
+        "Make a small list of priorities to focus on one thing at a time.",
+        "Take a short break to clear your mind with a mindfulness exercise."
+    ],
+    "hopeful": [
+        "Nurture that spark! Set a small goal for today to keep the momentum going.",
+        "Reflect on what’s inspiring you and share it with someone you trust."
+    ],
+    "lonely": [
+        "Reach out to someone you care about, even a quick message can help.",
+        "Do something kind for yourself, like enjoying a favorite hobby."
+    ],
     "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."
+        "Reflect on what's on your mind with a quick mindfulness moment: focus on your breath for 60 seconds.",
+        "Engage in a favorite activity to lift your spirits."
     ]
 }
 
@@ -81,17 +97,17 @@ therapists = {
     "Dr. Jane Smith": {
         "specialty": "Anxiety and Depression",
         "email": "[email protected]",
-        "times": ["09:00", "11:00", "14:00", "16:00"]
+        "times": ["09:00", "09:30", "10:00", "10:30", "11:00", "14:00", "14:30", "15:00", "15:30"]
     },
     "Dr. Amit Patel": {
         "specialty": "Stress Management",
         "email": "[email protected]",
-        "times": ["10:00", "12:00", "15:00", "17:00"]
+        "times": ["10:00", "10:30", "11:00", "11:30", "12:00", "15:00", "15:30", "16:00", "16:30"]
     },
     "Dr. Sarah Brown": {
         "specialty": "Trauma and PTSD",
         "email": "[email protected]",
-        "times": ["08:00", "13:00", "15:30", "18:00"]
+        "times": ["08:00", "08:30", "09:00", "13:00", "13:30", "14:00", "14:30", "15:00", "15:30"]
     }
 }
 
@@ -101,66 +117,60 @@ model = genai.GenerativeModel("gemini-1.5-flash",
 
 # Chatbot function
 def chatbot_function(message, mood, conversation_mode, region, state):
-    # Initialize chat history if not present
     if "chat_history" not in state:
         state["chat_history"] = []
     
     history = state["chat_history"]
     
-    # Multilingual Support: Detect input language
+    if not message.strip():
+        return "Please share how you're feeling, and I'm here to listen.", state
+    
     try:
-        lang = detect(message) if message.strip() else "en"
+        lang = detect(message)
     except:
         lang = "en"
     
-    # Append mood to history if provided
-    if mood and mood != "Select mood (optional)":
+    if mood and mood != "Select mood":
         history.append([f"I'm feeling {mood.lower()}.", None])
     
-    # Append user message
     history.append([message, None])
     
-    # Themed Conversation Modes: Adjust tone
     tone_instruction = {
+        "Neutral": "Maintain a balanced, empathetic tone.",
         "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."
+        "Empathetic": "Show deep understanding and validate the user's feelings.",
+        "Hopeful": "Adopt an optimistic tone, focusing on possibilities and growth."
     }.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}).
+    You are a compassionate mental health support chatbot, acting as a supportive friend. Focus exclusively on mental health, self-motivation, and uplifting the user, especially if they're feeling down. 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.
+    - Suggestions for coping strategies if distress is detected, tailored to their mood or input.
+    - Encouragement to build resilience and connection.
+    - Avoid unrelated topics (e.g., academics, tech). Keep responses concise, warm, and friendly.
     """
     
-    # Generate response
     try:
         response = model.generate_content(prompt)
         response_text = response.text
-    except Exception:
-        response_text = "I'm here for you. Could you share a bit more so I can support you better?"
+    except Exception as e:
+        response_text = "I'm here for you, but something went wrong. Want to share more?"
     
-    # Add coping strategy if mood is provided
-    if mood and mood != "Select mood (optional)":
+    if mood and mood != "Select mood":
         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])
+    resources = "\n\n**Resources for You**:\n" + "\n".join(regional_resources[region_key])
     response_text += resources
     
-    # Update history
     history[-1][1] = response_text
     state["chat_history"] = history
     
-    # Format history for display
     chat_display = ""
     for user_msg, bot_msg in history:
         if user_msg:
@@ -175,20 +185,23 @@ def clear_chat(state):
     state["chat_history"] = []
     return "", state
 
-# Mood journal function
+# Mood journal function with auto-update trigger
 def log_mood(mood, state):
-    if mood and mood != "Select mood (optional)":
+    if mood and mood != "Select mood":
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         state["mood_journal"].append({"timestamp": timestamp, "mood": mood.lower()})
-        return "Mood logged successfully!", state
-    return "Please select a mood to log.", state
+        # Trigger mood trends update
+        trend_output, state = show_mood_trends(state)
+        return "Mood logged!", trend_output, state
+    return "Please select a mood.", None, state
 
-# Mood trend visualization
+# Mood trend visualization (square chart)
 def show_mood_trends(state):
     if not state["mood_journal"]:
         return "No moods logged yet.", state
     df = pd.DataFrame(state["mood_journal"])
-    fig = px.line(df, x="timestamp", y="mood", title="Mood Trends Over Time", markers=True)
+    fig = px.line(df, x="timestamp", y="mood", title="Your Mood Trends", markers=True)
+    fig.update_layout(width=500, height=500, margin=dict(l=40, r=40, t=40, b=40))
     return fig, state
 
 # Feedback function
@@ -198,20 +211,16 @@ def submit_feedback(feedback_text, rating, state):
         return "Thank you for your feedback!", state
     return "Please provide feedback or a rating.", state
 
-# Emergency resources
-def show_emergency_resources():
-    return """
-    **Crisis Support:**
-    - National Suicide Prevention Lifeline: 1-800-273-8255
-    - Crisis Text Line: Text HOME to 741741
-    - [MentalHealth.gov](https://www.mentalhealth.gov/)
-    """
+# Emergency resources (region-specific)
+def show_emergency_resources(region):
+    region_key = region if region in regional_resources else "Global"
+    return f"**Crisis Support for {region_key}**:\n" + "\n".join(regional_resources[region_key])
 
 # Get available times for selected therapist
 def get_available_times(therapist):
     if therapist and therapist in therapists:
-        return gr.update(choices=therapists[therapist]["times"], value=None), f"Available times for {therapist} loaded."
-    return gr.update(choices=[], value=None), "Please select a therapist."
+        return therapists[therapist]["times"], f"Available times for {therapist} loaded."
+    return ["Select time"], "Please select a therapist."
 
 # Create MIME message for Gmail API
 def create_message(to, subject, message_text):
@@ -227,7 +236,6 @@ def send_emails(therapist, time_slot, date, user_email, therapist_email, state):
     if "failed_emails" not in state:
         state["failed_emails"] = []
     
-    # Therapist email
     therapist_body = f"""
 Dear {therapist},
 
@@ -243,30 +251,34 @@ Mental Health Chatbot
     """
     therapist_message = create_message(therapist_email, "New Appointment", therapist_body)
     
-    # User email
     user_body = f"""
 Dear User,
 
-Your appointment is booked:
+Your appointment has been successfully booked! We’re here to support your mental health journey. You’ll hear from your therapist soon to confirm the details.
+
+Details:
 - Therapist: {therapist}
 - Date: {date}
 - Time: {time_slot}
 
-Expect a confirmation from {therapist}.
-
 Best,
 Mental Health Chatbot
     """
     user_message = create_message(user_email, "Appointment Confirmation", user_body)
     
+    service, error = get_gmail_service()
+    if not service:
+        state["failed_emails"].append({
+            "therapist_email": {"to": therapist_email, "subject": "New Appointment", "body": therapist_body},
+            "user_email": {"to": user_email, "subject": "Appointment Confirmation", "body": user_body},
+            "error": error,
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        })
+        return False, error
+    
     try:
-        service = get_gmail_service()
-        
-        # Send therapist email
-        therapist_result = service.users().messages().send(userId="me", body=therapist_message).execute()
-        # Send user email
-        user_result = service.users().messages().send(userId="me", body=user_message).execute()
-        
+        service.users().messages().send(userId="me", body=therapist_message).execute()
+        service.users().messages().send(userId="me", body=user_message).execute()
         return True, ""
     except HttpError as e:
         error_msg = f"Gmail API error: {str(e)}"
@@ -291,128 +303,122 @@ Mental Health Chatbot
 def schedule_appointment(therapist, time_slot, date, user_email, state):
     if not therapist or therapist not in therapists:
         return "Please select a therapist.", state
-    if not time_slot or time_slot not in therapists[therapist]["times"]:
-        return "Please select a valid time slot.", state
+    if not time_slot or time_slot == "Select time":
+        return "Please select a time slot.", state
     if not date:
-        return "Please select a date.", state
+        return "Please enter a date.", state
     if not user_email or not re.match(r"[^@]+@[^@]+\.[^@]+", user_email):
-        return "Please enter a valid email address.", state
+        return "Please enter a valid email.", state
     
     try:
         appointment_date = datetime.strptime(date, "%Y-%m-%d")
         if appointment_date < datetime.now():
             return "Please select a future date.", state
     except ValueError:
-        return "Invalid date format. Use YYYY-MM-DD.", state
+        return "Invalid date format (use YYYY-MM-DD).", state
+    
+    try:
+        appointment_time = datetime.strptime(time_slot, "%H:%M").strftime("%H:%M")
+        if appointment_time not in therapists[therapist]["times"]:
+            return f"Time {appointment_time} is not available for {therapist}.", state
+    except ValueError:
+        return "Invalid time format (use HH:MM).", state
     
-    # Store appointment
     appointment = {
         "therapist": therapist,
-        "time": time_slot,
+        "time": appointment_time,
         "date": date,
         "user_email": user_email,
         "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     }
     state["appointments"].append(appointment)
     
-    # Send emails
     therapist_email = therapists[therapist]["email"]
-    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email, state)
+    success, error_msg = send_emails(therapist, appointment_time, date, user_email, therapist_email, state)
     if success:
-        return f"Appointment booked with {therapist} on {date} at {time_slot}. Emails sent from {GMAIL_ADDRESS} to {therapist_email} and {user_email}!", state
+        return f"Appointment booked with {therapist} on {date} at {appointment_time}. Emails sent!", state
     else:
-        return (f"Appointment booked with {therapist} on {date} at {time_slot}. "
-                f"Emails from {GMAIL_ADDRESS} could not be sent. "
-                f"Please email {therapist_email} with your details (date: {date}, time: {time_slot}) "
-                f"and check {user_email} for confirmation (spam/junk)."), state
+        return (f"Appointment booked with {therapist} on {date} at {appointment_time}. "
+                f"Email sending failed: {error_msg}. Please contact {therapist_email}."), state
 
 # Gradio Interface
 with gr.Blocks(title="Mental Health Support Chatbot") as demo:
-    # Initialize state
     state = gr.State({"chat_history": [], "mood_journal": [], "feedback": [], "appointments": [], "failed_emails": []})
     
-    gr.Markdown("# Mental Health Support Chatbot")
-    gr.Markdown("I'm here to listen and support you. Feel free to share how you're feeling.")
+    gr.Markdown("## Mental Health Support Chatbot")
+    gr.Markdown("I'm here to listen, support, and help you feel stronger. Share how you're feeling today.")
     
-    # Mood Indicator
-    mood = gr.Dropdown(
-        choices=["Select mood (optional)", "Happy", "Sad", "Anxious", "Stressed", "Other"],
-        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",
-        value="",
-        interactive=False,
-        lines=10
-    )
-    user_input = gr.Textbox(
-        placeholder="Type your message here...",
-        label="Your Message"
-    )
-    
-    # Buttons
     with gr.Row():
-        submit_btn = gr.Button("Send")
-        clear_btn = gr.Button("Clear Chat")
+        with gr.Column(scale=1):
+            mood = gr.Dropdown(
+                choices=["Select mood", "Happy", "Sad", "Anxious", "Stressed", "Angry", "Overwhelmed", "Hopeful", "Lonely", "Other"],
+                label="Your Mood",
+                value="Select mood"
+            )
+            conversation_mode = gr.Radio(
+                choices=["Neutral", "Calm", "Motivational", "Empathetic", "Hopeful"],
+                label="Conversation Style",
+                value="Neutral"
+            )
+            region = gr.Dropdown(
+                choices=["USA", "India", "UK", "Global"],
+                label="Your Region",
+                value="Global"
+            )
+        with gr.Column(scale=2):
+            chatbot = gr.Textbox(
+                label="Our Conversation",
+                value="",
+                interactive=False,
+                lines=12,
+                show_copy_button=True
+            )
+            user_input = gr.Textbox(
+                placeholder="Tell me how you're feeling...",
+                label="Your Message",
+                lines=3
+            )
+            with gr.Row():
+                submit_btn = gr.Button("Send Message")
+                clear_btn = gr.Button("Clear Chat")
     
-    # Emergency Resources
-    emergency_btn = gr.Button("Emergency Resources")
+    emergency_btn = gr.Button("Crisis Support")
     emergency_output = gr.Markdown("")
     
-    # 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()
+        mood_trend_output = gr.Plot(label="Your Mood Trends")
     
-    # 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")
+    with gr.Accordion("Share Feedback"):
+        feedback_text = gr.Textbox(label="Your Thoughts", placeholder="How can we make this better?")
+        feedback_rating = gr.Slider(minimum=1, maximum=5, step=1, label="How was this interaction?")
         feedback_btn = gr.Button("Submit Feedback")
         feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
     
-    # Appointment Scheduling
-    with gr.Accordion("Schedule Appointment"):
+    with gr.Accordion("Book an Appointment"):
         therapist = gr.Dropdown(
             choices=list(therapists.keys()),
-            label="Select a Therapist"
-        )
-        time_slot = gr.Dropdown(
-            choices=[],
-            label="Select a Time Slot",
-            value=None,
-            interactive=True
+            label="Choose a Therapist"
         )
         date = gr.Textbox(
-            label="Appointment Date (YYYY-MM-DD)",
-            placeholder="e.g., 2025-04-20"
+            label="Appointment Date",
+            placeholder="Enter date (YYYY-MM-DD)",
+            info="Use format YYYY-MM-DD, e.g., 2025-04-22"
+        )
+        time_slot = gr.Dropdown(
+            choices=["Select time"],
+            label="Appointment Time",
+            interactive=True,
+            allow_custom_value=True
         )
         user_email = gr.Textbox(
-            label="Your Email Address",
+            label="Your Email",
             placeholder="e.g., [email protected]"
         )
         schedule_btn = gr.Button("Book Appointment")
         schedule_output = gr.Textbox(label="Booking Status", interactive=False)
     
-    # Event Handlers
     submit_btn.click(
         fn=chatbot_function,
         inputs=[user_input, mood, conversation_mode, region, state],
@@ -430,18 +436,13 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     )
     emergency_btn.click(
         fn=show_emergency_resources,
-        inputs=None,
+        inputs=region,
         outputs=emergency_output
     )
     log_mood_btn.click(
         fn=log_mood,
         inputs=[mood, state],
-        outputs=[mood_log_output, state]
-    )
-    mood_trend_btn.click(
-        fn=show_mood_trends,
-        inputs=[state],
-        outputs=[mood_trend_output, state]
+        outputs=[mood_log_output, mood_trend_output, state]
     )
     feedback_btn.click(
         fn=submit_feedback,
@@ -459,15 +460,11 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         outputs=[schedule_output, state]
     )
 
-    # Resource Links
     gr.Markdown("""
-    ---
-    **Helpful Resources:**
-    - [National Suicide Prevention Lifeline](https://suicidepreventionlifeline.org/)
-    - [MentalHealth.gov](https://www.mentalhealth.gov/)
-    - [Crisis Text Line](https://www.crisistextline.org/)
+    **Resources**:
+    - [Befrienders Worldwide](https://befrienders.org/)
+    - [WHO Mental Health](https://www.who.int/health-topics/mental-health)
     """)
 
-# Launch (for local testing)
 if __name__ == "__main__":
     demo.launch()