Update services/map_service.py

services/map_service.py

@@ -1,15 +1,18 @@
 import folium
-from PIL import Image
-import io
+import os
 
 def generate_map(gps_coordinates, cracks):
     if not gps_coordinates or not cracks:
         return None
     
+    # Calculate the average latitude and longitude
     avg_lat = sum(coord[0] for coord in gps_coordinates) / len(gps_coordinates)
     avg_lon = sum(coord[1] for coord in gps_coordinates) / len(gps_coordinates)
+    
+    # Create a Folium map centered at the average GPS coordinates
     m = folium.Map(location=[avg_lat, avg_lon], zoom_start=15)
 
+    # Add markers for each crack
     for coord, crack in zip(gps_coordinates[-len(cracks):], cracks):
         folium.Marker(
             location=coord,
@@ -17,16 +20,9 @@ def generate_map(gps_coordinates, cracks):
             icon=folium.Icon(color='red' if crack['severity'] == 'Severe' else 'orange' if crack['severity'] == 'Moderate' else 'green')
         ).add_to(m)
 
-    # Saving the map to an HTML file (no conversion to image needed here)
+    # Save the map as an HTML file
     map_path_html = "map_temp.html"
     m.save(map_path_html)
 
-    # Convert the HTML map to an image using PIL (without wkhtmltoimage)
-    try:
-        img = Image.open(map_path_html)
-        img_path = "map_temp.png"
-        img.save(img_path)
-        return img_path
-    except Exception as e:
-        print(f"Error converting map to PNG: {str(e)}")
-        return None
+    # Return the HTML map file path
+    return map_path_html