Update app.py

app.py

@@ -25,7 +25,7 @@ ffmpeg_path = util.get_ffmpeg_path()
 mediapy.set_ffmpeg(ffmpeg_path)
 
 
-def do_interpolation(frame1, frame2, interpolation):
+def do_interpolation(frame1, frame2, interpolation, n):
     print("tween frames: " + str(interpolation))
     print(frame1, frame2)
     input_frames = [frame1, frame2]
@@ -34,8 +34,8 @@ def do_interpolation(frame1, frame2, interpolation):
             input_frames, int(interpolation), interpolator))
     
     #print(frames)
-    mediapy.write_video(f"{frame1}_to_{frame2}_out.mp4", frames, fps=25)
-    return f"{frame1}_to_{frame2}_out.mp4"
+    mediapy.write_video(f"{n}_to_{n+1}_out.mp4", frames, fps=25)
+    return f"{n}_to_{n+1}_out.mp4"
     
 def get_frames(video_in, step, name, resize_w):
     frames = []
@@ -117,17 +117,17 @@ def infer(glr_in, interpolation, fps_output):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
 
-            interpolated_frames = do_interpolation(frame[0], next_frame[0], interpolation) # should return a list of 3 interpolated frames
+            interpolated_frames = do_interpolation(frame[0], next_frame[0], interpolation, idx) # should return a list of interpolated frames
             break_interpolated_video = get_frames(interpolated_frames, "interpol", f"{idx}_", 0)
             print(break_interpolated_video[0])
             for j, img in enumerate(break_interpolated_video[0][0:len(break_interpolated_video[0])-1]):
                 print(f"IMG:{img}")
-                os.rename(img, f"{frame}_to_{next_frame}_{j}.png")
-                result_frames.append(f"{frame}_to_{next_frame}_{j}.png")
+                os.rename(img, f"{idx}_to_{idx+1}_{j}.png")
+                result_frames.append(f"{idx}_to_{idx+1}_{j}.png")
             
             print("frames " + str(idx) + " & " + str(idx+1) + "/" + str(n_frame) + ": done;")
             #print(f"CURRENT FRAMES: {result_frames}")
-    result_frames.append(f"{frames_list[n_frame-1]}")
+    result_frames.append(f"{frames_list[n_frame-1][0]}")
     final_vid = create_video(result_frames, fps_output, "interpolated")
 
     files = final_vid