Update services/map_service.py

services/map_service.py

@@ -7,7 +7,7 @@ def generate_map(gps_coordinates: List[List[float]], map_items: List[Dict[str, A
     Generate a map showing issue locations based on GPS coordinates.
     Args:
         gps_coordinates: List of [latitude, longitude] coordinates.
-        map_items: List of detected items (cracks, holes, missing patches) to plot.
+        map_items: List of detected items (cracks, holes, potholes, missing patches) to plot.
     Returns:
         Path to the generated map image, or None if generation fails.
     """
@@ -16,21 +16,36 @@ def generate_map(gps_coordinates: List[List[float]], map_items: List[Dict[str, A
         
         if gps_coordinates:
             lats, longs = zip(*gps_coordinates)
-            ax.plot(longs, lats, 'b-', label='Path')
+            ax.plot(longs, lats, 'b-', label='Path')  # Blue line for path
         
         for item in map_items:
             gps = item.get('gps', [0, 0])
             lat, lon = gps
             # Use a default severity if the key is missing
             severity = item.get('severity', 'Moderate')  # Default to 'Moderate'
-            color = 'darkred' if severity == 'Severe' else 'yellow' if severity == 'Moderate' else 'darkgreen'
-            marker = 'x' if item.get('type') == 'crack' else 'o' if item.get('type') == 'hole' else '^'
-            ax.scatter(lon, lat, c=color, marker=marker, label=item.get('type', 'Issue'), s=100)
+            dtype = item.get('type', '')
+            
+            # Custom colors for holes as per requirement
+            if dtype in ['hole', 'pothole']:
+                if severity == 'Mild':
+                    color = 'darkgreen'  # Green dot for minor holes
+                elif severity == 'Moderate':
+                    color = 'yellow'     # Yellow dot for moderate holes
+                else:
+                    color = 'darkred'    # Severe holes remain dark red
+            else:
+                # Existing logic for other types
+                color = 'darkred' if severity == 'Severe' else 'yellow' if severity == 'Moderate' else 'darkgreen'
+            
+            marker = 'x' if dtype == 'crack' else 'o' if dtype in ['hole', 'pothole'] else '^'
+            ax.scatter(lon, lat, c=color, marker=marker, label=dtype.capitalize(), s=100)
         
         ax.set_xlabel('Longitude')
         ax.set_ylabel('Latitude')
         ax.set_title('Issue Locations')
-        ax.legend()
+        handles, labels = ax.get_legend_handles_labels()
+        by_label = dict(zip(labels, handles))
+        ax.legend(by_label.values(), by_label.keys())
         ax.grid(True)
         
         map_path = "map_temp.png"