Update app.py

app.py

@@ -1442,12 +1442,9 @@ def login(username, password):
 
 def toggle_login_signup(show_login):
     if show_login:
-        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+        return gr.update(visible=True), gr.update(visible=False)
     else:
-        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
-
-def close_forms():
-    return gr.update(visible=False), gr.update(visible=False)
+        return gr.update(visible=False), gr.update(visible=True)
 
 def toggle_login_logout(is_logged_in):
     if is_logged_in:
@@ -1460,7 +1457,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     
     def toggle_login_status(is_logged_in):
         is_logged_in = not is_logged_in
-        return gr.update(value=toggle_login_logout(is_logged_in)), is_logged_in, gr.update(visible=False), gr.update(visible=False)
+        return gr.update(value=toggle_login_logout(is_logged_in)), is_logged_in
     
     with gr.Row():
         with gr.Column():
@@ -1512,9 +1509,11 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
                 login_password = gr.Textbox(placeholder="Password", type="password")
                 login_form_button = gr.Button("Login")
                 login_message = gr.Textbox(interactive=False)
-                login_form_button.click(fn=login, inputs=[login_username, login_password], outputs=[login_message])
-                login_button.click(fn=toggle_login_signup, inputs=[gr.State(True)], outputs=[login_form, gr.Column(visible=False)])
-                gr.Button("Close").click(fn=close_forms, inputs=None, outputs=[login_form, signup_form])
+                login_form_button.click(fn=login, inputs=[login_username, login_password], outputs=[login_message, login_form, signup_button]).then(
+                    fn=lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(value="Logout")), inputs=None, outputs=[login_form, signup_form, login_button]
+                )
+                gr.Button("Close").click(fn=lambda: gr.update(visible=False), inputs=None, outputs=login_form)
+                login_button.click(fn=toggle_login_signup, inputs=[gr.State(True)], outputs=[login_form, signup_form])
             
             signup_form = gr.Column(visible=False)
             with signup_form:
@@ -1524,12 +1523,14 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
                 signup_password_confirmation = gr.Textbox(placeholder="Confirm Password", type="password")
                 signup_form_button = gr.Button("Signup")
                 signup_message = gr.Textbox(interactive=False)
-                signup_form_button.click(fn=signup, inputs=[signup_username, signup_password, signup_password_confirmation], outputs=[signup_message])
-                signup_button.click(fn=toggle_login_signup, inputs=[gr.State(False)], outputs=[signup_form, gr.Column(visible=False)])
-                gr.Button("Close").click(fn=close_forms, inputs=None, outputs=[login_form, signup_form])
+                signup_form_button.click(fn=signup, inputs=[signup_username, signup_password, signup_password_confirmation], outputs=[signup_message, signup_form, signup_button]).then(
+                    fn=lambda: gr.update(visible=False), inputs=None, outputs=signup_form
+                )
+                gr.Button("Close").click(fn=lambda: gr.update(visible=False), inputs=None, outputs=signup_form)
+                signup_button.click(fn=toggle_login_signup, inputs=[gr.State(False)], outputs=[signup_form, login_form])
     
-    login_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in, login_form, signup_form])
-    signup_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in, login_form, signup_form])
+    login_form_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in])
+    login_button.click(fn=toggle_login_status, inputs=[is_logged_in], outputs=[login_button, is_logged_in])
 
 demo.queue()
 demo.launch(share=True)