Update app.py

app.py

@@ -1,15 +1,22 @@
 import gradio as gr
 import google.generativeai as genai
 import os
-from datetime import datetime
+from datetime import datetime, timedelta
 import plotly.express as px
 import pandas as pd
 from langdetect import detect
 import random
+import smtplib
+from email.message import EmailMessage
+import re
 
 # Configure Gemini API (expects GEMINI_API_KEY in Hugging Face secrets)
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
+# Gmail SMTP configuration (expects GMAIL_ADDRESS and GMAIL_APP_PASSWORD in secrets)
+GMAIL_ADDRESS = os.getenv("GMAIL_ADDRESS")
+GMAIL_APP_PASSWORD = os.getenv("GMAIL_APP_PASSWORD")
+
 # Coping Strategies Library
 coping_strategies = {
     "happy": [
@@ -57,14 +64,36 @@ regional_resources = {
     ]
 }
 
+# Mock Therapist Database
+therapists = {
+    "Dr. Jane Smith": {
+        "specialty": "Anxiety and Depression",
+        "email": "[email protected]",
+        "times": ["09:00", "11:00", "14:00", "16:00"]
+    },
+    "Dr. Amit Patel": {
+        "specialty": "Stress Management",
+        "email": "[email protected]",
+        "times": ["10:00", "12:00", "15:00", "17:00"]
+    },
+    "Dr. Sarah Brown": {
+        "specialty": "Trauma and PTSD",
+        "email": "[email protected]",
+        "times": ["08:00", "13:00", "15:30", "18:00"]
+    }
+}
+
 # Initialize Gemini model
 model = genai.GenerativeModel("gemini-1.5-flash", 
                              generation_config={"temperature": 0.7, "max_output_tokens": 200})
 
 # Chatbot function
-def chatbot_function(message, history, mood, conversation_mode, region, state):
-    if history is None:
-        history = []
+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
     try:
@@ -117,6 +146,7 @@ def chatbot_function(message, history, mood, conversation_mode, region, state):
     
     # Update history
     history[-1][1] = response_text
+    state["chat_history"] = history
     
     # Format history for display
     chat_display = ""
@@ -126,7 +156,12 @@ def chatbot_function(message, history, mood, conversation_mode, region, state):
         if bot_msg:
             chat_display += f"**Bot**: {bot_msg}\n\n"
     
-    return chat_display, history, state
+    return chat_display, state
+
+# Clear chat history
+def clear_chat(state):
+    state["chat_history"] = []
+    return "", state
 
 # Mood journal function
 def log_mood(mood, state):
@@ -160,26 +195,93 @@ def show_emergency_resources():
     - [MentalHealth.gov](https://www.mentalhealth.gov/)
     """
 
-# 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:
+# Get available times for selected therapist
+def get_available_times(therapist):
+    if therapist and therapist in therapists:
+        return therapists[therapist]["times"], f"Available times for {therapist} loaded."
+    return [], "Please select a therapist."
+
+# Schedule appointment
+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:
+        return "Please select a time slot.", state
+    if not date:
+        return "Please select a date.", state
+    if not user_email or not re.match(r"[^@]+@[^@]+\.[^@]+", user_email):
+        return "Please enter a valid email address.", 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
+    
+    # Store appointment
+    appointment = {
+        "therapist": therapist,
+        "time": time_slot,
+        "date": date,
+        "user_email": user_email,
+        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    }
+    state["appointments"].append(appointment)
+    
+    # Send email to therapist
+    therapist_email = therapists[therapist]["email"]
+    msg_therapist = EmailMessage()
+    msg_therapist.set_content(f"""
+    Dear {therapist},
+    
+    You have a new appointment scheduled:
+    - Date: {date}
+    - Time: {time_slot}
+    - Client Email: {user_email}
+    
+    Please confirm with the client.
+    
+    Best regards,
+    Mental Health Chatbot
+    """)
+    msg_therapist["Subject"] = "New Appointment Scheduled"
+    msg_therapist["From"] = GMAIL_ADDRESS
+    msg_therapist["To"] = therapist_email
+    
+    # Send email to user
+    msg_user = EmailMessage()
+    msg_user.set_content(f"""
+    Dear User,
+    
+    Your appointment has been booked:
+    - Therapist: {therapist}
+    - Date: {date}
+    - Time: {time_slot}
+    
+    You will receive a confirmation from {therapist} soon.
+    
+    Best regards,
+    Mental Health Chatbot
+    """)
+    msg_user["Subject"] = "Appointment Confirmation"
+    msg_user["From"] = GMAIL_ADDRESS
+    msg_user["To"] = user_email
+    
+    # Send emails
+    try:
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+            server.login(GMAIL_ADDRESS, GMAIL_APP_PASSWORD)
+            server.send_message(msg_therapist)
+            server.send_message(msg_user)
+        return f"Appointment booked with {therapist} on {date} at {time_slot}. Confirmation emails sent!", state
+    except Exception as e:
+        return f"Failed to send emails: {str(e)}. Appointment stored but emails not sent.", state
+
+# Gradio Interface
+with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     # Initialize state
-    state = gr.State({"mood_journal": [], "feedback": []})
-    history = gr.State()
-    
-    # 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])
+    state = gr.State({"chat_history": [], "mood_journal": [], "feedback": [], "appointments": []})
     
     gr.Markdown("# Mental Health Support Chatbot")
     gr.Markdown("I'm here to listen and support you. Feel free to share how you're feeling.")
@@ -239,21 +341,43 @@ with gr.Blocks(css=css, title="Mental Health Support Chatbot") as demo:
         feedback_btn = gr.Button("Submit Feedback")
         feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
     
+    # Appointment Scheduling
+    with gr.Accordion("Schedule Appointment"):
+        therapist = gr.Dropdown(
+            choices=list(therapists.keys()),
+            label="Select a Therapist"
+        )
+        time_slot = gr.Dropdown(
+            choices=[],
+            label="Select a Time Slot",
+            interactive=True
+        )
+        date = gr.Textbox(
+            label="Appointment Date (YYYY-MM-DD)",
+            placeholder="e.g., 2025-04-20"
+        )
+        user_email = gr.Textbox(
+            label="Your Email Address",
+            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, history, mood, conversation_mode, region, state],
-        outputs=[chatbot, history, state]
+        inputs=[user_input, mood, conversation_mode, region, state],
+        outputs=[chatbot, state]
     )
     user_input.submit(
         fn=chatbot_function,
-        inputs=[user_input, history, mood, conversation_mode, region, state],
-        outputs=[chatbot, history, state]
+        inputs=[user_input, mood, conversation_mode, region, state],
+        outputs=[chatbot, state]
     )
     clear_btn.click(
-        fn=lambda: ("", [], state),
-        inputs=None,
-        outputs=[chatbot, history, state]
+        fn=clear_chat,
+        inputs=[state],
+        outputs=[chatbot, state]
     )
     emergency_btn.click(
         fn=show_emergency_resources,
@@ -275,6 +399,16 @@ with gr.Blocks(css=css, title="Mental Health Support Chatbot") as demo:
         inputs=[feedback_text, feedback_rating, state],
         outputs=[feedback_output, state]
     )
+    therapist.change(
+        fn=get_available_times,
+        inputs=therapist,
+        outputs=[time_slot, schedule_output]
+    )
+    schedule_btn.click(
+        fn=schedule_appointment,
+        inputs=[therapist, time_slot, date, user_email, state],
+        outputs=[schedule_output, state]
+    )
 
     # Resource Links
     gr.Markdown("""