Update app.py

app.py

@@ -115,30 +115,15 @@ agent_with_chat_history = RunnableWithMessageHistory(
 
 username = ""
 
-def login(message, history):
-    if "Lina" in message.lower():
-        username = "Lina"
-        return username
-    elif "Guest" in message.lower():
-        username = "Guest"
-        return username
-    else:
-        return "Please ask about username.", None
+def set_username(name):
+    global username
+    username = name
+    return gr.update(visible=False), gr.update(visible=True)
 
 # Gradio chat response function
 
 def respond(message, history: list[dict]):
 
-    with gr.Blocks():
-        with gr.Row():
-            with gr.Column():
-                    gr.ChatInterface(
-                        login,
-                        examples=["Guest", "Lina"],
-                        additional_outputs=[code],
-                        type="messages"
-                    )
-
     # Step 1: Embed user message
     try:
         vectorized_input = embedding_model.embed_query(message)
@@ -219,12 +204,38 @@ def respond(message, history: list[dict]):
 
 
 # launch ChatInterface
+
+with gr.Blocks() as demo:
+    # Login interface
+    login_box = gr.Column(visible=True)
+    with login_box:
+        gr.Markdown("### Choose a username to begin:")
+        with gr.Row():
+            btn_lina = gr.Button("Lina")
+            btn_guest = gr.Button("Guest")
+    
+    # Chatbot interface
+    chatbot_box = gr.Column(visible=False)
+    with chatbot_box:
+        chatbot = gr.ChatInterface(
+            respond,
+            title="YAQIN Chatbot",
+            description="Culturally Sensitive Chatbot for Muslim Women Wanting Mental Healthcare. \n\n What\'s on your mind?",
+            type="messages",
+        )
+
+    # Button actions
+    btn_lina.click(fn=set_username, inputs=[], outputs=[login_box, chatbot_box], kwargs={"name": "Lina"})
+    btn_guest.click(fn=set_username, inputs=[], outputs=[login_box, chatbot_box], kwargs={"name": "Guest"})
+
+
+'''
 demo = gr.ChatInterface(
     respond,
     title="YAQIN Chatbot",
     description="Culturally Sensitive Chatbot for Muslim Women Wanting Mental Healthcare. \n\n What\'s on your mind?",
     type="messages",
 )
-
+'''
 if __name__ == "__main__":
     demo.launch()