Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import google.generativeai as genai
 import os
-from datetime import datetime, timedelta
+from datetime import datetime
 import plotly.express as px
 import pandas as pd
 from langdetect import detect
@@ -9,6 +9,7 @@ import random
 import smtplib
 from email.message import EmailMessage
 import re
+import time
 
 # Configure Gemini API (expects GEMINI_API_KEY in Hugging Face secrets)
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
@@ -201,82 +202,94 @@ def get_available_times(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."
 
-# 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 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 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"]
+# Send emails with retries
+def send_emails(therapist, time_slot, date, user_email, therapist_email):
     msg_therapist = EmailMessage()
     msg_therapist.set_content(f"""
     Dear {therapist},
     
-    You have a new appointment scheduled:
+    You have a new appointment:
     - Date: {date}
     - Time: {time_slot}
     - Client Email: {user_email}
     
     Please confirm with the client.
     
-    Best regards,
+    Best,
     Mental Health Chatbot
     """)
-    msg_therapist["Subject"] = "New Appointment Scheduled"
+    msg_therapist["Subject"] = "New Appointment"
     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:
+    Your appointment is booked:
     - Therapist: {therapist}
     - Date: {date}
     - Time: {time_slot}
     
-    You will receive a confirmation from {therapist} soon.
+    Expect a confirmation from {therapist}.
     
-    Best regards,
+    Best,
     Mental Health Chatbot
     """)
     msg_user["Subject"] = "Appointment Confirmation"
     msg_user["From"] = GMAIL_ADDRESS
     msg_user["To"] = user_email
     
-    # Send emails
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            with smtplib.SMTP_SSL("smtp.gmail.com", 465, timeout=10) as server:
+                server.login(GMAIL_ADDRESS, GMAIL_APP_PASSWORD)
+                server.send_message(msg_therapist)
+                server.send_message(msg_user)
+            return True, ""
+        except (smtplib.SMTPException, ConnectionRefusedError, OSError) as e:
+            if attempt == max_retries - 1:
+                return False, str(e)
+            time.sleep(2)  # Wait before retry
+    return False, "Max retries exceeded"
+
+# 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 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 address.", state
+    
     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
+        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 emails
+    therapist_email = therapists[therapist]["email"]
+    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email)
+    if success:
+        return f"Appointment booked with {therapist} on {date} at {time_slot}. Emails sent!", state
+    else:
+        return (f"Appointment booked with {therapist} on {date} at {time_slot}, but emails failed: {error_msg}. "
+                f"Please contact {therapist_email} manually with your appointment details."), state
 
 # Gradio Interface
 with gr.Blocks(title="Mental Health Support Chatbot") as demo: