Update app.py

app.py

@@ -241,8 +241,7 @@ def pano_depth_to_world_points(depth):
 def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
 
-def get_mesh(depth, blur_data, loadall):
-    global frames
+def get_mesh(image, depth, blur_data, loadall):
     global locations
     global mesh
     if loadall == False:
@@ -250,7 +249,7 @@ def get_mesh(depth, blur_data, loadall):
     fnum = frame_selected
 
     #print(depth["composite"])
-    blur_img = blur_image(frames[fnum], depth["composite"], blur_data)
+    blur_img = blur_image(image[fnum][0], depth["composite"], blur_data)
     gdepth = rgb2gray(depth["composite"])
     
     print('depth to gray - ok')
@@ -380,7 +379,7 @@ with gr.Blocks(css=css) as demo:
             input_url = gr.Textbox(value="./examples/streetview.mp4", label="URL")
             input_video = gr.Video(label="Input Video", format="mp4")
             input_url.change(fn=loadurl, inputs=[input_url], outputs=[input_video])
-            output_frame = gr.Gallery(label="Frames", preview=True, columns=8192, type="filepath")
+            output_frame = gr.Gallery(label="Frames", preview=True, columns=8192, type="numpy")
             output_switch = gr.Checkbox(label="Show depths")
             output_switch.input(fn=switch_rows, inputs=[output_switch], outputs=[output_frame])
             output_mask = gr.ImageEditor(interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(colors=['black', 'darkgray', 'gray', 'lightgray', 'white']), layers=True)
@@ -749,7 +748,7 @@ try {
         return output_video_path + (locations,)
 
     submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, coords])
-    render.click(partial(get_mesh), inputs=[output_mask, blur_in, load_all], outputs=[result])
+    render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()