Fix: move admin assignment logic into main.py startup for reliability

Dockerfile

@@ -31,5 +31,5 @@ ENV PORT=7860
 # Expose port
 EXPOSE ${PORT}
 
-# Start uvicorn with configurable port (and run admin script)
-CMD python hf_make_admin.py && uvicorn app.main:app --host 0.0.0.0 --port ${PORT}
+# Start uvicorn with configurable port
+CMD uvicorn app.main:app --host 0.0.0.0 --port ${PORT}

app/main.py

@@ -51,6 +51,30 @@ async def startup_event():
     print("Initializing database...")
     init_db()
     print("Database initialized successfully!")
+    
+    # Check for ADMIN_EMAIL to grant admin privileges
+    import os
+    from sqlalchemy.orm import Session
+    from app.database.connection import SessionLocal
+    from app.database.models import User
+    
+    admin_email = os.environ.get('ADMIN_EMAIL')
+    if admin_email:
+        db = SessionLocal()
+        try:
+            print(f"Checking admin status for: {admin_email}")
+            user = db.query(User).filter(User.email == admin_email).first()
+            if user:
+                if not user.is_admin:
+                    user.is_admin = 1
+                    db.commit()
+                    print(f"✅ User {admin_email} has been granted ADMIN privileges!")
+                else:
+                    print(f"ℹ️ User {admin_email} is already an admin.")
+            else:
+                print(f"⚠️ User {admin_email} not found. Signup first to become admin.")
+        finally:
+            db.close()
 
 
 @app.on_event("shutdown")

hf_make_admin.py

@@ -1,20 +0,0 @@
-import sqlite3
-import os
-
-# This script will run on HF Spaces and make the email provided in the ADMIN_EMAIL secret an admin
-email = os.environ.get('ADMIN_EMAIL')
-
-if not email:
-    print("❌ ADMIN_EMAIL environment variable not set. Please add it to HF Space Secrets.")
-else:
-    try:
-        conn = sqlite3.connect('/app/data/database.db')
-        conn.execute("UPDATE users SET is_admin = 1 WHERE email = ?", (email,))
-        conn.commit()
-        if conn.total_changes > 0:
-            print(f"✅ User {email} is now an ADMIN!")
-        else:
-            print(f"⚠️ User {email} not found in database. Sign up with this email first.")
-        conn.close()
-    except Exception as e:
-        print(f"❌ Error: {e}")