Update app.py

app.py

@@ -103,6 +103,10 @@ model = genai.GenerativeModel("gemini-1.5-flash",
 def chatbot_function(message, mood, conversation_mode, region, state):
     if "chat_history" not in state:
         state["chat_history"] = []
+    if "conversation_archive" not in state:
+        state["conversation_archive"] = []
+    if "current_conversation_id" not in state:
+        state["current_conversation_id"] = 0
     
     history = state["chat_history"]
     
@@ -150,20 +154,64 @@ def chatbot_function(message, mood, conversation_mode, region, state):
     history[-1][1] = response_text
     state["chat_history"] = history
     
-    chat_display = ""
+    chat_display = generate_chat_display(history)
+    
+    return chat_display, state
+
+# Generate chat display with left/right alignment
+def generate_chat_display(history):
+    chat_display = """
+    <style>
+        .chat-container { max-width: 600px; margin: auto; }
+        .message { margin: 10px 0; padding: 10px; border-radius: 10px; width: 80%; }
+        .bot-message { background-color: #e6f3ff; float: left; clear: both; }
+        .user-message { background-color: #d4edda; float: right; clear: both; }
+    </style>
+    <div class='chat-container'>
+    """
     for user_msg, bot_msg in history:
         if user_msg:
-            chat_display += f"**You**: {user_msg}\n\n"
+            chat_display += f"<div class='message user-message'><strong>You</strong>: {user_msg}</div>"
         if bot_msg:
-            chat_display += f"**Healora**: {bot_msg}\n\n"
-    
-    return chat_display, state
+            chat_display += f"<div class='message bot-message'><strong>Healora</strong>: {bot_msg}</div>"
+    chat_display += "</div>"
+    return chat_display
 
 # Clear chat history
 def clear_chat(state):
     state["chat_history"] = []
     return "", state
 
+# Start new conversation
+def new_conversation(state):
+    if state["chat_history"]:
+        state["conversation_archive"].append({
+            "id": state["current_conversation_id"],
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "history": state["chat_history"]
+        })
+        state["current_conversation_id"] += 1
+        state["chat_history"] = []
+    return "", update_conversation_dropdown(state), state
+
+# Update conversation dropdown
+def update_conversation_dropdown(state):
+    choices = ["Current Conversation"] + [
+        f"Conversation {conv['id']} ({conv['timestamp']})" 
+        for conv in state["conversation_archive"]
+    ]
+    return gr.update(choices=choices, value="Current Conversation")
+
+# Load selected conversation
+def load_conversation(selected_conversation, state):
+    if selected_conversation == "Current Conversation":
+        return generate_chat_display(state["chat_history"]), state
+    else:
+        for conv in state["conversation_archive"]:
+            if f"Conversation {conv['id']} ({conv['timestamp']})" == selected_conversation:
+                return generate_chat_display(conv["history"]), state
+        return generate_chat_display(state["chat_history"]), state
+
 # Mood journal function
 def log_mood(mood, state):
     if mood and mood != "Select mood (optional)":
@@ -207,10 +255,18 @@ def create_message(to, subject, message_text):
     return {"raw": raw}
 
 # Send emails using Gmail API
-def send_emails(therapist, time_slot, date, user_email, therapist_email, appointment_note, state):
+def send_emails(therapist, time_slot, date, user_email, therapist_email, appointment_note, chat_history, state):
     if "failed_emails" not in state:
         state["failed_emails"] = []
     
+    # Format chat history for email
+    chat_history_text = "Chat History:\n"
+    for user_msg, bot_msg in chat_history:
+        if user_msg:
+            chat_history_text += f"You: {user_msg}\n"
+        if bot_msg:
+            chat_history_text += f"Healora: {bot_msg}\n"
+    
     therapist_body = f"""
 Dear {therapist},
 
@@ -222,6 +278,7 @@ You have a new appointment:
 - Therapist Details:
   - Specialty: {therapists[therapist]["specialty"]}
   - Email: {therapist_email}
+- {chat_history_text}
 
 Please confirm with the client.
 
@@ -301,7 +358,7 @@ def schedule_appointment(therapist, time_slot, date, user_email, appointment_not
     state["appointments"].append(appointment)
     
     therapist_email = therapists[therapist]["email"]
-    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email, appointment_note, state)
+    success, error_msg = send_emails(therapist, time_slot, date, user_email, therapist_email, appointment_note, state["chat_history"], 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
     else:
@@ -390,7 +447,7 @@ Healora
 
 # Gradio Interface
 with gr.Blocks(title="Healora: Mental Health Support Chatbot") as demo:
-    state = gr.State({"chat_history": [], "mood_journal": [], "appointments": [], "failed_emails": []})
+    state = gr.State({"chat_history": [], "conversation_archive": [], "current_conversation_id": 0, "mood_journal": [], "appointments": [], "failed_emails": []})
     
     gr.Markdown("# Healora: Your Safe Space for Healing and Hope")
     gr.Markdown("I'm here to listen and support you. Feel free to share how you're feeling.")
@@ -412,11 +469,17 @@ with gr.Blocks(title="Healora: Mental Health Support Chatbot") as demo:
         value="Global"
     )
     
-    chatbot = gr.Textbox(
+    with gr.Row():
+        conversation_dropdown = gr.Dropdown(
+            choices=["Current Conversation"],
+            label="Select Conversation",
+            value="Current Conversation"
+        )
+        new_conversation_btn = gr.Button("New Conversation")
+    
+    chatbot = gr.HTML(
         label="Conversation",
-        value="",
-        interactive=False,
-        lines=10
+        value=""
     )
     user_input = gr.Textbox(
         placeholder="Type your message here...",
@@ -507,6 +570,16 @@ with gr.Blocks(title="Healora: Mental Health Support Chatbot") as demo:
         inputs=[state],
         outputs=[chatbot, state]
     )
+    new_conversation_btn.click(
+        fn=new_conversation,
+        inputs=[state],
+        outputs=[chatbot, conversation_dropdown, state]
+    )
+    conversation_dropdown.change(
+        fn=load_conversation,
+        inputs=[conversation_dropdown, state],
+        outputs=[chatbot, state]
+    )
     emergency_btn.click(
         fn=show_emergency_resources,
         inputs=region,