Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 
 app = FastAPI()
 
-# Allow all CORS origins (change this if needed for production)
+# Allow cross-origin requests (adjust for production)
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -16,16 +16,18 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ✅ Ensure DB path is in a writable directory for Hugging Face Spaces
-HF_DATA_DIR = os.getenv("HF_HOME", "/data")
-os.makedirs(HF_DATA_DIR, exist_ok=True)
-db_path = os.path.join(HF_DATA_DIR, "users.db")
+# ✅ Use a safe, writable location
+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)
+conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 
-# Create users table if it doesn't exist
+# Create users table
 cursor.execute('''
     CREATE TABLE IF NOT EXISTS users (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -35,11 +37,11 @@ cursor.execute('''
 ''')
 conn.commit()
 
-# Utility function to hash passwords securely
+# Hash password using SHA256
 def hash_password(password: str) -> str:
     return hashlib.sha256(password.encode()).hexdigest()
 
-# Request models
+# Pydantic models
 class SignupForm(BaseModel):
     email: str
     password: str
@@ -48,7 +50,7 @@ class LoginForm(BaseModel):
     email: str
     password: str
 
-# Route to sign up new users
+# Signup route
 @app.post("/signup")
 def signup(data: SignupForm):
     email = data.email.lower().strip()
@@ -61,7 +63,7 @@ def signup(data: SignupForm):
     except sqlite3.IntegrityError:
         raise HTTPException(status_code=409, detail="Email already registered")
 
-# Route to log in existing users
+# Login route
 @app.post("/login")
 def login(data: LoginForm):
     email = data.email.lower().strip()
@@ -75,7 +77,7 @@ def login(data: LoginForm):
     else:
         raise HTTPException(status_code=401, detail="Invalid credentials")
 
-# Test root route
+# Root route
 @app.get("/")
 def root():
-    return {"message": "FastAPI Auth API is running 🚀"}
+    return {"message": "FastAPI Auth API is running ✅"}