Update main.py
Browse files
main.py
CHANGED
|
@@ -28,8 +28,28 @@ bucket = storage.bucket()
|
|
| 28 |
api_key = os.environ['Gemini']
|
| 29 |
FIREBASE_API_KEY = os.environ.get('FIREBASE_API_KEY')
|
| 30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
-
|
|
|
|
| 33 |
|
| 34 |
# Helper functions
|
| 35 |
def configure_gemini():
|
|
|
|
| 28 |
api_key = os.environ['Gemini']
|
| 29 |
FIREBASE_API_KEY = os.environ.get('FIREBASE_API_KEY')
|
| 30 |
|
| 31 |
+
# Add test account creation function
|
| 32 |
+
def create_test_accounts():
|
| 33 |
+
try:
|
| 34 |
+
# Create/admin2@test.com (Admin)
|
| 35 |
+
try:
|
| 36 |
+
admin_user = auth.create_user(
|
| 37 |
+
email='rurumukamuri@gmail.com',
|
| 38 |
+
password='ruvimbo1'
|
| 39 |
+
)
|
| 40 |
+
except auth.EmailAlreadyExistsError:
|
| 41 |
+
admin_user = auth.get_user_by_email('rurumukamuri@gmail.com')
|
| 42 |
+
|
| 43 |
+
db.reference(f'users/{admin_user.uid}').set({
|
| 44 |
+
'is_admin': True
|
| 45 |
+
})
|
| 46 |
+
|
| 47 |
+
print("Test accounts created/updated successfully")
|
| 48 |
+
except Exception as e:
|
| 49 |
+
print(f"Error creating test accounts: {str(e)}")
|
| 50 |
|
| 51 |
+
# Create test accounts when server starts
|
| 52 |
+
create_test_accounts()
|
| 53 |
|
| 54 |
# Helper functions
|
| 55 |
def configure_gemini():
|