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")
 
@@ -27,7 +27,7 @@ def get_gmail_service():
         creds = Credentials.from_authorized_user_info(json.loads(GMAIL_TOKEN_JSON))
         return build("gmail", "v1", credentials=creds)
     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 = {
@@ -106,12 +106,15 @@ def chatbot_function(message, mood, conversation_mode, region, state):
     
     history = state["chat_history"]
     
+    if not message.strip():
+        return "Please enter a message.", state
+    
     try:
-        lang = detect(message) if message.strip() else "en"
+        lang = detect(message)
     except:
         lang = "en"
     
-    if mood and mood != "Select mood (optional)":
+    if mood and mood != "Select mood":
         history.append([f"I'm feeling {mood.lower()}.", None])
     
     history.append([message, None])
@@ -134,17 +137,17 @@ def chatbot_function(message, mood, conversation_mode, region, state):
     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 = f"I'm here for you, but something went wrong: {str(e)}. Please try again."
     
-    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}"
     
     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**:\n" + "\n".join(regional_resources[region_key])
     response_text += resources
     
     history[-1][1] = response_text
@@ -166,25 +169,25 @@ def clear_chat(state):
 
 # Mood journal function
 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
+        return "Mood logged!", state
+    return "Please select a mood.", 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 Over Time", markers=True)
+    fig = px.line(df, x="timestamp", y="mood", title="Mood Trends", 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 "Thank you for your feedback!", state
+        return "Feedback submitted!", state
     return "Please provide feedback or a rating.", state
 
 # Emergency resources
@@ -199,8 +202,8 @@ def show_emergency_resources():
 # Get available times for selected therapist
 def get_available_times(therapist):
     if therapist and therapist in therapists:
-        return f"Available times for {therapist} loaded. Please select a time."
-    return "Please select a therapist."
+        return therapists[therapist]["times"], f"Available times for {therapist} loaded."
+    return [], "Please select a therapist."
 
 # Create MIME message for Gmail API
 def create_message(to, subject, message_text):
@@ -246,10 +249,19 @@ 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()
-        therapist_result = service.users().messages().send(userId="me", body=therapist_message).execute()
-        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)}"
@@ -275,25 +287,25 @@ 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 valid time slot.", state
+        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
+        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
+        return "Invalid time format (use HH:MM).", state
     
     appointment = {
         "therapist": therapist,
@@ -307,127 +319,115 @@ def schedule_appointment(therapist, time_slot, date, user_email, state):
     therapist_email = therapists[therapist]["email"]
     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 {appointment_time}. 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 {appointment_time}. "
-                f"Emails from {GMAIL_ADDRESS} could not be sent. "
-                f"Please email {therapist_email} with your details (date: {date}, time: {appointment_time}) "
-                f"and check {user_email} for confirmation (spam/junk)."), state
+                f"Email sending failed: {error_msg}. Please contact {therapist_email}."), state
 
