Update app.py (#2)

- Update app.py (07b6e2157f03e204ec1afd29cb40c7840d356b2a)


Co-authored-by: Nishith Jain <[email protected]>

app.py

@@ -8,7 +8,9 @@ import moviepy.editor as mp
 from pydub import AudioSegment
 from PIL import Image
 import numpy as np
-
+import os
+import tempfile
+import uuid
 
 torch.set_float32_matmul_precision(["high", "highest"][0])
 
@@ -48,9 +50,15 @@ def fn(vid):
  # Add the original audio back to the processed video
  processed_video = processed_video.set_audio(audio)
 
- # Return the processed video
- return processed_video
+ # Save the processed video to a temporary file
+ temp_dir = "temp"
+ os.makedirs(temp_dir, exist_ok=True)
+ unique_filename = str(uuid.uuid4()) + ".mp4"
+ temp_filepath = os.path.join(temp_dir, unique_filename)
+ processed_video.write_videofile(temp_filepath, codec="libx264")
 
+ # Return the path to the temporary file
+ return temp_filepath
 
 
 def process(image):