Create app.py

app.py

@@ -0,0 +1,139 @@
+import sqlite3
+import gradio as gr
+import os
+
+DB_PATH = "/tmp/temp.db"
+current_user = None  # global to store logged-in user
+
+# Initialize DB with users table
+def init_db():
+    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT UNIQUE,
+            password TEXT
+        )
+    """)
+    conn.commit()
+    conn.close()
+
+# Signup function
+def signup(username, password):
+    init_db()
+    conn = sqlite3.connect(DB_PATH, check_same_thread=False)
+    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."
+    except sqlite3.IntegrityError:
+        conn.close()
+        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"
+
+    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"
+
+# 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."
+
+# Download DB file
+def get_db_file():
+    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]
+            )
+
+demo.launch()