Update main.py

main.py

@@ -25,7 +25,51 @@ firebase_admin.initialize_app(cred, {
 
 bucket = storage.bucket()
 api_key = os.environ['Gemini']
-FIREBASE_API_KEY = os.environ.get('FIREBASE_API_KEY')  # For auth REST API calls
+FIREBASE_API_KEY = os.environ.get('FIREBASE_API_KEY')
+
+# Add test account creation function
+def create_test_accounts():
+    try:
+        # Create/admin2@test.com (Admin)
+        try:
+            admin_user = auth.create_user(
+                email='admin2@test.com',
+                password='password1'
+            )
+        except auth.EmailAlreadyExistsError:
+            admin_user = auth.get_user_by_email('admin2@test.com')
+        
+        db.reference(f'users/{admin_user.uid}').set({
+            'daily_cash': 300,
+            'remaining_cash': 300,
+            'last_reset': '2000-01-01T00:00:00+00:00',
+            'is_admin': True
+        })
+
+        # Create user1@test.com (Regular user)
+        try:
+            regular_user = auth.create_user(
+                email='user1@test.com',
+                password='123456'
+            )
+        except auth.EmailAlreadyExistsError:
+            regular_user = auth.get_user_by_email('user1@test.com')
+        
+        db.reference(f'users/{regular_user.uid}').set({
+            'daily_cash': 300,
+            'remaining_cash': 300,
+            'last_reset': '2000-01-01T00:00:00+00:00',
+            'is_admin': False
+        })
+        
+        print("Test accounts created/updated successfully")
+    except Exception as e:
+        print(f"Error creating test accounts: {str(e)}")
+
+# Create test accounts when server starts
+create_test_accounts()
+
+
 
 # Helper functions
 def configure_gemini():