Update chatbot.py

chatbot.py

@@ -12,7 +12,6 @@ DATABASE_URL = os.getenv('DATABASE_URL')
 
 # Parse the database URL
 result = urlparse(DATABASE_URL)
-
 username = result.username
 password = result.password
 database = result.path[1:]
@@ -142,13 +141,15 @@ with gr.Blocks() as demo:
     password = gr.Textbox(label="Password", type="password")
     login_button = gr.Button("Login")
     register_button = gr.Button("Register")
+    user_id_state = gr.State()
 
     chat_interface = gr.Interface(
         fn=chatbot,
-        inputs=[gr.Textbox(label="User ID"), gr.Textbox(lines=7, label="Chat with AttachmentBot")],
+        inputs=[user_id_state, gr.Textbox(lines=7, label="Chat with AttachmentBot")],
         outputs=gr.Textbox(label="Conversation"),
         title="AttachmentBot",
-        description="Let me survey you about your attachment with certain people in your life. To begin, enter 'start'."
+        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):
@@ -165,7 +166,7 @@ with gr.Blocks() as demo:
         else:
             return gr.update(visible=False), "", "Registration failed, try a different username."
 
-    login_button.click(login, inputs=[username, password], outputs=[chat_interface, gr.Variable(), gr.Textbox(visible=False)])
-    register_button.click(register, inputs=[username, password], outputs=[chat_interface, gr.Variable(), gr.Textbox(visible=False)])
+    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)])
 
 demo.launch()