Update app.py

app.py

@@ -171,7 +171,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         global frame_selected
         global masks
         masks = orig_frames
-        return final_vid, final_zip, orig_frames, depth_frames, masks[frame_selected] #output_path
+        return final_vid, final_zip, np.concatenate((orig_frames, depth_frames), axis=0), masks[frame_selected] #output_path
 
 def depth_edges_mask(depth):
     """Returns a mask of edges in the depth map.
@@ -304,7 +304,7 @@ def blur_image(image, depth, blur_data):
 def loadurl(url):
     return url
 
-def select_frame(v, evt: gr.SelectData):
+def select_frame(v):
     global frame_selected
     global masks
     masks[frame_selected] = v
@@ -313,7 +313,7 @@ def select_frame(v, evt: gr.SelectData):
         frame_selected = evt.index
     v = masks[frame_selected]
     #print(v)
-    return gr.Gallery(selected_index=evt.index, preview=True), v, frame_selected
+    return v, frame_selected
 
 
 css = """
@@ -361,8 +361,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="Frame", type='numpy', preview=True, columns=8192)
-            output_depth = gr.Gallery(label="Depth", type='numpy', preview=True, columns=8192, interactive=False)
+            output_frame = gr.Gallery(label="Frames", type='numpy', preview=True, columns=256)
             output_mask = gr.ImageEditor(interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(colors=['black', 'darkgray', 'gray', 'lightgray', 'white']), layers=True)
             submit = gr.Button("Submit")
         with gr.Column():
@@ -574,8 +573,7 @@ with gr.Blocks(css=css) as demo:
       <a id='move' href='#'>move</a>    <a id='rotate' href='#'>rotate</a>    <a id='scale' href='#'>scale</a>
                 </pre>""")
                 selected = gr.Number(elem_id="fnum", value=0, minimum=0, maximum=256, interactive=False)
-                output_frame.select(fn=select_frame, inputs=[output_mask], outputs=[output_depth, output_mask, selected], show_progress='hidden')
-                output_depth.select(fn=select_frame, inputs=[output_mask], outputs=[output_frame, output_mask, selected], show_progress='hidden')
+                output_frame.select(fn=select_frame, inputs=[output_mask], outputs=[output_mask, selected], show_progress='hidden')
                 example_coords = """[
                   {"latLng": { "lat": 50.07379596793083, "lng": 14.437146122950555 } },
                   {"latLng": { "lat": 50.073799567020004, "lng": 14.437146774240507 } },
@@ -729,14 +727,14 @@ with gr.Blocks(css=css) as demo:
 
         return output_video_path + (locations,)
 
-    submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_depth, output_mask, coords])
+    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_frame, output_depth, blur_in, load_all], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()
     example_files = [os.path.join('examples', filename) for filename in example_files]
     
-    examples = gr.Examples(examples=example_files, inputs=[input_video], outputs=[processed_video, processed_zip, output_frame, output_depth, output_mask, coords], fn=on_submit, cache_examples=True)
+    examples = gr.Examples(examples=example_files, inputs=[input_video], outputs=[processed_video, processed_zip, output_frame, output_mask, coords], fn=on_submit, cache_examples=True)
     
 
 if __name__ == '__main__':