4th tab for street grids

app.py

@@ -8,6 +8,7 @@ import logging
 from inference_tab import get_inference_widgets, run_inference,georefImg
 from annotation_tab import get_annotation_widgets
 from map_tab import get_map_widgets
+from strgrid_tab import get_strgrid_widgets
 
 # setup logging
 logging.basicConfig(level=logging.DEBUG)
@@ -21,7 +22,9 @@ with gr.Blocks() as demo:
     with gr.Tab("Annotation"):
         get_annotation_widgets(selected_tile_state)
     with gr.Tab("Map"): 
-        get_map_widgets(city_name)  
+        get_map_widgets(city_name) 
+    with gr.Tab("Street Grid"):
+        get_strgrid_widgets()
 
 
 

strgrid_tab/__init__.py

@@ -0,0 +1,6 @@
+from .strgrid_setup import get_strgrid_widgets
+from .strgrid_logic import run_strgrid_logic
+
+
+
+__all__ = ["get_strgrid_widgets", "run_strgrid_logic"]

strgrid_tab/strgrid_logic.py

@@ -0,0 +1,61 @@
+import geopandas as gpd
+import osmnx as ox
+from pyproj import Transformer
+from config import OUTPUT_DIR
+import os
+
+
+def extract_osm_from_shp_extent(shp_path):
+    gdf = gpd.read_file(shp_path)
+    gdf = gdf.to_crs(epsg=4326)
+
+    bbox = gdf.total_bounds
+   
+    G = ox.graph_from_bbox(bbox,network_type="all")
+    G_proj = ox.project_graph(G, to_crs="EPSG:3395")
+
+    edges = ox.graph_to_gdfs(
+        G_proj,
+        nodes=False,
+        edges=True,
+        fill_edge_geometry=True
+    )
+
+    edges = edges[["name", "geometry"]]
+    edges = edges[edges["name"].notnull()]
+    edges["name"] = edges["name"].apply(
+        lambda x: x[0] if isinstance(x, list) else x
+    )
+
+    return edges
+
+
+def find_nearest_osm_roads(input_shp,osm_roads):
+    roads = gpd.read_file(input_shp)
+    roads = roads.to_crs(epsg=3395)
+
+    nearest = gpd.sjoin_nearest(
+        roads,
+        osm_roads,
+        how="left",
+        distance_col="distance_m"
+    )
+
+
+    nearest.rename(columns={"name_left":"strgrid_name","name_right":"osm_name"}, inplace=True)
+    nearest = nearest.sort_values("distance_m").drop_duplicates(subset=["strgrid_name","geometry"])
+
+    return nearest
+
+
+def write_nearest_roads_csv(nearest_gdf):
+    OSM_NEAR_PATH = os.path.join(OUTPUT_DIR, "strgrid_osm_nearest.csv")
+    nearest_gdf.to_csv(OSM_NEAR_PATH, index=False)
+    return OSM_NEAR_PATH
+
+
+def run_strgrid_logic(input_shp):
+    osm_roads = extract_osm_from_shp_extent(input_shp)
+    nearest = find_nearest_osm_roads(input_shp,osm_roads)
+    csv_path = write_nearest_roads_csv(nearest)
+    return csv_path

strgrid_tab/strgrid_setup.py

@@ -0,0 +1,53 @@
+import gradio as gr
+from .strgrid_logic import run_strgrid_logic
+
+
+def get_strgrid_widgets():
+    with gr.Row():
+        with gr.Column(scale=1, min_width=500):
+            shp_input = gr.File(
+                label="Upload Zipped Shapefile with existing street grid"
+            )
+
+            run_osm_btn = gr.Button(
+                "Extract Nearest OSM Roads",
+                interactive=False
+            )
+
+            osm_log = gr.Textbox(
+                label="Progress",
+                lines=6,
+                interactive=False
+            )
+
+            osm_download = gr.File(
+                label="Download CSV",
+                file_types=[".csv"],
+                type="filepath"
+            )
+
+       
+
+    def enable_run_btn(shp):
+        return gr.update(interactive=bool(shp))
+
+    shp_input.change(
+        fn=enable_run_btn,
+        inputs=shp_input,
+        outputs=run_osm_btn
+    )
+
+
+    def run_osm_wrapper(shp_file):
+        csv_path = run_strgrid_logic(
+            shp_file.name
+        )
+        return "Find nearest OSM roads is complete.", csv_path
+
+    run_osm_btn.click(
+        fn=run_osm_wrapper,
+        inputs=[shp_input],
+        outputs=[osm_log, osm_download]
+    )
+
+    return shp_input, run_osm_btn, osm_log, osm_download