Update app.py

app.py

@@ -20,7 +20,6 @@ from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from moviepy.editor import *
 
 frame_selected = 0
-depth_selected = 0
 
 def zip_files(files_in, files_out):
     with ZipFile("depth_result.zip", "w") as zipObj:
@@ -229,7 +228,7 @@ def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
 
 def get_mesh(image, depth):
-    fnum = image.selected_index
+    fnum = frame_selected
     gdepth = rgb2gray(depth[fnum][0])
     print('depth to gray - ok')
     points = pano_depth_to_world_points(gdepth, 1)
@@ -271,20 +270,13 @@ def loadurl(url):
     return url
 
 def select_frame(evt: gr.SelectData):
-    global depth_selected
     global frame_selected
     if evt.index != frame_selected:
-        print(f'index: {evt.index}, d_s: {depth_selected}, f_s: {frame_selected}')
-        depth_selected = evt.index
-        return gr.Gallery(selected_index=evt.index, preview=True)
-
-def select_depth(evt: gr.SelectData):
-    global frame_selected
-    global depth_selected
-    if evt.index != depth_selected:
-        print(f'index: {evt.index}, f_s: {frame_selected}, d_s: {depth_selected}')
+        print(f'index: {evt.index}, f_s: {frame_selected}')
         frame_selected = evt.index
         return gr.Gallery(selected_index=evt.index, preview=True)
+    else:
+        return
 
 
 css = """
@@ -336,7 +328,7 @@ with gr.Blocks(css=css) as demo:
             output_frame = gr.Gallery(label="Frame", type='numpy', preview=True, columns=8192)
             output_depth = gr.Gallery(label="Depth", type='numpy', preview=True, columns=8192)
             output_frame.select(fn=select_frame, inputs=None, outputs=[output_depth])
-            output_depth.select(fn=select_depth, inputs=None, outputs=[output_frame])
+            output_depth.select(fn=select_frame, inputs=None, outputs=[output_frame])
             submit = gr.Button("Submit")
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')