Update main.py

main.py

@@ -51,13 +51,24 @@ async def generate_video_api(
                 frame.save(frame_file)
                 frame_files.append(frame_file)
 
+            # Create a temporary file for the video
+            with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_video:
+                temp_video_path = temp_video.name
+
             # Use ffmpeg-python to combine images into a video
             input_stream = ffmpeg.input(os.path.join(tmpdir, 'frame_%03d.png'), framerate=fps)
-            output_stream = ffmpeg.output(input_stream, 'pipe:', vcodec='libx264', pix_fmt='yuv420p', format='mp4')
-            out, err = ffmpeg.run(output_stream, capture_stdout=True, capture_stderr=True)
+            output_stream = ffmpeg.output(input_stream, temp_video_path, vcodec='libx264', pix_fmt='yuv420p')
+            ffmpeg.run(output_stream, overwrite_output=True)
+
+            # Read the temporary video file
+            with open(temp_video_path, 'rb') as video_file:
+                video_content = video_file.read()
+
+            # Delete the temporary video file
+            os.unlink(temp_video_path)
 
         # Return the video as a streaming response
-        return StreamingResponse(io.BytesIO(out), media_type="video/mp4")
+        return StreamingResponse(io.BytesIO(video_content), media_type="video/mp4")
 
     except requests.exceptions.HTTPError as http_err:
         # Handle HTTP errors