Update main.py

main.py

@@ -1,12 +1,10 @@
 from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 import uvicorn
-from fastapi.responses import StreamingResponse, JSONResponse
+from fastapi.responses import StreamingResponse
 import io
-import base64
 import requests
-import os
-import subprocess
 from PIL import Image
+import subprocess
 
 app = FastAPI()
 
@@ -23,12 +21,8 @@ async def generate_video_api(
         # Read the uploaded image file
         image_content = await file.read()
         
-        # Save the image temporarily
-        with open("input_image.png", "wb") as f:
-            f.write(image_content)
-
         # Generate a sequence of images
-        images = []
+        frames = []
         for i in range(num_frames):
             # Prepare the payload
             payload = {
@@ -43,26 +37,39 @@ async def generate_video_api(
             response = requests.post(API_URL, json=payload)
             response.raise_for_status()
 
-            # Save the generated image
-            image = Image.open(io.BytesIO(response.content))
-            image.save(f"frame_{i:03d}.png")
-            images.append(f"frame_{i:03d}.png")
+            # Add the generated image to frames
+            frames.append(Image.open(io.BytesIO(response.content)))
+
+        # Create a video from frames using PIL
+        video_frames = []
+        for frame in frames:
+            img_byte_arr = io.BytesIO()
+            frame.save(img_byte_arr, format='PNG')
+            video_frames.append(img_byte_arr.getvalue())
 
         # Use FFmpeg to combine images into a video
-        subprocess.run([
-            'ffmpeg', '-framerate', str(fps), '-i', 'frame_%03d.png',
-            '-c:v', 'libx264', '-pix_fmt', 'yuv420p', 'output.mp4'
-        ])
+        ffmpeg_cmd = [
+            'ffmpeg',
+            '-f', 'image2pipe',
+            '-framerate', str(fps),
+            '-i', '-',
+            '-c:v', 'libx264',
+            '-pix_fmt', 'yuv420p',
+            '-movflags', '+faststart',
+            '-f', 'mp4',
+            '-'
+        ]
 
-        # Read the video file
-        with open("output.mp4", "rb") as f:
-            video_content = f.read()
+        process = subprocess.Popen(ffmpeg_cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        
+        for frame in video_frames:
+            process.stdin.write(frame)
+        
+        process.stdin.close()
+        video_content, stderr = process.communicate()
 
-        # Clean up temporary files
-        for img in images:
-            os.remove(img)
-        os.remove("input_image.png")
-        os.remove("output.mp4")
+        if process.returncode != 0:
+            raise Exception(f"FFmpeg error: {stderr.decode()}")
 
         # Return the video as a streaming response
         return StreamingResponse(io.BytesIO(video_content), media_type="video/mp4")
@@ -70,9 +77,7 @@ async def generate_video_api(
     except requests.exceptions.HTTPError as http_err:
         # Handle HTTP errors
         error_detail = f"HTTP error occurred: {http_err}"
-        if response.status_code == 500:
-            error_detail += f"\nResponse content: {response.text}"
-        raise HTTPException(status_code=response.status_code, detail=error_detail)
+        raise HTTPException(status_code=500, detail=error_detail)
 
     except Exception as err:
         # Handle any other errors