Update app.py

app.py

@@ -203,9 +203,12 @@ def get_available_times(therapist):
     return gr.update(choices=[], value=None), "Please select a therapist."
 
 # Send emails with retries
-def send_emails(therapist, time_slot, date, user_email, therapist_email):
+def send_emails(therapist, time_slot, date, user_email, therapist_email, state):
+    if "failed_emails" not in state:
+        state["failed_emails"] = []
+    
     msg_therapist = EmailMessage()
-    msg_therapist.set_content(f"""
+    therapist_body = f"""
     Dear {therapist},
     
     You have a new appointment:
@@ -217,13 +220,14 @@ def send_emails(therapist, time_slot, date, user_email, therapist_email):
     
     Best,
     Mental Health Chatbot
-    """)
+    """
+    msg_therapist.set_content(therapist_body)
     msg_therapist["Subject"] = "New Appointment"
     msg_therapist["From"] = GMAIL_ADDRESS
     msg_therapist["To"] = therapist_email
     
     msg_user = EmailMessage()
-    msg_user.set_content(f"""
+    user_body = f"""
     Dear User,
     
     Your appointment is booked:
@@ -235,7 +239,8 @@ def send_emails(therapist, time_slot, date, user_email, therapist_email):
     
     Best,
     Mental Health Chatbot
-    """)
+    """
+    msg_user.set_content(user_body)
     msg_user["Subject"] = "Appointment Confirmation"
     msg_user["From"] = GMAIL_ADDRESS
     msg_user["To"] = user_email
@@ -250,8 +255,15 @@ def send_emails(therapist, time_slot, date, user_email, therapist_email):
             return True, ""
         except (smtplib.SMTPException, ConnectionRefusedError, OSError) as e:
             if attempt == max_retries - 1:
+                # Store failed email content
+                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": str(e),
+                    "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                })
                 return False, str(e)
-            time.sleep(2)  # Wait before retry
+            time.sleep(2)
     return False, "Max retries exceeded"
 
 # Schedule appointment
@@ -284,17 +296,18 @@ def schedule_appointment(therapist, time_slot, date, user_email, state):
     
     # Send emails
     therapist_email = therapists[therapist]["email"]
-    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email)
+    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 {time_slot}. Emails sent!", state
+        return f"Appointment booked with {therapist} on {date} at {time_slot}. Emails sent to {therapist_email} and {user_email}!", 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
+                f"Please email {therapist_email} with your details (date: {date}, time: {time_slot}) and check {user_email} for confirmation. "
+                f"Emails may be in spam/junk."), state
 
 # Gradio Interface
 with gr.Blocks(title="Mental Health Support Chatbot") as demo:
     # Initialize state
-    state = gr.State({"chat_history": [], "mood_journal": [], "feedback": [], "appointments": []})
+    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.")