Update app.py

app.py

@@ -106,10 +106,10 @@ def get_frames(video_in):
     return frames, fps
 
 
-def create_video(frames, fps):
+def create_video(frames, fps, type):
     print("building video result")
     clip = ImageSequenceClip(frames, fps=fps)
-    clip.write_videofile("movie.mp4", fps=fps)
+    clip.write_videofile(f"video_{type}_result.mp4", fps=fps)
     
     return 'movie.mp4'
 
@@ -324,9 +324,9 @@ def infer(video_in, trim_value, prompt, background_prompt):
         with_matte_result_frames.append(f"matte_result_img-{i}.jpg")
         print("frame " + i + "/" + str(n_frame) + ": done;")
 
-    vid_bg = create_video(with_bg_result_frames, fps)
-    vid_green = create_video(with_green_result_frames, fps)
-    vid_matte = create_video(with_matte_result_frames, fps)
+    vid_bg = create_video(with_bg_result_frames, fps, "bg")
+    vid_green = create_video(with_green_result_frames, fps, "greenscreen")
+    vid_matte = create_video(with_matte_result_frames, fps, "matte")
     print("finished !")
     
     return vid_bg, vid_green, vid_matte