Update main.py

main.py

@@ -334,87 +334,87 @@ def generate_story_endpoint():
         return jsonify({'error': str(e)}), 500
 
 # ---------- Video Generation Endpoint ----------
-app.route('/api/video/generate', methods=['POST'])
+@app.route('/api/video/generate', methods=['POST'])
 def generate_video_endpoint():
     try:
+        print("➡️ Received video generation request...")
+        
         # Verify user token
         auth_header = request.headers.get('Authorization', '')
         if not auth_header.startswith('Bearer '):
+            print("❌ ERROR: Missing or invalid token")
             return jsonify({'error': 'Missing or invalid token'}), 401
+
         token = auth_header.split(' ')[1]
         uid = verify_token(token)
         if not uid:
+            print("❌ ERROR: Invalid or expired token")
             return jsonify({'error': 'Invalid or expired token'}), 401
 
         data = request.get_json()
         story_id = data.get('story_id')
         if not story_id:
+            print("❌ ERROR: story_id is required")
             return jsonify({'error': 'story_id is required'}), 400
 
-        # Retrieve the story record from the database.
+        print(f"Fetching story {story_id} from Firebase...")
         story_ref = db.reference(f"stories/{story_id}")
         story_data = story_ref.get()
+
         if not story_data:
+            print("❌ ERROR: Story not found")
             return jsonify({'error': 'Story not found'}), 404
 
         sections = story_data.get("sections", [])
         if not sections:
+            print("❌ ERROR: No sections found in the story")
             return jsonify({'error': 'No sections found in the story'}), 404
 
-        # Download the image and audio files from their URLs.
         image_files = []
         audio_files = []
+        
+        print(f"Processing {len(sections)} sections...")
+
         for section in sections:
             image_url = section.get("image_url")
             audio_url = section.get("audio_url")
-            # Download image
+
+            print(f"➡️ Downloading image from: {image_url}")
             img_resp = requests.get(image_url)
             if img_resp.status_code == 200:
-                img_path = f"/tmp/{uuid.uuid4().hex}.jpg"
-                with open(img_path, "wb") as f:
-                    f.write(img_resp.content)
-                image_files.append(img_path)
-            # Download audio
+                img = Image.open(io.BytesIO(img_resp.content))
+                image_files.append(img)
+            else:
+                print(f"❌ ERROR: Failed to download image {image_url}")
+
+            print(f"➡️ Downloading audio from: {audio_url}")
             aud_resp = requests.get(audio_url)
             if aud_resp.status_code == 200:
                 aud_path = f"/tmp/{uuid.uuid4().hex}.mp3"
                 with open(aud_path, "wb") as f:
                     f.write(aud_resp.content)
                 audio_files.append(aud_path)
+            else:
+                print(f"❌ ERROR: Failed to download audio {audio_url}")
 
         if not image_files:
+            print("❌ ERROR: No valid images found")
             return jsonify({'error': 'No images available for video generation'}), 500
 
-        # Generate video using the video generation function (from video_gen.py)
-        from video_gen import create_video  # from your video_gen.py
+        # Generate video
         video_output_path = f"/tmp/{uuid.uuid4().hex}.mp4"
-        video_file = create_video(image_files, audio_files, output_path=video_output_path)
+        video_file = create_silent_video(image_files, [5.0] * len(image_files), video_output_path)
+
         if not video_file:
+            print("❌ ERROR: Video generation failed")
             return jsonify({'error': 'Video generation failed'}), 500
 
-        # Upload the video to Firebase Storage.
-        video_blob_name = f"stories/{uid}/{uuid.uuid4().hex}.mp4"
-        video_url = upload_to_storage(video_file, video_blob_name)
-
-        # Optionally, update the story record with the video URL.
-        story_ref.update({"video_url": video_url})
-
-          # --- Subtract 5 Credits from the User ---
-        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})
+        print(f"✅ Video generated successfully: {video_file}")
 
-        # Clean up temporary local files.
-        for f in image_files:
-            os.remove(f)
-        for f in audio_files:
-            os.remove(f)
-        os.remove(video_file)
+        return jsonify({"video_url": video_file})
 
-        return jsonify({"video_url": video_url,"new_credits": new_credits})
     except Exception as e:
+        print(f"❌ ERROR: {e}")
         return jsonify({'error': str(e)}), 500