Update app.py

app.py

@@ -26,10 +26,10 @@ mediapy.set_ffmpeg(ffmpeg_path)
 
 
         
-def do_interpolation(frame1, frame2):
+def do_interpolation(frame1, frame2, times_to_interpolate):
     print(frame1, frame2)
     input_frames = [frame1, frame2]
-    times_to_interpolate = 2
+    #times_to_interpolate = 2
     frames = list(
         util.interpolate_recursively_from_files(
             input_frames, times_to_interpolate, interpolator))
@@ -83,12 +83,8 @@ def create_video(frames, fps, type):
     
     return type + "_result.mp4"
 
-def convertG2V(imported_gif):
-    clip = VideoFileClip(imported_gif.name)
-    clip.write_videofile("my_gif_video.mp4")
-    return "my_gif_video.mp4"
     
-def infer(video_in):
+def infer(video_in,interpolation,fps_output):
     
     
     # 1. break video into frames and get FPS
@@ -112,7 +108,7 @@ def infer(video_in):
     for idx, frame in enumerate(frames_list):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
-            interpolated_frames = do_interpolation(frame, next_frame) # should return a list of 3 interpolated frames
+            interpolated_frames = do_interpolation(frame, next_frame,interpolation) # should return a list of 3 interpolated frames
             break_interpolated_video = get_frames(interpolated_frames, "interpol")
             print(break_interpolated_video[0])
             for j, img in enumerate(break_interpolated_video[0]):
@@ -121,7 +117,7 @@ def infer(video_in):
                 result_frames.append(f"{frame}_to_{next_frame}_{j}.jpg")
             print("frames " + str(idx) + " & " + str(idx+1) + "/" + str(n_frame) + ": done;")
     
-    final_vid = create_video(result_frames, 8, "interpolated")
+    final_vid = create_video(result_frames, fps_output, "interpolated")
 
     files = [final_vid]
 
@@ -153,8 +149,8 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 video_input = gr.Video(source="upload", type="filepath")
-                gif_input = gr.File(label="import a GIF instead", file_types=['.gif'])
-                gif_input.change(fn=convertG2V, inputs=gif_input, outputs=video_input)
+                interpolation = gr.Slider(minimum=2,maximum=4,step=1, value=1, label="Interpolation Steps")
+                fps_output = gr.Radio([8, 12, 24], label="FPS output")
                 submit_btn = gr.Button("Submit")
             
             with gr.Column():
@@ -169,6 +165,6 @@ with gr.Blocks() as demo:
         #cache_examples=False
     #)
     
-    submit_btn.click(fn=infer, inputs=[video_input], outputs=[video_output, file_output])
+    submit_btn.click(fn=infer, inputs=[video_input,interpolation,fps_output], outputs=[video_output, file_output])
 
 demo.launch()