-# Custom CSS for beautiful UI
+# Custom CSS for clean UI
 custom_css = """
-/* General Styling */
 body {
-    font-family: 'Inter', sans-serif;
-    background-color: #f0f4f8;
+    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Arial, sans-serif;
+    background: #f5f7fa;
+    color: #1a202c;
 }
 
-/* Containers */
 .gradio-container {
-    max-width: 900px;
-    margin: 0 auto;
-    padding: 20px;
-    background: white;
-    border-radius: 16px;
-    box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
+    max-width: 800px;
+    margin: 20px auto;
+    padding: 24px;
+    background: #ffffff;
+    border-radius: 12px;
+    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
 }
 
-/* Headings */
 h1 {
-    color: #2c3e50;
-    font-size: 2.5em;
+    font-size: 1.8em;
+    color: #2d3748;
     text-align: center;
-    margin-bottom: 10px;
+    margin-bottom: 16px;
 }
 
 h2, h3 {
-    color: #34495e;
+    color: #4a5568;
 }
 
-/* Inputs and Buttons */
 input, select, textarea {
-    border-radius: 8px !important;
-    border: 1px solid #d1d9e0 !important;
-    padding: 10px !important;
-    transition: border-color 0.2s;
+    border: 1px solid #e2e8f0 !important;
+    border-radius: 6px !important;
+    padding: 8px !important;
+    font-size: 1em !important;
+    width: 100% !important;
+    box-sizing: border-box;
 }
 
 input:focus, select:focus, textarea:focus {
-    border-color: #3498db !important;
+    border-color: #3182ce !important;
     outline: none;
+    box-shadow: 0 0 0 2px rgba(49, 130, 206, 0.2);
 }
 
 button {
-    background: linear-gradient(90deg, #3498db, #2980b9) !important;
-    color: white !important;
+    background: #3182ce !important;
+    color: #ffffff !important;
     border: none !important;
-    border-radius: 8px !important;
-    padding: 12px 24px !important;
-    font-weight: 600 !important;
-    transition: transform 0.2s, box-shadow 0.2s !important;
+    border-radius: 6px !important;
+    padding: 10px 20px !important;
+    font-size: 1em !important;
+    cursor: pointer;
+    transition: background 0.2s;
 }
 
 button:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2) !important;
+    background: #2b6cb0 !important;
 }
 
-/* Chatbox */
 textarea[readonly] {
-    background: #f8fafc !important;
-    border-radius: 12px !important;
-    padding: 15px !important;
+    background: #edf2f7 !important;
+    border-radius: 6px !important;
+    padding: 12px !important;
+    font-size: 1em;
 }
 
-/* Appointment Section */
 #appointment-section .gr-box {
-    background: #e6f0fa !important;
-    border-radius: 12px !important;
-    padding: 20px !important;
+    background: #f7fafc !important;
+    border-radius: 8px !important;
+    padding: 16px !important;
 }
 
-/* Date and Time Pickers */
-#date-picker input, #time-picker input {
+#date-picker, #time-picker {
     width: 100%;
-    padding: 10px;
-    border: 1px solid #d1d9e0;
-    border-radius: 8px;
-    font-size: 1em;
-    cursor: pointer;
-}
-
-#date-picker input::-webkit-calendar-picker-indicator,
-#time-picker input::-webkit-calendar-picker-indicator {
-    cursor: pointer;
 }
 
-/* Responsive Design */
 @media (max-width: 600px) {
     .gradio-container {
-        padding: 15px;
+        padding: 16px;
+        margin: 10px;
     }
     button {
-        padding: 10px 20px !important;
+        padding: 8px 16px !important;
     }
 }
 """
 
-# JavaScript to convert textboxes to date/time pickers
+# JavaScript for date/time pickers
 custom_js = """
 document.addEventListener('DOMContentLoaded', function() {
-    // Convert date textbox to date picker
-    const dateInput = document.querySelector('#date-picker input');
-    if (dateInput) {
-        dateInput.type = 'date';
-        dateInput.placeholder = 'Select a date';
-    }
-    
-    // Convert time textbox to time picker
-    const timeInput = document.querySelector('#time-picker input');
-    if (timeInput) {
-        timeInput.type = 'time';
-        timeInput.step = '1800'; // 30-minute intervals
-        timeInput.placeholder = 'Select a time';
+    try {
+        const dateInput = document.querySelector('#date-picker input');
+        if (dateInput) {
+            dateInput.type = 'date';
+            dateInput.placeholder = 'YYYY-MM-DD';
+        }
+        
+        const timeInput = document.querySelector('#time-picker input');
+        if (timeInput) {
+            timeInput.type = 'time';
+            timeInput.step = '1800';
+            timeInput.placeholder = 'HH:MM';
+        }
+    } catch (e) {
+        console.error('Picker initialization failed:', e);
     }
 });
 """
