Update app.py

app.py

@@ -76,22 +76,22 @@ regional_resources = {
     ]
 }
 
-# Mock Therapist Database
+# Mock Therapist Database (times adjusted for time picker compatibility)
 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", "16:00"]
     },
     "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", "17:00"]
     },
     "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", "15:30", "16:00", "18:00"]
     }
 }
 
@@ -101,33 +101,27 @@ 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
     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
     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}).
@@ -137,30 +131,25 @@ 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:
         response_text = "I'm here for you. Could you share a bit more so I can support you better?"
     
-    # 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
     state["chat_history"] = history
     
-    # Format history for display
     chat_display = ""
     for user_msg, bot_msg in history:
         if user_msg:
@@ -210,8 +199,8 @@ def show_emergency_resources():
 # 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 f"Available times for {therapist} loaded. Please select a time."
+    return "Please select a therapist."
 
 # Create MIME message for Gmail API
 def create_message(to, subject, message_text):
@@ -227,7 +216,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,7 +231,6 @@ Mental Health Chatbot
     """
     therapist_message = create_message(therapist_email, "New Appointment", therapist_body)
     
-    # User email
     user_body = f"""
 Dear User,
 
@@ -261,12 +248,8 @@ Mental Health Chatbot
     
     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()
-        
         return True, ""
     except HttpError as e:
         error_msg = f"Gmail API error: {str(e)}"
@@ -291,7 +274,7 @@ 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"]:
+    if not time_slot:
         return "Please select a valid time slot.", state
     if not date:
         return "Please select a date.", state
@@ -303,107 +286,193 @@ def schedule_appointment(therapist, time_slot, date, user_email, state):
         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.", 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.", 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 from {GMAIL_ADDRESS} to {therapist_email} and {user_email}!", state
     else:
-        return (f"Appointment booked with {therapist} on {date} at {time_slot}. "
+        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: {time_slot}) "
+                f"Please email {therapist_email} with your details (date: {date}, time: {appointment_time}) "
                 f"and check {user_email} for confirmation (spam/junk)."), state
 
+# Custom CSS for beautiful UI
+custom_css = """
+/* General Styling */
+body {
+    font-family: 'Inter', sans-serif;
+    background-color: #f0f4f8;
+}
+
+/* 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);
+}
+
+/* Headings */
+h1 {
+    color: #2c3e50;
+    font-size: 2.5em;
+    text-align: center;
+    margin-bottom: 10px;
+}
+
+h2, h3 {
+    color: #34495e;
+}
+
+/* Inputs and Buttons */
+input, select, textarea {
+    border-radius: 8px !important;
+    border: 1px solid #d1d9e0 !important;
+    padding: 10px !important;
+    transition: border-color 0.2s;
+}
+
+input:focus, select:focus, textarea:focus {
+    border-color: #3498db !important;
+    outline: none;
+}
+
+button {
+    background: linear-gradient(90deg, #3498db, #2980b9) !important;
+    color: white !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;
+}
+
+button:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.2) !important;
+}
+
+/* Chatbox */
+textarea[readonly] {
+    background: #f8fafc !important;
+    border-radius: 12px !important;
+    padding: 15px !important;
+}
+
+/* Appointment Section */
+#appointment-section .gr-box {
+    background: #e6f0fa !important;
+    border-radius: 12px !important;
+    padding: 20px !important;
+}
+
+/* Date and Time Pickers */
+input[type="date"], input[type="time"] {
+    background: white !important;
+    cursor: pointer;
+}
+
+/* Responsive Design */
+@media (max-width: 600px) {
+    .gradio-container {
+        padding: 15px;
+    }
+    button {
+        padding: 10px 20px !important;
+    }
+}
+"""
+
 # Gradio Interface
-with gr.Blocks(title="Mental Health Support Chatbot") as demo:
-    # Initialize state
+with gr.Blocks(title="Mental Health Support Chatbot", css=custom_css) as demo:
     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.")
-    
-    # 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"
-    )
+    gr.Markdown("A safe space to share how you're feeling, with tools to support your well-being.")
     
-    # 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=3):
+            mood = gr.Dropdown(
+                choices=["Select mood (optional)", "Happy", "Sad", "Anxious", "Stressed", "Other"],
+                label="How are you feeling today?",
+                value="Select mood (optional)"
+            )
+            conversation_mode = gr.Radio(
+                choices=["Neutral", "Calm", "Motivational"],
+                label="Conversation Style",
+                value="Neutral"
+            )
+            region = gr.Dropdown(
+                choices=["USA", "India", "UK", "Global"],
+                label="Your Region",
+                value="Global"
+            )
+        with gr.Column(scale=7):
+            chatbot = gr.Textbox(
+                label="Conversation",
+                value="",
+                interactive=False,
+                lines=12,
+                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")
+                clear_btn = gr.Button("Clear Chat")
     
-    # 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("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_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)
     
-    # Appointment Scheduling
-    with gr.Accordion("Schedule Appointment"):
+    with gr.Accordion("Schedule Appointment", elem_id="appointment-section"):
         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
-        )
         date = gr.Textbox(
-            label="Appointment Date (YYYY-MM-DD)",
-            placeholder="e.g., 2025-04-20"
+            label="Appointment Date",
+            type="date",
+            placeholder="Select a date"
+        )
+        time_slot = gr.Textbox(
+            label="Appointment Time",
+            type="time",
+            placeholder="Select a time"
         )
         user_email = gr.Textbox(
             label="Your Email Address",
@@ -412,7 +481,6 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         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],
@@ -451,7 +519,7 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     therapist.change(
         fn=get_available_times,
         inputs=therapist,
-        outputs=[time_slot, schedule_output]
+        outputs=schedule_output
     )
     schedule_btn.click(
         fn=schedule_appointment,
@@ -459,7 +527,6 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         outputs=[schedule_output, state]
     )
 
-    # Resource Links
     gr.Markdown("""
     ---
     **Helpful Resources:**
@@ -468,6 +535,5 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     - [Crisis Text Line](https://www.crisistextline.org/)
     """)
 
-# Launch (for local testing)
 if __name__ == "__main__":
     demo.launch()