Update app.py

app.py

@@ -1,45 +1,32 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 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
+import hashlib
+import motor.motor_asyncio
 
-# === FastAPI setup ===
 app = FastAPI()
 
+# === CORS ===
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # For testing; restrict this in production
+    allow_origins=["*"],  # Open CORS (limit for prod)
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# === Database Setup ===
-DB_DIR = "/tmp"
-DB_PATH = os.path.join(DB_DIR, "users.db")
-os.makedirs(DB_DIR, exist_ok=True)
-
-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,
-        email TEXT UNIQUE NOT NULL,
-        password TEXT NOT NULL
-    )
-''')
-conn.commit()
+# === MongoDB Setup ===
+MONGO_URI = os.getenv("MONGO_URI")
+client = motor.motor_asyncio.AsyncIOMotorClient(MONGO_URI)
+db = client["cps_db"]  # database name
+users_collection = db["users"]
 
+# === Utils ===
 def hash_password(password: str) -> str:
     return hashlib.sha256(password.encode()).hexdigest()
 
-# === Pydantic Models ===
+# === Models ===
 class SignupForm(BaseModel):
     email: str
     password: str
@@ -48,24 +35,25 @@ class LoginForm(BaseModel):
     email: str
     password: str
 
-# === FastAPI Routes ===
+# === Routes ===
 @app.post("/signup")
-def signup(data: SignupForm):
+async 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()
-        return {"success": True, "message": "Account created successfully"}
-    except sqlite3.IntegrityError:
+
+    existing = await users_collection.find_one({"email": email})
+    if existing:
         raise HTTPException(status_code=409, detail="Email already registered")
 
+    await users_collection.insert_one({"email": email, "password": password})
+    return {"success": True, "message": "Account created successfully"}
+
 @app.post("/login")
-def login(data: LoginForm):
+async 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()
+
+    user = await users_collection.find_one({"email": email, "password": password})
     if user:
         return {"success": True, "message": "Login successful"}
     else:
@@ -73,22 +61,4 @@ def login(data: LoginForm):
 
 @app.get("/")
 def root():
-    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")
+    return {"message": "MongoDB FastAPI is running ✅"}