Update app.py

app.py

@@ -14,10 +14,10 @@ from email.mime.text import MIMEText
 import re
 import json
 
-# Configure Gemini API
+# Configure Gemini API (expects GEMINI_API_KEY in Hugging Face secrets)
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-# Gmail API configuration
+# Gmail API configuration (expects GMAIL_ADDRESS, GMAIL_TOKEN_JSON in secrets)
 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), None
+        return build("gmail", "v1", credentials=creds)
     except Exception as e:
-        return None, f"Failed to initialize Gmail API: {str(e)}"
+        raise Exception(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.",
@@ -48,7 +48,7 @@ coping_strategies = {
         "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.",
+        "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."
     ]
 }
@@ -81,17 +81,17 @@ therapists = {
     "Dr. Jane Smith": {
         "specialty": "Anxiety and Depression",
         "email": "[email protected]",
-        "times": ["09:00", "09:30", "10:00", "10:30", "11:00", "14:00", "14:30", "15:00", "16:00"]
+        "times": ["09:00", "11:00", "14:00", "16:00"]
     },
     "Dr. Amit Patel": {
         "specialty": "Stress Management",
         "email": "[email protected]",
-        "times": ["10:00", "10:30", "11:00", "11:30", "12:00", "15:00", "15:30", "16:00", "17:00"]
+        "times": ["10:00", "12:00", "15:00", "17:00"]
     },
     "Dr. Sarah Brown": {
         "specialty": "Trauma and PTSD",
         "email": "[email protected]",
-        "times": ["08:00", "08:30", "09:00", "13:00", "13:30", "14:00", "15:30", "16:00", "18:00"]
+        "times": ["08:00", "13:00", "15:30", "18:00"]
     }
 }
 
@@ -101,30 +101,33 @@ 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"]
     
-    if not message.strip():
-        return "Please enter a message.", state
-    
+    # Multilingual Support: Detect input language
     try:
-        lang = detect(message)
+        lang = detect(message) if message.strip() else "en"
     except:
         lang = "en"
     
-    if mood and mood != "Select mood":
+    # 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
     history.append([message, None])
     
+    # 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}).
@@ -134,25 +137,30 @@ def chatbot_function(message, mood, conversation_mode, region, state):
     - Keep responses concise, warm, and encouraging.
     """
     
+    # Generate response
     try:
         response = model.generate_content(prompt)
         response_text = response.text
-    except Exception as e:
-        response_text = "I'm here for you, but something went wrong. Please try again."
+    except Exception:
+        response_text = "I'm here for you. Could you share a bit more so I can support you better?"
     
-    if mood and mood != "Select mood":
+    # 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**Resources**:\n" + "\n".join(regional_resources[region_key])
+    resources = "\n\n**Recommended Resources**:\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:
@@ -169,25 +177,25 @@ def clear_chat(state):
 
 # Mood journal function
 def log_mood(mood, state):
-    if mood and mood != "Select mood":
+    if mood and mood != "Select mood (optional)":
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         state["mood_journal"].append({"timestamp": timestamp, "mood": mood.lower()})
-        return "Mood logged!", state
-    return "Please select a mood.", state
+        return "Mood logged successfully!", state
+    return "Please select a mood to log.", state
 
 # Mood trend visualization
 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", markers=True)
+    fig = px.line(df, x="timestamp", y="mood", title="Mood Trends Over Time", markers=True)
     return fig, state
 
 # Feedback function
 def submit_feedback(feedback_text, rating, state):
     if feedback_text or rating:
         state["feedback"].append({"text": feedback_text, "rating": rating, "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")})
-        return "Feedback submitted!", state
+        return "Thank you for your feedback!", state
     return "Please provide feedback or a rating.", state
 
 # Emergency resources
@@ -202,9 +210,8 @@ def show_emergency_resources():
 # Get available times for selected therapist
 def get_available_times(therapist):
     if therapist and therapist in therapists:
-        times = therapists[therapist]["times"]
-        return times, f"Available times for {therapist}: {', '.join(times)}"
-    return [], "Please select a therapist."
+        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."
 
 # Create MIME message for Gmail API
 def create_message(to, subject, message_text):
@@ -220,6 +227,7 @@ 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},
 
@@ -235,6 +243,7 @@ Mental Health Chatbot
     """
     therapist_message = create_message(therapist_email, "New Appointment", therapist_body)
     
