Update app.py

app.py

@@ -280,18 +280,18 @@ def update(
                 raise gr.Error(e)
 
 
-# Password protection logic
+# Custom login system
 def check_password(password):
     return password == PASSWORD
 
 
-# Gradio app with password protection
+# Gradio app with custom login system
 with gr.Blocks() as demo:
     # Login interface
     with gr.Row():
         password_input = gr.Textbox(label="Enter Password", type="password")
-        login_button = gr.LoginButton(label="Login")
-        logout_button = gr.LogoutButton(label="Logout")
+        login_button = gr.Button("Login")
+        login_status = gr.Textbox(label="Login Status", interactive=False)
 
     # Main app interface (hidden by default)
     with gr.Row(visible=False) as main_interface:
@@ -405,20 +405,16 @@ with gr.Blocks() as demo:
             )
 
     # Show/hide main interface based on login status
-    def toggle_interface(is_logged_in):
-        return gr.update(visible=is_logged_in)
+    def toggle_interface(password):
+        if check_password(password):
+            return gr.update(visible=True), "Login successful!"
+        else:
+            return gr.update(visible=False), "Invalid password. Please try again."
 
     login_button.click(
-        fn=check_password,
+        fn=toggle_interface,
         inputs=password_input,
-        outputs=None,
-        success=toggle_interface,
-        queue=False,
-    )
-    logout_button.click(
-        fn=lambda: False,
-        outputs=main_interface,
-        queue=False,
+        outputs=[main_interface, login_status],
     )
 
 # Launch the app with embedding enabled