Update app.py

app.py

@@ -20,6 +20,7 @@ from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from moviepy.editor import *
 
 frame_selected = 0
+locations = []
 
 def zip_files(files_in, files_out):
     with ZipFile("depth_result.zip", "w") as zipObj:
@@ -526,17 +527,32 @@ with gr.Blocks(css=css) as demo:
               ctxt.drawImage(this, 0, 0); 
             '/><br/>
             <canvas id='cnv_out'/>""")
-            example_coords = ['50.07379596793083,14.437146122950555 50.073799567020004,14.43714677424050 50.07377647505558,14.437161000659017 50.07379496839027,14.437148958238538 50.073823157821664,14.437124189538856']
-            with gr.Accordion(label="Coordinates", open=False):
-                coords = gr.Textbox(value=example_coords[0], label="Precise locations", show_label=False)
+            example_coords = '50.07379596793083,14.437146122950555 50.073799567020004,14.437146774240507 50.07377647505558,14.437161000659017 50.07379496839027,14.437148958238538 50.073823157821664,14.437124189538856'
+            with gr.Accordion(label="Locations", open=False):
+                coords = gr.Textbox(value=example_coords, label="Precise coordinates", show_label=False)
             render = gr.Button("Render")
     
     def on_submit(uploaded_video,model_type,coordinates):
-
+        
+        global locations
         if not coordinates:
-            print(example_coords[0])
+            locations = example_coords.split()
         else:
-            print(coordinates)
+            locations = coordinates.split()
+        print(locations)
+
+        avg_coords = [0.0, 0.0]
+        for location in locations:
+            location = location.split(',')
+            avg_coords[0] = avg_coords[0] + location[0]
+            avg_coords[1] = avg_coords[1] + location[1]
+        avg_coords[0] = avg_coords[0] / len(locations)
+        avg_coords[1] = avg_coords[1] / len(locations)
+        
+        for location in locations:
+            location[0] = location[0] - avg_coords[0]
+            location[1] = location[1] - avg_coords[1]
+        print(locations)
             
         # Process the video and get the path of the output video
         output_video_path = make_video(uploaded_video,encoder=model_type)