Update chatbot.py

chatbot.py

@@ -142,6 +142,7 @@ with gr.Blocks() as demo:
     login_button = gr.Button("Login")
     register_button = gr.Button("Register")
     user_id_state = gr.State()
+    auth_message = gr.Textbox(visible=False)
 
     chat_input = gr.Textbox(lines=7, label="Chat with AttachmentBot")
     chat_output = gr.Textbox(label="Conversation")
@@ -161,8 +162,8 @@ 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=[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)])
+    login_button.click(login, inputs=[username, password], outputs=[chat_input, user_id_state, auth_message])
+    register_button.click(register, inputs=[username, password], outputs=[chat_input, user_id_state, auth_message])
 
     chat_interface = gr.Interface(
         fn=chatbot,
@@ -170,7 +171,6 @@ with gr.Blocks() as demo:
         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()