rairo commited on
Commit
950f620
·
verified ·
1 Parent(s): d0c9194

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +21 -1
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
- 500
 
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():