Update app.py

app.py

@@ -4,30 +4,29 @@ from pydantic import BaseModel
 import sqlite3
 import hashlib
 import os
+import gradio as gr
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import RedirectResponse
+from gradio.routes import mount_gradio_app
 
+# === FastAPI setup ===
 app = FastAPI()
 
-# Allow cross-origin requests (adjust for production)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=["*"],  # For testing; restrict this in production
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# ✅ Use a safe, writable location
+# === Database Setup ===
 DB_DIR = "/tmp"
 DB_PATH = os.path.join(DB_DIR, "users.db")
-
-# Ensure the directory exists
 os.makedirs(DB_DIR, exist_ok=True)
 
-# Connect to the database
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
-
-# Create users table
 cursor.execute('''
     CREATE TABLE IF NOT EXISTS users (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -37,11 +36,10 @@ cursor.execute('''
 ''')
 conn.commit()
 
-# Hash password using SHA256
 def hash_password(password: str) -> str:
     return hashlib.sha256(password.encode()).hexdigest()
 
-# Pydantic models
+# === Pydantic Models ===
 class SignupForm(BaseModel):
     email: str
     password: str
@@ -50,12 +48,11 @@ class LoginForm(BaseModel):
     email: str
     password: str
 
-# Signup route
+# === FastAPI Routes ===
 @app.post("/signup")
 def signup(data: SignupForm):
     email = data.email.lower().strip()
     password = hash_password(data.password)
-
     try:
         cursor.execute("INSERT INTO users (email, password) VALUES (?, ?)", (email, password))
         conn.commit()
@@ -63,21 +60,35 @@ def signup(data: SignupForm):
     except sqlite3.IntegrityError:
         raise HTTPException(status_code=409, detail="Email already registered")
 
-# Login route
 @app.post("/login")
 def login(data: LoginForm):
     email = data.email.lower().strip()
     password = hash_password(data.password)
-
     cursor.execute("SELECT * FROM users WHERE email = ? AND password = ?", (email, password))
     user = cursor.fetchone()
-
     if user:
         return {"success": True, "message": "Login successful"}
     else:
         raise HTTPException(status_code=401, detail="Invalid credentials")
 
-# Root route
 @app.get("/")
 def root():
-    return {"message": "FastAPI Auth API is running ✅"}
+    return RedirectResponse(url="/gradio")
+
+# === Gradio Interface ===
+def view_users():
+    cursor.execute("SELECT id, email FROM users")
+    rows = cursor.fetchall()
+    return [["ID", "Email"]] + [[str(r[0]), r[1]] for r in rows]
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 👥 Registered Users")
+    user_table = gr.Dataframe(
+        value=view_users(),
+        headers=None,
+        interactive=False
+    )
+    gr.Button("🔄 Refresh").click(fn=view_users, outputs=user_table)
+
+# === Mount Gradio App ===
+app = mount_gradio_app(app, demo, path="/gradio")