@@ -436,84 +436,85 @@ document.addEventListener('DOMContentLoaded', function() {
 with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_js) as demo:
     state = gr.State({"chat_history": [], "mood_journal": [], "feedback": [], "appointments": [], "failed_emails": []})
     
-    gr.Markdown("# Mental Health Support Chatbot")
-    gr.Markdown("A safe space to share how you're feeling, with tools to support your well-being.")
+    gr.Markdown("## Mental Health Support Chatbot")
+    gr.Markdown("A safe space to share your feelings and access support.")
     
     with gr.Row():
-        with gr.Column(scale=3):
+        with gr.Column(scale=1):
             mood = gr.Dropdown(
-                choices=["Select mood (optional)", "Happy", "Sad", "Anxious", "Stressed", "Other"],
-                label="How are you feeling today?",
-                value="Select mood (optional)"
+                choices=["Select mood", "Happy", "Sad", "Anxious", "Stressed", "Other"],
+                label="Mood",
+                value="Select mood"
             )
             conversation_mode = gr.Radio(
                 choices=["Neutral", "Calm", "Motivational"],
-                label="Conversation Style",
+                label="Style",
                 value="Neutral"
             )
             region = gr.Dropdown(
                 choices=["USA", "India", "UK", "Global"],
-                label="Your Region",
+                label="Region",
                 value="Global"
             )
-        with gr.Column(scale=7):
+        with gr.Column(scale=2):
             chatbot = gr.Textbox(
-                label="Conversation",
+                label="Chat",
                 value="",
                 interactive=False,
-                lines=12,
+                lines=10,
                 show_copy_button=True
             )
             user_input = gr.Textbox(
-                placeholder="Type your message here...",
-                label="Your Message",
+                placeholder="Type your message...",
+                label="Message",
                 lines=2
             )
             with gr.Row():
                 submit_btn = gr.Button("Send")
-                clear_btn = gr.Button("Clear Chat")
+                clear_btn = gr.Button("Clear")
     
-    emergency_btn = gr.Button("Emergency Resources")
+    emergency_btn = gr.Button("Crisis Support")
     emergency_output = gr.Markdown("")
     
     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_log_output = gr.Textbox(label="Status", interactive=False)
+        mood_trend_btn = gr.Button("View Trends")
         mood_trend_output = gr.Plot()
     
-    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="Rate this interaction")
-        feedback_btn = gr.Button("Submit Feedback")
-        feedback_output = gr.Textbox(label="Feedback Status", interactive=False)
+    with gr.Accordion("Feedback"):
+        feedback_text = gr.Textbox(label="Feedback", placeholder="Your thoughts...")
+        feedback_rating = gr.Slider(minimum=1, maximum=5, step=1, label="Rating")
+        feedback_btn = gr.Button("Submit")
+        feedback_output = gr.Textbox(label="Status", interactive=False)
     
     with gr.Accordion("Schedule Appointment", elem_id="appointment-section"):
         therapist = gr.Dropdown(
             choices=list(therapists.keys()),
-            label="Select a Therapist"
+            label="Therapist"
         )
         date = gr.Textbox(
-            label="Appointment Date",
-            placeholder="Select a date (YYYY-MM-DD)",
+            label="Date",
+            placeholder="YYYY-MM-DD",
             elem_id="date-picker"
         )
-        time_slot = gr.Textbox(
-            label="Appointment Time",
-            placeholder="Select a time (HH:MM)",
-            elem_id="time-picker"
+        time_slot = gr.Dropdown(
+            choices=[],
+            label="Time",
+            interactive=True
         )
         user_email = gr.Textbox(
-            label="Your Email Address",
-            placeholder="e.g., [email protected]"
+            label="Email",
+            placeholder="[email protected]"
         )
-        schedule_btn = gr.Button("Book Appointment")
-        schedule_output = gr.Textbox(label="Booking Status", interactive=False)
+        schedule_btn = gr.Button("Book")
+        schedule_output = gr.Textbox(label="Status", interactive=False)
     
     submit_btn.click(
         fn=chatbot_function,
         inputs=[user_input, mood, conversation_mode, region, state],
-        outputs=[chatbot, state]
+        outputs=[chatbot, state],
+        _js="() => { return true; }"
     )
     user_input.submit(
         fn=chatbot_function,
@@ -548,7 +549,7 @@ with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_
     therapist.change(
         fn=get_available_times,
         inputs=therapist,
-        outputs=schedule_output
+        outputs=[time_slot, schedule_output]
     )
     schedule_btn.click(
         fn=schedule_appointment,
@@ -557,9 +558,8 @@ with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css, js=custom_
     )
 
     gr.Markdown("""
-    ---
-    **Helpful Resources:**
-    - [National Suicide Prevention Lifeline](https://suicidepreventionlifeline.org/)
+    **Resources:**
+    - [Suicide Prevention Lifeline](https://suicidepreventionlifeline.org/)
     - [MentalHealth.gov](https://www.mentalhealth.gov/)
     - [Crisis Text Line](https://www.crisistextline.org/)
     """)