Update chatbot.py

chatbot.py

@@ -147,15 +147,6 @@ with gr.Blocks() as demo:
     chat_output = gr.Textbox(label="Conversation")
     state = gr.State()
 
-    chat_interface = gr.Interface(
-        fn=chatbot,
-        inputs=[user_id_state, chat_input, state],
-        outputs=[chat_output, state],
-        title="AttachmentBot",
-        description="Let me survey you about your attachment with certain people in your life. To begin, enter 'start'.",
-        visible=False  # Initially hidden, shown after login
-    )
-
     def login(username, password):
         user_id = authenticate(username, password)
         if user_id:
@@ -170,7 +161,16 @@ with gr.Blocks() as demo:
         else:
             return gr.update(visible=False), None, "Registration failed, try a different username."
 
-    login_button.click(login, inputs=[username, password], outputs=[chat_interface, user_id_state, gr.Textbox(visible=False)])
-    register_button.click(register, inputs=[username, password], outputs=[chat_interface, user_id_state, gr.Textbox(visible=False)])
+    login_button.click(login, inputs=[username, password], outputs=[gr.update(visible=True), user_id_state, gr.Textbox(visible=False)])
+    register_button.click(register, inputs=[username, password], outputs=[gr.update(visible=True), user_id_state, gr.Textbox(visible=False)])
+
+    chat_interface = gr.Interface(
+        fn=chatbot,
+        inputs=[user_id_state, chat_input, state],
+        outputs=[chat_output, state],
+        title="AttachmentBot",
+        description="Let me survey you about your attachment with certain people in your life. To begin, enter 'start'.",
+        visible=False  # Initially hidden, shown after login
+    )
 
-demo.launch()
+    demo.launch()