Update app.py

app.py

@@ -92,27 +92,36 @@ regional_resources = {
     ]
 }
 
-# Mock Therapist Database
+# Mock Therapist Database with 12-hour time format
 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", "15:30"]
+        "times": [
+            "7:00 AM", "7:30 AM", "8:00 AM", "8:30 AM", "9:00 AM", "9:30 AM",
+            "2:00 PM", "2:30 PM", "3:00 PM", "3:30 PM", "4:00 PM"
+        ]
     },
     "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", "16:30"]
+        "times": [
+            "8:00 AM", "8:30 AM", "9:00 AM", "9:30 AM", "10:00 AM",
+            "3:00 PM", "3:30 PM", "4:00 PM", "4:30 PM", "5:00 PM"
+        ]
     },
     "Dr. Sarah Brown": {
         "specialty": "Trauma and PTSD",
         "email": "[email protected]",
-        "times": ["08:00", "08:30", "09:00", "13:00", "13:30", "14:00", "14:30", "15:00", "15:30"]
+        "times": [
+            "7:00 AM", "7:30 AM", "8:00 AM", "1:00 PM", "1:30 PM",
+            "6:00 PM", "6:30 PM", "7:00 PM", "8:00 PM", "9:00 PM", "10:00 PM"
+        ]
     }
 }
 
 # Initialize Gemini model
-model = genai.GenerativeModel("gemini-1.5-flash", 
+model = genai.GenerativeModel("learnlm-1.5-pro-experimental", 
                              generation_config={"temperature": 0.7, "max_output_tokens": 200})
 
 # Chatbot function
@@ -144,7 +153,7 @@ def chatbot_function(message, mood, conversation_mode, region, state):
     }.get(conversation_mode, "Maintain a balanced, empathetic tone.")
     
     prompt = f"""
-    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:
+    You are Healora, a compassionate chatbot focused on mental health, self-motivation, and being a supportive friend. Based on the user's input: {message}, provide:
     - Empathetic, sensitive responses in the user's language (detected as {lang}).
     - {tone_instruction}
     - Suggestions for coping strategies if distress is detected, tailored to their mood or input.
@@ -176,7 +185,7 @@ def chatbot_function(message, mood, conversation_mode, region, state):
         if user_msg:
             chat_display += f"**You**: {user_msg}\n\n"
         if bot_msg:
-            chat_display += f"**Bot**: {bot_msg}\n\n"
+            chat_display += f"**Healora**: {bot_msg}\n\n"
     
     return chat_display, state
 
@@ -190,7 +199,6 @@ def log_mood(mood, state):
     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()})
-        # Trigger mood trends update
         trend_output, state = show_mood_trends(state)
         return "Mood logged!", trend_output, state
     return "Please select a mood.", None, state
@@ -247,7 +255,7 @@ You have a new appointment:
 Please confirm with the client.
 
 Best,
-Mental Health Chatbot
+Healora
     """
     therapist_message = create_message(therapist_email, "New Appointment", therapist_body)
     
@@ -262,7 +270,7 @@ Details:
 - Time: {time_slot}
 
 Best,
