Update app.py

app.py

@@ -1,13 +1,12 @@
-import sqlite3
 import gradio as gr
+import sqlite3
 import os
 
-DB_PATH = "/tmp/temp.db"
-current_user = None  # global to store logged-in user
+DB_PATH = "temp.db"
 
-# Initialize DB with users table
+# --- Database Setup ---
 def init_db():
-    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("""
         CREATE TABLE IF NOT EXISTS users (
@@ -19,121 +18,80 @@ def init_db():
     conn.commit()
     conn.close()
 
-# Signup function
+init_db()
+
+# --- Functions ---
 def signup(username, password):
-    init_db()
-    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    if not username or not password:
+        return "⚠️ Please enter both username and password."
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     try:
         cursor.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, password))
         conn.commit()
         conn.close()
-        return "✅ Signup successful! You can now login."
+        return "✅ Signup successful! Please login."
     except sqlite3.IntegrityError:
         conn.close()
-        return "⚠️ Username already exists. Try another."
+        return "❌ Username already exists. Try another."
 
-# Login function
 def login(username, password):
-    global current_user
-    init_db()
-    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
-    cursor = conn.cursor()
-
-    # Special case: admin 1234/1234
     if username == "1234" and password == "1234":
-        current_user = "Admin"
-        conn.close()
-        return "admin"
-
+        return "✅ Admin logged in! You can download the database.", "Admin"
+    
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
     cursor.execute("SELECT * FROM users WHERE username=? AND password=?", (username, password))
     user = cursor.fetchone()
     conn.close()
+    
     if user:
-        current_user = username
-        return "home"
-    return "fail"
+        return f"✅ Welcome, {username}!", username
+    else:
+        return "❌ Invalid credentials.", None
 
-# Logout function
 def logout():
-    global current_user
-    current_user = None
-    return "🚪 You have been logged out.", home_page(), gr.update(visible=False)
-
-# Home page view
-def home_page():
-    if current_user:
-        return f"🏠 Welcome, **{current_user}**! You are logged in."
-    else:
-        return "🏠 Welcome to the Home Page! Please signup or login to continue."
+    return "✅ Logged out successfully!", None
 
-# Download DB file
-def get_db_file():
+def download_db():
     if os.path.exists(DB_PATH):
         return DB_PATH
     return None
 
-# Build Gradio app
-with gr.Blocks(css="""
-    body {background-color: #f5f7fa; font-family: 'Segoe UI', sans-serif;}
-    .gr-button {border-radius: 10px !important; font-weight: bold;}
-    .gr-textbox, .gr-number {border-radius: 8px !important;}
-    h1, h2 {color: #2c3e50;}
-""") as demo:
-    
-    gr.Markdown("<h1 style='text-align:center;'>🔐 Simple Login System</h1>")
-
-    with gr.Tabs():
-        with gr.Tab("🏠 Home"):
-            home_out = gr.Textbox(label="Message", value=home_page(), interactive=False)
-            logout_btn = gr.Button("🚪 Logout")
-            logout_status = gr.Textbox(label="Logout Status")
-
-            logout_btn.click(logout, outputs=[logout_status, home_out, None])
-
-        with gr.Tab("📝 Signup"):
-            su_user = gr.Textbox(label="New Username")
-            su_pass = gr.Textbox(label="New Password", type="password")
-            su_btn = gr.Button("Signup")
-            su_out = gr.Textbox(label="Status")
-            su_btn.click(signup, inputs=[su_user, su_pass], outputs=su_out)
-
-        with gr.Tab("🔑 Login"):
-            lg_user = gr.Textbox(label="Username")
-            lg_pass = gr.Textbox(label="Password", type="password")
-            lg_btn = gr.Button("Login")
-            lg_status = gr.Textbox(label="Status")
-
-            file_dl = gr.File(label="⬇️ Download temp.db (Admin only)", type="file", visible=False)
-
-            def handle_login(username, password):
-                result = login(username, password)
-                if result == "admin":
-                    return (
-                        "✅ Logged in as Admin. You can download the DB.", 
-                        get_db_file(), 
-                        gr.update(visible=True), 
-                        home_page()
-                    )
-                elif result == "home":
-                    return (
-                        "✅ Login successful!", 
-                        None, 
-                        gr.update(visible=False), 
-                        home_page()
-                    )
-                else:
-                    return (
-                        "❌ Invalid credentials. Try again.", 
-                        None, 
-                        gr.update(visible=False), 
-                        home_page()
-                    )
-
-            lg_btn.click(
-                handle_login, 
-                inputs=[lg_user, lg_pass], 
-                outputs=[lg_status, file_dl, file_dl, home_out]
-            )
+# --- Gradio UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔐 Simple Login System with Signup & DB Download")
+
+    with gr.Tab("Signup"):
+        signup_user = gr.Textbox(label="Username")
+        signup_pass = gr.Textbox(label="Password", type="password")
+        signup_btn = gr.Button("Signup")
+        signup_out = gr.Textbox(label="Signup Status", interactive=False)
+
+        signup_btn.click(signup, inputs=[signup_user, signup_pass], outputs=signup_out)
+
+    with gr.Tab("Login"):
+        login_user = gr.Textbox(label="Username")
+        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])
+
+    with gr.Tab("Home"):
+        home_status = gr.Textbox(label="Home Status", interactive=False)
+        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])
+
+        # 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)
+
+        login_btn.click(fn=lambda u: download_db() if u == "Admin" else None,
+                        inputs=current_user, outputs=download_btn)
 
 demo.launch()