Update main.py

main.py

@@ -365,19 +365,28 @@ def generate_video_endpoint():
         auth_header = request.headers.get('Authorization', '')
         if not auth_header.startswith('Bearer '):
             logging.error("❌ ERROR: Missing or invalid token")
-            return jsonify({'error': 'Missing or invalid token', 'log_url': upload_log()}), 401
+            return jsonify({
+                'error': 'Missing or invalid token',
+                'log_url': upload_log()  # Upload log file so you can see the error
+            }), 401
 
         token = auth_header.split(' ')[1]
         uid = verify_token(token)
         if not uid:
             logging.error("❌ ERROR: Invalid or expired token")
-            return jsonify({'error': 'Invalid or expired token', 'log_url': upload_log()}), 401
+            return jsonify({
+                'error': 'Invalid or expired token',
+                'log_url': upload_log()
+            }), 401
 
         data = request.get_json()
         story_id = data.get('story_id')
         if not story_id:
             logging.error("❌ ERROR: story_id is required")
-            return jsonify({'error': 'story_id is required', 'log_url': upload_log()}), 400
+            return jsonify({
+                'error': 'story_id is required',
+                'log_url': upload_log()
+            }), 400
 
         logging.info(f"Fetching story {story_id} from Firebase...")
         story_ref = db.reference(f"stories/{story_id}")
@@ -385,18 +394,25 @@ def generate_video_endpoint():
 
         if not story_data:
             logging.error("❌ ERROR: Story not found")
-            return jsonify({'error': 'Story not found', 'log_url': upload_log()}), 404
+            return jsonify({
+                'error': 'Story not found',
+                'log_url': upload_log()
+            }), 404
 
         sections = story_data.get("sections", [])
         if not sections:
             logging.error("❌ ERROR: No sections found in the story")
-            return jsonify({'error': 'No sections found in the story', 'log_url': upload_log()}), 404
+            return jsonify({
+                'error': 'No sections found in the story',
+                'log_url': upload_log()
+            }), 404
 
         image_files = []
         audio_files = []
 
         logging.info(f"Processing {len(sections)} sections...")
 
+        # Download each image and audio file
         for section in sections:
             image_url = section.get("image_url")
             audio_url = section.get("audio_url")
@@ -406,6 +422,7 @@ def generate_video_endpoint():
             if img_resp.status_code == 200:
                 img = Image.open(io.BytesIO(img_resp.content))
                 image_files.append(img)
+                logging.info("✅ Image downloaded successfully.")
             else:
                 logging.error(f"❌ ERROR: Failed to download image {image_url}")
 
@@ -416,27 +433,61 @@ def generate_video_endpoint():
                 with open(aud_path, "wb") as f:
                     f.write(aud_resp.content)
                 audio_files.append(aud_path)
+                logging.info("✅ Audio downloaded successfully.")
             else:
                 logging.error(f"❌ ERROR: Failed to download audio {audio_url}")
 
         if not image_files:
             logging.error("❌ ERROR: No valid images found")
-            return jsonify({'error': 'No images available for video generation', 'log_url': upload_log()}), 500
+            return jsonify({
+                'error': 'No images available for video generation',
+                'log_url': upload_log()
+            }), 500
 
+        # Create the video
         video_output_path = f"/tmp/{uuid.uuid4().hex}.mp4"
+        logging.info("Starting create_video...")
         video_file = create_video(image_files, audio_files, output_path=video_output_path)
 
         if not video_file:
             logging.error("❌ ERROR: Video generation failed")
-            return jsonify({'error': 'Video generation failed', 'log_url': upload_log()}), 500
+            return jsonify({
+                'error': 'Video generation failed',
+                'log_url': upload_log()
+            }), 500
 
         logging.info(f"✅ Video generated successfully: {video_file}")
 
-        return jsonify({"video_url": video_file, "log_url": upload_log()})
+        # Upload the video to Firebase Storage
+        logging.info("Uploading video to Firebase Storage...")
+        video_blob_name = f"stories/{uid}/{uuid.uuid4().hex}.mp4"
+        video_url = upload_to_storage(video_file, video_blob_name)
+        logging.info(f"✅ Video uploaded to {video_url}")
+
+        # Update the story record with the video URL
+        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}")
+
+        return jsonify({
+            "video_url": video_url,
+            "new_credits": new_credits,
+            "log_url": upload_log()  # Upload the final log so you can inspect it
+        })
 
     except Exception as e:
-        logging.error(f"❌ ERROR: {e}\n{traceback.format_exc()}")
-        return jsonify({'error': str(e), 'log_url': upload_log()}), 500
+        trace = traceback.format_exc()
+        logging.error(f"❌ EXCEPTION: {str(e)}\n{trace}")
+        return jsonify({
+            'error': str(e),
+            'log_url': upload_log()
+        }), 500
 
 
 #----------Image Generation Endpoint ----------