Fix credits balance

main.py

@@ -235,7 +235,6 @@ def upload_log():
 # Content
 # -----------------------
 # ---------- Story Generation Endpoint ----------
-
 @app.route('/api/story/generate', methods=['POST'])
 def generate_story_endpoint():
     try:
@@ -248,6 +247,14 @@ def generate_story_endpoint():
         if not uid:
             return jsonify({'error': 'Invalid or expired token'}), 401
 
+        # --- Check User Credits Before Generation ---
+        user_ref = db.reference(f"users/{uid}")
+        user_data = user_ref.get() or {}
+        current_credits = user_data.get("credits", 0)
+        
+        if current_credits < 5:
+            return jsonify({'error': 'Insufficient credits. You need at least 5 credits to generate a story.'}), 403
+
         # --- Read Request Data ---
         data = request.form.to_dict()  # For multipart/form-data
         input_type = data.get('input_type', 'text')  # "text", "pdf", "wiki", "bible", "youtube", "dataframe"
@@ -416,10 +423,7 @@ def generate_story_endpoint():
         story_ref.set(story_record)
 
         # Subtract 5 Credits
-        user_ref = db.reference(f"users/{uid}")
-        user_data = user_ref.get() or {}
-        current_credits = user_data.get("credits", 0)
-        new_credits = max(0, current_credits - 5)
+        new_credits = current_credits - 5
         user_ref.update({"credits": new_credits})
 
         preview = sections[0] if sections else {}
@@ -464,6 +468,14 @@ def generate_video_endpoint():
                 'log_url': upload_log()
             }), 401
 
+        user_ref = db.reference(f"users/{uid}")
+        user_data = user_ref.get() or {}
+        current_credits = user_data.get("credits", 0)
+
+        if current_credits < 5:
+            return jsonify({'error': 'Insufficient credits. You need at least 5 credits to generate a video.'}), 403
+
+
         data = request.get_json()
         story_id = data.get('story_id')
         if not story_id:
@@ -553,9 +565,7 @@ def generate_video_endpoint():
         story_ref.update({"video_url": video_url})
 
         # Deduct 5 credits
-        user_ref = db.reference(f"users/{uid}")
-        user_data = user_ref.get() or {}
-        current_credits = user_data.get("credits", 0)
+        
         new_credits = max(0, current_credits - 5)
         user_ref.update({"credits": new_credits})
         logging.info(f"✅ Deducted 5 credits. New credit balance: {new_credits}")