Update app.py

app.py

@@ -93,15 +93,13 @@ def create_video(frames, fps, type):
     mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
     return type + "_result.mp4"
 
-def sortFiles(e):
-  return e[0]
     
-def infer(glr_in, interpolation, fps_output):
+def infer(f_in, interpolation, fps_output):
     
     fps_output = logscale(fps_output)
     # 1. break video into frames and get FPS
     #break_vid = get_frames(url_in, "vid_input_frame", "origin", resize_n)
-    frames_list = glr_in.sort(key=sortFiles) #break_vid[0]
+    frames_list = f_in #break_vid[0]
     fps = 1 #break_vid[1]
     print(f"ORIGIN FPS: {fps}")
     n_frame = int(300) #limited to 300 frames
@@ -119,7 +117,7 @@ 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, idx) # should return a list of interpolated frames
+            interpolated_frames = do_interpolation(frame, next_frame, 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]):
@@ -129,7 +127,7 @@ def infer(glr_in, interpolation, fps_output):
             
             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][0]}")
+    result_frames.append(f"{frames_list[n_frame-1]}")
     final_vid = create_video(result_frames, fps_output, "interpolated")
 
     files = final_vid
@@ -145,8 +143,9 @@ def logscale(linear):
 def linscale(linear):
     return int(math.log2(linear))
 
-def loadurl(url):
-    return url
+def loadf(f):
+    f = f.sort()
+    return f
 
 
 title="""
@@ -177,8 +176,9 @@ with gr.Blocks() as demo:
         gr.HTML(title)
         with gr.Row():
             with gr.Column():
-                gallery_input = gr.Gallery(label="Slides", preview=True, columns=8192)
-                #video_input.change(fn=loadurl, inputs=[video_input], outputs=[url_input])
+                files_input = gr.File(file_count="directory")
+                gallery_input = gr.Gallery(label="Slides", preview=True, columns=8192, interactive=False)
+                files_input.change(fn=loadf, inputs=[files_input], outputs=[gallery_input])
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")
                     interpolation = gr.Number(value=1, show_label=False, interactive=False)
@@ -196,11 +196,11 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=[[["./examples/0.png", "./examples/1.png", "./examples/2.png", "./examples/3.png", "./examples/4.png"], 1, 0]],
         fn=infer,
-        inputs=[gallery_input, interpolation_slider, fps_output_slider],
+        inputs=[files_input, interpolation_slider, fps_output_slider],
         outputs=[video_output, file_output],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[gallery_input, interpolation_slider, fps_output_slider], outputs=[video_output, file_output])
+    submit_btn.click(fn=infer, inputs=[files_input, interpolation_slider, fps_output_slider], outputs=[video_output, file_output])
 
 demo.launch()