Update main.py

main.py

@@ -81,16 +81,69 @@ def verify_admin(auth_header):
     return uid
 
 # ─── Ensure dummy admin exists ───────────────────────────────────────────────────
+import firebase_admin
+from firebase_admin import credentials, auth, db # Ensure db is imported
+import os
+import json
+from datetime import datetime
+
+# --- Firebase Initialization (ensure this runs once) ---
+# You already have this in your app, just showing it here for context
+# credentials_json_string = os.environ.get("FIREBASE")
+# if credentials_json_string:
+#     credentials_json = json.loads(credentials_json_string)
+#     cred = credentials.Certificate(credentials_json)
+#     firebase_admin.initialize_app(cred, {
+#         'databaseURL': 'YOUR_FIREBASE_DB_URL', # Replace with your actual DB URL
+#         'storageBucket': 'YOUR_FIREBASE_STORAGE_BUCKET' # Replace with your actual Storage Bucket
+#     })
+# else:
+#     print("FIREBASE secret not set. Firebase Admin SDK not initialized.")
+# --------------------------------------------------------
+
+# ─── Ensure dummy admin exists and has DB entry ───────────────────────────────────
+admin_email = "rairorr@gmail.com"
+admin_password = "math1034!" # REMINDER: Change this for production!
+
 try:
-    admin_user = auth.get_user_by_email("rairorr@gmail.com")
+    admin_user = auth.get_user_by_email(admin_email)
+    print(f"Admin user '{admin_email}' already exists in Firebase Auth.")
 except firebase_admin._auth_utils.UserNotFoundError:
     admin_user = auth.create_user(
-        email="rairorr@gmail.com",
+        email=admin_email,
         email_verified=True,
-        password="admin2025",
+        password=admin_password,
         display_name="Admin"
     )
+    print(f"Admin user '{admin_email}' created in Firebase Auth with UID: {admin_user.uid}")
+
+# Set custom claim for Firebase ID Token (for future direct claim checks)
 auth.set_custom_user_claims(admin_user.uid, {"admin": True})
+print(f"Custom claim 'admin: True' set for UID: {admin_user.uid}")
+
+# --- IMPORTANT: Create/Update entry in Realtime Database ---
+user_ref = db.reference(f'users/{admin_user.uid}')
+user_data = user_ref.get()
+
+if not user_data:
+    # If no entry exists, create a new one
+    user_data_to_set = {
+        'email': admin_email,
+        'credits': 999999, # Give admin lots of credits or set to 0 if not needed
+        'is_admin': True, # Explicitly set is_admin to True in DB
+        'created_at': datetime.utcnow().isoformat(),
+        'suspended': False
+    }
+    user_ref.set(user_data_to_set)
+    print(f"Admin user data created in Realtime Database for UID: {admin_user.uid}")
+elif not user_data.get('is_admin', False):
+    # If entry exists but is_admin is false, update it
+    user_ref.update({'is_admin': True})
+    print(f"Admin user data updated in Realtime Database (is_admin set to True) for UID: {admin_user.uid}")
+else:
+    print(f"Admin user data already exists and is_admin is True in Realtime Database for UID: {admin_user.uid}")
+
+print("Admin setup complete.")
 
 # Decorator for credit deduction
 def credit_required(cost=1):