Update app.py

app.py

@@ -245,19 +245,19 @@ def save_video_to_mp4(video_tensor, fps=20):
     if video_numpy.max() <= 1.0:
         video_numpy = (video_numpy * 255).astype(np.uint8)
     
-    ## Create a temporary file to save the video
-    #temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-    #output_path = temp_file.name
-    # Create an in-memory byte buffer to store the video
-    byte_buffer = io.BytesIO()
+    # Create a temporary file to save the video
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    output_path = temp_file.name
+    ## Create an in-memory byte buffer to store the video
+    #byte_buffer = io.BytesIO()
     
     # Get video dimensions
     height, width, channels = video_numpy[0].shape
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec for .mp4
     
     # Create VideoWriter object
-    #out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
-    out = cv2.VideoWriter(byte_buffer, fourcc, fps, (width, height), isColor=True)
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+    #out = cv2.VideoWriter(byte_buffer, fourcc, fps, (width, height), isColor=True)
     
     # Write the frames to the output file
     for frame in video_numpy:
@@ -266,9 +266,9 @@ def save_video_to_mp4(video_tensor, fps=20):
         out.write(frame_bgr)
     
     out.release()
-    # Return the byte buffer's content (the video as bytes)
-    byte_buffer.seek(0)
-    return byte_buffer.read() #output_path
+    ## Return the byte buffer's content (the video as bytes)
+    #byte_buffer.seek(0)
+    return output_path #byte_buffer.read()
     
 # Function to list available videos dynamically
 def list_videos():