Update app.py

app.py

@@ -611,19 +611,26 @@ with gr.Blocks(css=css) as demo:
     def on_submit(uploaded_video,model_type,coordinates):
         global locations
         locations = []
+        avg = [0, 0]
         
         if not coordinates:
             locations = json.loads(example_coords)
             for k, location in enumerate(locations):
                 locations[k] = location["latLng"]
-                locations[k]["lat"] = locations[k]["lat"] - locations[0]["lat"]
-                locations[k]["lng"] = locations[k]["lng"] - locations[0]["lng"]
+                avg[0] = avg[0] + locations[k]["lat"]
+                avg[1] = avg[1] + locations[k]["lng"]
         else:
             locations = json.loads(coordinates)
             for k, location in enumerate(locations):
                 locations[k] = location["location"]["latLng"]
-                locations[k]["lat"] = locations[k]["lat"] - locations[0]["lat"]
-                locations[k]["lng"] = locations[k]["lng"] - locations[0]["lng"]
+                avg[0] = avg[0] + locations[k]["lat"]
+                avg[1] = avg[1] + locations[k]["lng"]
+                
+        avg[0] = avg[0] / len(locations)
+        avg[1] = avg[1] / len(locations)
+        for k, location in enumerate(locations):
+            locations[k]["lat"] = location["lat"] - avg[0]
+            locations[k]["lng"] = location["lng"] - avg[1]
         print(locations)
             
         # Process the video and get the path of the output video