-Mental Health Chatbot
+Healora
     """
     user_message = create_message(user_email, "Appointment Confirmation", user_body)
     
@@ -318,15 +326,16 @@ def schedule_appointment(therapist, time_slot, date, user_email, 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
+        # Convert 12-hour format to 24-hour for validation
+        appointment_time = datetime.strptime(time_slot, "%I:%M %p").strftime("%H:%M")
+        if time_slot not in therapists[therapist]["times"]:
+            return f"Time {time_slot} is not available for {therapist}.", state
     except ValueError:
-        return "Invalid time format (use HH:MM).", state
+        return "Invalid time format.", state
     
     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")
@@ -334,30 +343,40 @@ def schedule_appointment(therapist, time_slot, date, user_email, state):
     state["appointments"].append(appointment)
     
     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!", state
     else:
-        return (f"Appointment booked with {therapist} on {date} at {appointment_time}. "
+        return (f"Appointment booked with {therapist} on {date} at {time_slot}. "
                 f"Email sending failed: {error_msg}. Please contact {therapist_email}."), state
 
+# Minimal CSS for UI polish
+custom_css = """
+button { padding: 10px 20px; margin: 5px; }
+input, select { border: 1px solid #ccc; border-radius: 4px; padding: 8px; }
+.gr-row { gap: 10px; }
+.gr-column { padding: 10px; }
+.gr-textbox, .gr-dropdown { margin-bottom: 10px; }
+"""
+
 # Gradio Interface
-with gr.Blocks(title="Mental Health Support Chatbot") as demo:
+with gr.Blocks(title="Healora 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, support, and help you feel stronger. Share how you're feeling today.")
+    gr.Markdown("## Healora")
+    gr.Markdown("Your safe space for healing and hope.")
+    gr.Markdown("---")
     
     with gr.Row():
         with gr.Column(scale=1):
             mood = gr.Dropdown(
                 choices=["Select mood", "Happy", "Sad", "Anxious", "Stressed", "Angry", "Overwhelmed", "Hopeful", "Lonely", "Other"],
-                label="Your Mood",
+                label="How Are You Feeling?",
                 value="Select mood"
             )
             conversation_mode = gr.Radio(
                 choices=["Neutral", "Calm", "Motivational", "Empathetic", "Hopeful"],
-                label="Conversation Style",
+                label="Choose a Tone",
                 value="Neutral"
             )
             region = gr.Dropdown(
@@ -367,57 +386,59 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
             )
         with gr.Column(scale=2):
             chatbot = gr.Textbox(
-                label="Our Conversation",
+                label="Chat with Healora",
                 value="",
                 interactive=False,
-                lines=12,
+                lines=14,
                 show_copy_button=True
             )
             user_input = gr.Textbox(
-                placeholder="Tell me how you're feeling...",
-                label="Your Message",
+                placeholder="Share your thoughts...",
+                label="Your Thoughts",
                 lines=3
             )
             with gr.Row():
-                submit_btn = gr.Button("Send Message")
+                submit_btn = gr.Button("Send", variant="primary")
                 clear_btn = gr.Button("Clear Chat")
     
+    gr.Markdown("---")
+    
     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_output = gr.Plot(label="Your Mood Trends")
+        mood_log_output = gr.Textbox(label="Status", interactive=False)
+        mood_trend_output = gr.Plot(label="Mood Trends")
     
-    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)
+    with gr.Accordion("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 Your Experience")
+        feedback_btn = gr.Button("Submit")
+        feedback_output = gr.Textbox(label="Status", interactive=False)
     
-    with gr.Accordion("Book an Appointment"):
+    with gr.Accordion("Appointments"):
         therapist = gr.Dropdown(
             choices=list(therapists.keys()),
-            label="Choose a Therapist"
+            label="Select Therapist"
         )
         date = gr.Textbox(
-            label="Appointment Date",
-            placeholder="Enter date (YYYY-MM-DD)",
-            info="Use format YYYY-MM-DD, e.g., 2025-04-22"
+            label="Date",
+            placeholder="YYYY-MM-DD (e.g., 2025-04-22)",
+            info="Enter date in YYYY-MM-DD format"
         )
         time_slot = gr.Dropdown(
             choices=["Select time"],
-            label="Appointment Time",
+            label="Time",
             interactive=True,
             allow_custom_value=True
         )
         user_email = gr.Textbox(
-            label="Your Email",
+            label="Email",
             placeholder="e.g., [email protected]"
         )
-        schedule_btn = gr.Button("Book Appointment")
-        schedule_output = gr.Textbox(label="Booking Status", interactive=False)
+        schedule_btn = gr.Button("Book Appointment", variant="primary")
+        schedule_output = gr.Textbox(label="Status", interactive=False)
     
     submit_btn.click(
         fn=chatbot_function,
@@ -460,6 +481,7 @@ with gr.Blocks(title="Mental Health Support Chatbot") as demo:
         outputs=[schedule_output, state]
     )
 
+    gr.Markdown("---")
     gr.Markdown("""
     **Resources**:
     - [Befrienders Worldwide](https://befrienders.org/)