Merge pull request #152 from hotosm/feature/gpx_feedback_aoi

Feature : GPX Generator for feedback

backend/core/urls.py

@@ -11,6 +11,7 @@
     FeedbackLabelViewset,
     FeedbackView,
     FeedbackViewset,
+    GenerateFeedbackAOIGpxView,
     GenerateGpxView,
     LabelViewSet,
     ModelViewSet,
@@ -49,10 +50,13 @@
     path("training/status/<str:run_id>/", run_task_status),
     path("training/publish/<int:training_id>/", publish_training),
     path("prediction/", PredictionView.as_view()),
-    path("apply/feedback/", FeedbackView.as_view()),
+    path("feedback/training/submit/", FeedbackView.as_view()),
     path("status/", APIStatus.as_view()),
     path("geojson2osm/", geojson2osmconverter, name="geojson2osmconverter"),
     path("aoi/gpx/<int:aoi_id>/", GenerateGpxView.as_view()),
+    path(
+        "feedback-aoi/gpx/<int:feedback_aoi_id>/", GenerateFeedbackAOIGpxView.as_view()
+    ),
     path("workspace/", TrainingWorkspaceView.as_view()),
     path(
         "workspace/download/<path:lookup_dir>/", TrainingWorkspaceDownloadView.as_view()

backend/core/utils.py

@@ -2,12 +2,14 @@
 import json
 import math
 import os
+from datetime import datetime
 from uuid import uuid4
 from xml.dom import ValidationErr
 from zipfile import ZipFile
 
 import requests
 from django.conf import settings
+from gpxpy.gpx import GPX, GPXTrack, GPXTrackSegment, GPXWaypoint
 from tqdm import tqdm
 
 from .models import AOI, FeedbackAOI, FeedbackLabel, Label
@@ -243,6 +245,33 @@ def remove_file(path: str) -> None:
     os.unlink(path)
 
 
+def gpx_generator(geom_json):
+    """Generates GPX for give geojson geometry
+
+    Args:
+        geom_json (_type_): _description_
+
+    Returns:
+        xml: gpx
+    """
+
+    gpx = GPX()
+    gpx_track = GPXTrack()
+    gpx.tracks.append(gpx_track)
+    gpx_segment = GPXTrackSegment()
+    gpx_track.segments.append(gpx_segment)
+    for point in geom_json["coordinates"][0]:
+        # Append each point as a GPXWaypoint to the GPXTrackSegment
+        gpx_segment.points.append(GPXWaypoint(point[1], point[0]))
+    gpx.creator = "fAIr"
+    gpx_track.name = "Don't Edit this Boundary"
+    gpx_track.description = "Map inside this boundary and go back to fAIr UI"
+    gpx.time = datetime.now()
+    gpx.link = "https://github.com/hotosm/fAIr"
+    gpx.link_text = "AI Assisted Mapping - fAIr : HOTOSM"
+    return gpx.to_xml()
+
+
 def process_feature(feature, aoi_id, foreign_key_id, feedback=False):
     """Multi thread process of features"""
     properties = feature["properties"]

backend/core/views.py

@@ -27,7 +27,6 @@
 from django_filters.rest_framework import DjangoFilterBackend
 from drf_yasg.utils import swagger_auto_schema
 from geojson2osm import geojson2osm
-from gpxpy.gpx import GPX, GPXTrack, GPXTrackSegment, GPXWaypoint
 from hot_fair_utilities import polygonize, predict, vectorize
 from login.authentication import OsmAuthentication
 from login.permissions import IsOsmAuthenticated
@@ -66,6 +65,7 @@
     download_imagery,
     get_dir_size,
     get_start_end_download_coords,
+    gpx_generator,
     process_rawdata,
     request_rawdata,
 )
@@ -624,21 +624,19 @@ def get(self, request, aoi_id: int):
         # Convert the polygon field to GPX format
         geom_json = json.loads(aoi.geom.json)
         # Create a new GPX object
-        gpx = GPX()
-        gpx_track = GPXTrack()
-        gpx.tracks.append(gpx_track)
-        gpx_segment = GPXTrackSegment()
-        gpx_track.segments.append(gpx_segment)
-        for point in geom_json["coordinates"][0]:
-            # Append each point as a GPXWaypoint to the GPXTrackSegment
-            gpx_segment.points.append(GPXWaypoint(point[1], point[0]))
-        gpx.creator = "fAIr Backend"
-        gpx_track.name = f"AOI of id {aoi_id} , Don't Edit this Boundary"
-        gpx_track.description = "This is coming from AI Assisted Mapping - fAIr : HOTOSM , Map inside this boundary and go back to fAIr UI"
-        gpx.time = datetime.now()
-        gpx.link = "https://github.com/hotosm/fAIr"
-        gpx.link_text = "AI Assisted Mapping - fAIr : HOTOSM"
-        return HttpResponse(gpx.to_xml(), content_type="application/xml")
+        gpx_xml=gpx_generator(geom_json)
+        return HttpResponse(gpx_xml, content_type="application/xml")
+
+
+class GenerateFeedbackAOIGpxView(APIView):
+    def get(self, request, feedback_aoi_id: int):
+        aoi = get_object_or_404(FeedbackAOI, id=feedback_aoi_id)
+        # Convert the polygon field to GPX format
+        geom_json = json.loads(aoi.geom.json)
+        # Create a new GPX object
+        gpx_xml=gpx_generator(geom_json)
+        return HttpResponse(gpx_xml, content_type="application/xml")
+
 
 
 class TrainingWorkspaceView(APIView):