Update app.py

app.py

@@ -37,7 +37,7 @@ def do_interpolation(frame1, frame2, interpolation, n):
     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):
+def get_frames(video_in, step, name):
     frames = []
     cap = cv2.VideoCapture(video_in)
     cframes = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -69,9 +69,9 @@ def get_frames(video_in, step, name, resize_w):
         ret, frame = cap.read()
         if ret == False:
             break
-        if resize_w > 0:
-          resize_h = resize_w / 2.0
-          frame = cv2.resize(frame, (int(resize_w), int(resize_h)))
+        #if resize_w > 0:
+          #resize_h = resize_w / 2.0
+          #frame = cv2.resize(frame, (int(resize_w), int(resize_h)))
         
         cv2.imwrite(f"{name}_{step}{str(i)}.png",frame)
         frames.append(f"{name}_{step}{str(i)}.png")
@@ -92,6 +92,15 @@ def create_video(frames, fps, type):
 
     mediapy.write_video(type + "_result.mp4", imgs, fps=fps)
     return type + "_result.mp4"
+    
+
+def sharpest(f):
+    break_vid = get_frames(f, "vid_input_frame", "origin")
+    blur_s = []
+    for jdx, fr in enumerate(break_vid[0]):
+        blur_s.append(cv2.Laplacian(cv2.cvtColor(cv2.imread(fr).astype(np.uint8), cv2.COLOR_BGR2GRAY), cv2.CV_64F).var())
+    
+    return break_vid[0][np.argmax(blur_s)]
 
     
 def infer(f_in, interpolation, fps_output):
@@ -117,6 +126,11 @@ def infer(f_in, interpolation, fps_output):
         if idx < len(frames_list) - 1:
             next_frame = frames_list[idx+1]
 
+            ftype = frame.split('/')
+            if ftype[len(ftype)-1].split('.')[1] == 'mp4':
+                frame = sharpest(frame)
+                next_frame = sharpest(next_frame)
+
             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])
@@ -182,7 +196,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 with gr.Accordion(label="Upload files here", open=True):
-                    files_input = gr.File(file_count="multiple", file_types=['image'])
+                    files_input = gr.File(file_count="multiple", file_types=['image', '.mp4'])
                 gallery_input = gr.Gallery(label="Slideshow", preview=True, columns=8192, interactive=False)
                 files_input.change(fn=loadf, inputs=[files_input], outputs=[files_input, gallery_input])
                 with gr.Row():