Update app.py

app.py

@@ -85,11 +85,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         filenames = os.listdir(video_path)
         filenames = [os.path.join(video_path, filename) for filename in filenames if not filename.startswith('.')]
         filenames.sort()
-
-    uploads = os.listdir("/tmp/gradio/")
-    for k, ul in enumerate(uploads):
-        print(ul)
-    print(len(uploads))
     
     # os.makedirs(outdir, exist_ok=True)
     
@@ -345,6 +340,12 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         with gr.Column():
+            coords = gr.Textbox(value="""50.07379596793083,14.437146122950555
+            50.073799567020004,14.437146774240507
+            50.07377647505558,14.437161000659017
+            50.07379496839027,14.437148958238538
+            50.073823157821664,14.437124189538856
+            """, label="Coordinates")
             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])
@@ -533,20 +534,21 @@ with gr.Blocks(css=css) as demo:
             <canvas id='cnv_out'/>""")
             render = gr.Button("Render")
     
-    def on_submit(uploaded_video,model_type):
-                
+    def on_submit(uploaded_video,model_type,coordinates):
+
+        print(coordinates)
         # Process the video and get the path of the output video
         output_video_path = make_video(uploaded_video,encoder=model_type)
 
         return output_video_path
 
-    submit.click(on_submit, inputs=[input_video, model_type], outputs=[processed_video, processed_zip, output_frame, output_depth])
+    submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_depth])
     render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()
-    example_files = [[os.path.join('examples', filename), os.path.join('coords', filename+'.vtt')] for filename in example_files]
-    examples = gr.Examples(examples=example_files, inputs=[input_video], outputs=[processed_video, processed_zip, output_frame, output_depth], fn=on_submit, cache_examples=True)
+    example_files = [os.path.join('examples', filename) for filename in example_files]
+    examples = gr.Examples(examples=example_files, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_depth], fn=on_submit, cache_examples=True)
     
 
 if __name__ == '__main__':