Update app.py

app.py

@@ -38,7 +38,7 @@ def do_interpolation(frame1, frame2, times_to_interpolate):
     mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=12)
     return f"{frame1}_to_{frame2}_out.mp4"
     
-def get_frames(video_in, step):
+def get_frames(video_in, step, name):
     frames = []
     #resize the video
     clip = VideoFileClip(video_in)
@@ -65,8 +65,8 @@ def get_frames(video_in, step):
         ret, frame = cap.read()
         if ret == False:
             break
-        cv2.imwrite(f"{step}{str(i)}.jpg",frame)
-        frames.append(f"{step}{str(i)}.jpg")
+        cv2.imwrite(f"{name}_{step}{str(i)}.jpg",frame)
+        frames.append(f"{name}_{step}{str(i)}.jpg")
         i+=1
     
     cap.release()
@@ -88,7 +88,7 @@ def infer(video_in,interpolation,fps_output):
     
     
     # 1. break video into frames and get FPS
-    break_vid = get_frames(video_in, "vid_input_frame")
+    break_vid = get_frames(video_in, "vid_input_frame", "origin")
     frames_list= break_vid[0]
     fps = break_vid[1]
     n_frame = int(4*fps) #limited to 4 seconds
@@ -109,7 +109,7 @@ def infer(video_in,interpolation,fps_output):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
             interpolated_frames = do_interpolation(frame, next_frame,interpolation) # should return a list of 3 interpolated frames
-            break_interpolated_video = get_frames(interpolated_frames, "interpol")
+            break_interpolated_video = get_frames(interpolated_frames, "interpol",f"{idx}_")
             print(break_interpolated_video[0])
             for j, img in enumerate(break_interpolated_video[0]):
                 print(f"IMG:{img}")