Update app.py

app.py

@@ -82,9 +82,41 @@ def download_road_shapefile(coordinates_upper_left, coordinates_bottom_right):
         
         return name if name else ref if ref else ''
 
+    # Process road names and numbers
+    def process_highway_name(row):
+        # Safely get values, converting NaN to empty string
+        #print(row['name'])
+        #name = row.get('name', '')
+        ref = row.get('ref', '')
+        highway = row.get('highway', '')
+        
+        # Handle lists
+        #if isinstance(name, list): name = name[0] if name else ''
+        if isinstance(ref, list): ref = ref[0] if ref else None
+        if isinstance(highway, list): highway = highway[0] if highway else None
+        
+        # Process interstate highways
+        if highway == 'motorway' and ref:
+            ref = ref.split(';')
+            try:
+                ref = ref[0]
+            except:
+                #print('here', ref[0])
+                ref = ref[0]
+
+            if ref.startswith('I'):
+                #print(ref)
+                #print(f"I-{ref.replace('I', '').strip()}")
+                return f"I-{ref.replace('I', '').strip()}"
+            return ref
+            
+        
+        return ref
+    
+
     gdf_edges = gdf_edges[gdf_edges['name'].notnull()]
     # Apply the processing to create a combined name field
-    gdf_edges['display_name'] = gdf_edges.apply(process_road_name, axis = 1)  
+    gdf_edges['highway'] = gdf_edges.apply(process_road_name, axis = 1)  
     
     # Clean and abbreviate the display names
     gdf_edges['display_name'] = gdf_edges['display_name'].apply(abbreviate_address)
@@ -95,7 +127,7 @@ def download_road_shapefile(coordinates_upper_left, coordinates_bottom_right):
     gdf_edges = gdf_edges[gdf_edges['display_name'].str.len() > 0]
     
     # Select final columns
-    gdf_edges = gdf_edges[['display_name', 'geometry']]
+    gdf_edges = gdf_edges[['display_name', 'highway', 'geometry']]
     gdf_edges = gdf_edges.to_crs(epsg=4326)
     
     # Save the file