+    # User email
     user_body = f"""
 Dear User,
 
@@ -250,19 +259,14 @@ 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.users().messages().send(userId="me", body=therapist_message).execute()
-        service.users().messages().send(userId="me", body=user_message).execute()
+        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()
+        
         return True, ""
     except HttpError as e:
         error_msg = f"Gmail API error: {str(e)}"
@@ -287,160 +291,128 @@ 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:
-        return "Please select a time.", state
+    if not time_slot or time_slot not in therapists[therapist]["times"]:
+        return "Please select a valid 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.", state
+        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
-    
-    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}. Please choose from available times.", state
-    except ValueError:
-        return "Invalid time format (use HH:MM).", state
+        return "Invalid date format. Use YYYY-MM-DD.", state
     
+    # Store appointment
     appointment = {
         "therapist": therapist,
-        "time": appointment_time,
+        "time": time_slot,
         "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, appointment_time, date, user_email, therapist_email, state)
+    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email, state)
     if success:
-        return f"Appointment booked with {therapist} on {date} at {appointment_time}. Emails sent!", state
+        return f"Appointment booked with {therapist} on {date} at {time_slot}. Emails sent from {GMAIL_ADDRESS} to {therapist_email} and {user_email}!", state
     else:
-        return (f"Appointment booked with {therapist} on {date} at {appointment_time}. "
-                f"Email sending failed: {error_msg}. Please contact {therapist_email}."), state
-
-# Minimal CSS for date and time pickers
-custom_css = """
-#date-picker input[type="date"], #time-picker input[type="time"] {
-    width: 100%;
-    padding: 10px;
-    border: 1px solid #d1d5db;
-    border-radius: 6px;
-    font-size: 16px;
-    box-sizing: border-box;
-}
-#date-picker input[type="date"]:focus, #time-picker input[type="time"]:focus {
-    border-color: #3b82f6;
-    outline: none;
-    box-shadow: 0 0 0 2px rgba(59, 130, 246, 0.2);
-}
-#date-picker input[type="date"]::-webkit-calendar-picker-indicator,
-#time-picker input[type="time"]::-webkit-calendar-picker-indicator {
-    cursor: pointer;
-}
-"""
-
-# JavaScript for date and time pickers
-custom_js = """
-document.addEventListener('DOMContentLoaded', function() {
-    const dateInput = document.querySelector('#date-picker input');
-    const timeInput = document.querySelector('#time-picker input');
-    if (dateInput) {
-        dateInput.type = 'date';
-        dateInput.placeholder = 'YYYY-MM-DD';
-    }
-    if (timeInput) {
-        timeInput.type = 'time';
-        timeInput.placeholder = 'HH:MM';
-    }
-});
-"""
+        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
 
 # Gradio Interface
-with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_js) as demo:
+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("A safe space to share your feelings and access support.")
+    gr.Markdown("# Mental Health Support Chatbot")
+    gr.Markdown("I'm here to listen and support you. Feel free to share how you're feeling.")
+    
+    # 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():
-        with gr.Column(scale=1):
-            mood = gr.Dropdown(
-                choices=["Select mood", "Happy", "Sad", "Anxious", "Stressed", "Other"],
-                label="Mood",
-                value="Select mood"
-            )
-            conversation_mode = gr.Radio(
-                choices=["Neutral", "Calm", "Motivational"],
-                label="Conversation Style",
-                value="Neutral"
-            )
-            region = gr.Dropdown(
-                choices=["USA", "India", "UK", "Global"],
-                label="Region",
-                value="Global"
-            )
-        with gr.Column(scale=2):
-            chatbot = gr.Textbox(
-                label="Conversation",
-                value="",
-                interactive=False,
-                lines=10,
-                show_copy_button=True
-            )
-            user_input = gr.Textbox(
-                placeholder="Type your message here...",
-                label="Your Message",
-                lines=2
-            )
-            with gr.Row():
-                submit_btn = gr.Button("Send Message")
-                clear_btn = gr.Button("Clear Chat")
+        submit_btn = gr.Button("Send")
+        clear_btn = gr.Button("Clear Chat")
     
-    emergency_btn = gr.Button("Show Crisis Support")
+    # Emergency Resources
+    emergency_btn = gr.Button("Emergency Resources")
     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("View Mood Trends")
+        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", placeholder="How can we improve?")
-        feedback_rating = gr.Slider(minimum=1, maximum=5, step=1, label="Rating")
+        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)
     
+    # Appointment Scheduling
     with gr.Accordion("Schedule Appointment"):
         therapist = gr.Dropdown(
             choices=list(therapists.keys()),
-            label="Select Therapist"
+            label="Select a Therapist"
         )
-        date = gr.Textbox(
-            label="Appointment Date",
-            placeholder="Select date (YYYY-MM-DD)",
-            elem_id="date-picker"
+        time_slot = gr.Dropdown(
+            choices=[],
+            label="Select a Time Slot",
+            value=None,
+            interactive=True
         )
-        time_slot = gr.Textbox(
-            label="Appointment Time",
-            placeholder="Select time (HH:MM)",
-            elem_id="time-picker"
+        date = gr.Textbox(
+            label="Appointment Date (YYYY-MM-DD)",
+            placeholder="e.g., 2025-04-20"
         )
-        time_suggestions = gr.Markdown("Select a therapist to see available times.")
         user_email = gr.Textbox(
-            label="Your Email",
+            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, mood, conversation_mode, region, state],
@@ -479,7 +451,7 @@ with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_
     therapist.change(
         fn=get_available_times,
         inputs=therapist,
-        outputs=[time_slot, time_suggestions]
+        outputs=[time_slot, schedule_output]
     )
     schedule_btn.click(
         fn=schedule_appointment,
@@ -487,12 +459,15 @@ with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_
         outputs=[schedule_output, state]
     )
 
+    # Resource Links
     gr.Markdown("""
-    **Resources:**
-    - [Suicide Prevention Lifeline](https://suicidepreventionlifeline.org/)
+    ---
+    **Helpful Resources:**
+    - [National Suicide Prevention Lifeline](https://suicidepreventionlifeline.org/)
     - [MentalHealth.gov](https://www.mentalhealth.gov/)
     - [Crisis Text Line](https://www.crisistextline.org/)
     """)
 
+# Launch (for local testing)
 if __name__ == "__main__":
     demo.launch()