Update app.py

app.py

@@ -2,6 +2,18 @@ from flask import Flask, render_template, request, redirect, url_for, session, f
 from flask_sqlalchemy import SQLAlchemy
 from werkzeug.security import generate_password_hash, check_password_hash
 
+import os
+
+# Check if running on Hugging Face
+if os.getenv("SPACE_ID"):  # Hugging Face sets this env var
+    db_path = os.path.join("/tmp", "database.db")
+else:
+    db_path = os.path.join(os.path.dirname(__file__), "database.db")
+
+app.config["SQLALCHEMY_DATABASE_URI"] = f"sqlite:///{db_path}"
+app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+
+
 # Initialize Flask app
 app = Flask(__name__)
 app.secret_key = 'your_super_secret_key'