Update app.py

app.py

@@ -53,8 +53,16 @@ def login(username, password):
 def logout():
     return "✅ Logged out successfully!", None
 
-def download_db():
-    if os.path.exists(DB_PATH):
+def update_home(user):
+    if user == "Admin":
+        return "👑 Welcome Admin! You can download the DB."
+    elif user:
+        return f"🏠 Welcome Home, {user}!"
+    else:
+        return "❌ Please login first."
+
+def download_db(user):
+    if user == "Admin" and os.path.exists(DB_PATH):
         return DB_PATH
     return None
 
@@ -62,6 +70,8 @@ def download_db():
 with gr.Blocks() as demo:
     gr.Markdown("## 🔐 Simple Login System with Signup & DB Download")
 
+    current_user = gr.State()
+
     with gr.Tab("Signup"):
         signup_user = gr.Textbox(label="Username")
         signup_pass = gr.Textbox(label="Password", type="password")
@@ -75,7 +85,6 @@ with gr.Blocks() as demo:
         login_pass = gr.Textbox(label="Password", type="password")
         login_btn = gr.Button("Login")
         login_status = gr.Textbox(label="Login Status", interactive=False)
-        current_user = gr.State()  # store logged-in user
 
         login_btn.click(login, inputs=[login_user, login_pass], outputs=[login_status, current_user])
 
@@ -84,14 +93,13 @@ with gr.Blocks() as demo:
         logout_btn = gr.Button("Logout")
         download_btn = gr.File(label="Download DB", interactive=False)
 
-        # logout clears user
-        logout_btn.click(logout, outputs=[home_status, current_user])
+        # Update home page whenever current_user changes
+        current_user.change(update_home, inputs=current_user, outputs=home_status)
 
-        # allow admin to download DB
-        login_btn.click(fn=lambda u: "Welcome Home!" if u else "❌ Please login first.",
-                        inputs=current_user, outputs=home_status)
+        # Logout clears user
+        logout_btn.click(logout, outputs=[home_status, current_user])
 
-        login_btn.click(fn=lambda u: download_db() if u == "Admin" else None,
-                        inputs=current_user, outputs=download_btn)
+        # Enable DB download only for Admin
+        current_user.change(download_db, inputs=current_user, outputs=download_btn)
 
 demo.launch()