refactor: Update remaining generations to 20 in get_generations function

Arghya721 · Aug 18, 2024 · 53234d6 · 53234d6
1 parent 48149fc
commit 53234d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -594,11 +594,11 @@ def get_generations(token_info: dict = Depends(verify_google_token)):
         # create a new document for the user with the remaining generations
         user_generations_ref.set({
             'google_user_id': token_info['sub'],
-            'remaining_generations': 30,
+            'remaining_generations': 20,
             'created_at': google_firestore.SERVER_TIMESTAMP,
             'updated_at': google_firestore.SERVER_TIMESTAMP,
         })
-        return 30
+        return 20
 
 def update_generations_left(token_info: dict = Depends(verify_google_token), generations_left: int = 30):
     """Update the number of generations left for the user."""