Upload folder using huggingface_hub

src/gradio_app.py

@@ -1,3 +1,11 @@
+"""
+Gradio river network explorer — click directly on the river line (or
+use the slider) to read interpolated elevation and estimated groundwater
+level at that point.
+
+Usage:
+    python app.py --data-root datasets
+"""
 import argparse
 import sys
 from pathlib import Path
@@ -266,13 +274,9 @@ with gr.Blocks(title="River Network Explorer") as demo:
     gr.Markdown("# River Network Explorer")
 
     with gr.Row():
-        view_radio = gr.Radio(options=["Explore", "Network validation"], value="Explore", label="View", interactive=True)
-        basin_radio = gr.Radio(choices=list(BASIN_NAMES.keys()), value=0, label="River", interactive=True,
-                               type="index")
-        # Update labels nicely for radio buttons
-        basin_radio.choices = [(name, idx) for idx, name in BASIN_NAMES.items()]
+        view_radio = gr.Radio(choices=["Explore", "Network validation"], value="Explore", label="View", interactive=True)
+        basin_radio = gr.Radio(choices=[(name, idx) for idx, name in BASIN_NAMES.items()], value=0, label="River", interactive=True)
 
-    # Dynamic containers
     explore_group = gr.Group()
     with explore_group:
         status_md = gr.Markdown("Loading datasets...")
@@ -306,7 +310,6 @@ with gr.Blocks(title="River Network Explorer") as demo:
         val_plot_output = gr.Plot(label="Reach Graph Validation")
         val_snap_caption = gr.Markdown("")
 
-    # --- Event Handlers & Logic ---
     def update_view(view_val, basin_id):
         basin_name = BASIN_NAMES[basin_id]
         color = _PALETTE[basin_id % len(_PALETTE)]
@@ -366,11 +369,9 @@ with gr.Blocks(title="River Network Explorer") as demo:
                 slider: 50
             }
 
-    # Bind view and basin switcher
     view_radio.change(update_view, inputs=[view_radio, basin_radio], outputs=[explore_group, validation_group, status_md, plot_output, slider])
     basin_radio.change(update_view, inputs=[view_radio, basin_radio], outputs=[explore_group, validation_group, status_md, plot_output, slider])
 
-    # Handle slider or map clicks
     def compute_position(basin_id, percent):
         fraction = percent / 100.0
         basin_name = BASIN_NAMES[basin_id]
@@ -379,7 +380,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
         try:
             nodes, edges = get_graph(DATA_ROOT)
         except Exception:
-            return None
+            return [None] * 8
 
         centerlines = get_centerlines(DATA_ROOT, nodes)
         gw_df = get_groundwater(DATA_ROOT)
@@ -387,7 +388,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
 
         fig2 = None
         elev, lat, lon, dist_label, dist_sub = None, 0, 0, "", ""
-        nearest_station, nearest_station_km = "", 0
+        nearest_station = ""
 
         if has_centerline:
             info = centerlines[basin_id]
@@ -396,7 +397,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
             lat, lon = cp.latitude, cp.longitude
             dist_label = f"{cp.distance_from_mouth_km:.2f} km"
             dist_sub = f"from mouth, of {cp.total_length_km:.0f} km total"
-            nearest_station, nearest_station_km = info["gauges"].iloc[(info["gauges"]["centerline_km"] - cp.distance_from_mouth_km).abs().idxmin()]["station_code"], float((info["gauges"]["centerline_km"] - cp.distance_from_mouth_km).abs().min())
+            nearest_station = info["gauges"].iloc[(info["gauges"]["centerline_km"] - cp.distance_from_mouth_km).abs().idxmin()]["station_code"]
             fig2 = build_figure_real(info, basin_name, color, wells_df=gw_df, marker_lon=lon, marker_lat=lat)
         else:
             targets = schematic_click_targets(nodes, edges, basin_id)
@@ -405,7 +406,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
             lat, lon = point.latitude, point.longitude
             dist_label = f"{point.distance_from_upstream_km:.2f} km"
             dist_sub = f"of {point.edge_length_km:.2f} km segment"
-            nearest_station, nearest_station_km = point.upstream_station, fraction * point.edge_length_km
+            nearest_station = point.upstream_station
             x_pos = fraction * (len(targets) - 1)
             fig2 = build_figure_schematic(targets, basin_name, color, marker_x=x_pos, marker_y=elev)
 
@@ -418,7 +419,6 @@ with gr.Blocks(title="River Network Explorer") as demo:
 
         gw_text = f"{gwl:.1f} m" if gwl is not None else (f"No well (nearest {nearest_well_km:.0f}km away)" if nearest_well_km else "No data")
 
-        # Hydrometric plots
         hydro = get_hydrometric(DATA_ROOT)
         fig_wl, fig_disc, fig_rc = None, None, None
         if hydro is not None:
@@ -438,17 +438,16 @@ with gr.Blocks(title="River Network Explorer") as demo:
                 except Exception:
                     pass
 
-        return (
+        return [
             fig2, elev, dist_label, gw_text, f"Coordinates: {lat:.4f}, {lon:.4f}",
             fig_wl, fig_disc, fig_rc
-        )
+        ]
 
     slider.change(compute_position, inputs=[basin_radio, slider], outputs=[plot_output, metric_elev, metric_dist, metric_gw, coords_md, plot_waterlevel, plot_discharge, plot_rating])
 
     def on_plot_select(basin_id, select_data: gr.SelectData):
         if not select_data or select_data.index is None:
             return gr.skip()
-        # Customdata carries distance or fraction
         try:
             point_idx = select_data.index
             nodes, edges = get_graph(DATA_ROOT)
@@ -470,8 +469,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
 
     plot_output.select(on_plot_select, inputs=[basin_radio], outputs=[slider])
 
-    # Initial load trigger
     demo.load(update_view, inputs=[view_radio, basin_radio], outputs=[explore_group, validation_group, status_md, plot_output, slider])
 
-if __name__ == "__main__":
-    demo.launch()
+# if __name__ == "__main__":
+demo.launch()