Update app.py

app.py

@@ -14,6 +14,7 @@ import tempfile
 from functools import partial
 import spaces
 from zipfile import ZipFile
+import json
 
 from depth_anything.dpt import DepthAnything
 from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
@@ -451,8 +452,14 @@ with gr.Blocks(css=css) as demo:
                 selected = gr.Number(elem_id="fnum", value=0, minimum=0, maximum=256, interactive=False)
                 output_frame.select(fn=select_frame, inputs=None, outputs=[output_depth, selected], show_progress='hidden')
                 output_depth.select(fn=select_frame, inputs=None, outputs=[output_frame, selected], show_progress='hidden')
-                example_coords = '50.07379596793083,14.437146122950555 50.073799567020004,14.437146774240507 50.07377647505558,14.437161000659017 50.07379496839027,14.437148958238538 50.073823157821664,14.437124189538856'
-                coords = gr.Textbox(elem_id="coords", value=example_coords, label="Precise coordinates", show_label=False)
+                example_coords = """[
+                  {latLng: { lat: 50.07379596793083, lng: 14.437146122950555 } },
+                  {latLng: { lat: 50.073799567020004, lng: 14.437146774240507 } },
+                  {latLng: { lat: 50.07377647505558, lng: 14.437161000659017 } },
+                  {latLng: { lat: 50.07379496839027, lng: 14.437148958238538 } },
+                  {latLng: { lat: 50.073823157821664, lng: 14.437124189538856 } }
+                ]"""
+                coords = gr.JSON(elem_id="coords", value=example_coords, label="Precise coordinates", show_label=False)
                 offset = gr.HTML(value="""<svg id='offset' height='256' width='256' viewBox='0 0 256 256' xmlns='http://www.w3.org/2000/svg' xmlns:xlink='http://www.w3.org/1999/xlink' style='touch-action:none;background-color:transparent;' onpointerdown='
                 if (!BABYLON.Engine.LastCreatedScene.activeCamera.metadata) {
                   var evt = document.createEvent(\"Event\");
@@ -605,30 +612,18 @@ with gr.Blocks(css=css) as demo:
         locations = []
         
         if not coordinates:
-            locations_str = example_coords.split()
+            locations = json.loads(example_coords)
         else:
-            locations_str = coordinates.split()
-        print(locations_str)
-
-        avg_coords = [0.0, 0.0]
-        for location in locations_str:
-            location = location.split(',')
-            avg_coords[0] = avg_coords[0] + float(location[0])
-            avg_coords[1] = avg_coords[1] + float(location[1])
-            
-        avg_coords[0] = avg_coords[0] / len(locations_str)
-        avg_coords[1] = avg_coords[1] / len(locations_str)
+            locations = json.loads(coordinates)
 
-        for location in locations_str:
-            location = location.split(',')
-            locations.append([ float(location[0]) - avg_coords[0], float(location[1]) - avg_coords[1] ])
+        for location in locations:
+            location = location.latLng
         print(locations)
             
         # Process the video and get the path of the output video
         output_video_path = make_video(uploaded_video,encoder=model_type)
-        output_video_path.append(locations)
 
-        return output_video_path
+        return output_video_path + (locations)
 
     submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_depth, coords])
     render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in, load_all], outputs=[result])