Upload folder using huggingface_hub

src/gradio_app.py

@@ -312,6 +312,7 @@ with gr.Blocks(title="River Network Explorer") as demo:
     def update_view(view_val, basin_id):
         basin_name = BASIN_NAMES[basin_id]
         color = _PALETTE[basin_id % len(_PALETTE)]
+        default_fig = go.Figure()
         
         try:
             nodes, edges = get_graph(DATA_ROOT)
@@ -319,7 +320,16 @@ with gr.Blocks(title="River Network Explorer") as demo:
             return {
                 explore_group: gr.update(visible=True),
                 validation_group: gr.update(visible=False),
-                status_md: gr.update(value="❌ Could not find station_elevations.csv under data root.")
+                status_md: "❌ Could not find station_elevations.csv under data root.",
+                plot_output: default_fig,
+                slider: 50,
+                metric_elev: 0.0,
+                metric_dist: "",
+                metric_gw: "No data",
+                coords_md: "Coordinates: —",
+                plot_waterlevel: default_fig,
+                plot_discharge: default_fig,
+                plot_rating: default_fig
             }
 
         if view_val == "Network validation":
@@ -328,7 +338,8 @@ with gr.Blocks(title="River Network Explorer") as demo:
                 return {
                     explore_group: gr.update(visible=False),
                     validation_group: gr.update(visible=True),
-                    val_caption: gr.update(value=f"No reach graph found for {basin_name}. Run `python -m scripts.build_reach_graphs` first.")
+                    val_caption: f"No reach graph found for {basin_name}. Run `python -m scripts.build_reach_graphs` first.",
+                    val_plot_output: default_fig
                 }
             reach_nodes, reach_edges = reach
             n_confluences = int(reach_nodes["is_confluence"].sum()) if "is_confluence" in reach_nodes.columns else 0
@@ -343,8 +354,8 @@ with gr.Blocks(title="River Network Explorer") as demo:
                 m_confl: n_confluences,
                 m_gauges: n_gauged,
                 val_plot_output: fig,
-                val_caption: gr.update(value="Confirm visually: confluences (◆) should sit where a tributary joins."),
-                val_snap_caption: gr.update(value="")
+                val_caption: "Confirm visually: confluences (◆) should sit where a tributary joins.",
+                val_snap_caption: ""
             }
         else:
             centerlines = get_centerlines(DATA_ROOT, nodes)
@@ -363,48 +374,48 @@ with gr.Blocks(title="River Network Explorer") as demo:
             return {
                 explore_group: gr.update(visible=True),
                 validation_group: gr.update(visible=False),
-                status_md: gr.update(value=status_text),
+                status_md: status_text,
                 plot_output: fig,
                 slider: 50
             }
 
-    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])
-
     def compute_position(basin_id, percent):
         fraction = percent / 100.0
         basin_name = BASIN_NAMES[basin_id]
         color = _PALETTE[basin_id % len(_PALETTE)]
+        default_fig = go.Figure()
         
         try:
             nodes, edges = get_graph(DATA_ROOT)
         except Exception:
-            return [None] * 8
+            return {
+                plot_output: default_fig, metric_elev: 0.0, metric_dist: "", 
+                metric_gw: "No data", coords_md: "Coordinates: —", 
+                plot_waterlevel: default_fig, plot_discharge: default_fig, plot_rating: default_fig
+            }
 
         centerlines = get_centerlines(DATA_ROOT, nodes)
         gw_df = get_groundwater(DATA_ROOT)
         has_centerline = basin_id in centerlines
 
-        fig2 = None
-        elev, lat, lon, dist_label, dist_sub = None, 0, 0, "", ""
+        fig2 = default_fig
+        elev, lat, lon, dist_label = 0.0, 0.0, 0.0, ""
         nearest_station = ""
 
         if has_centerline:
             info = centerlines[basin_id]
             cp = interpolate_by_fraction(info["centerline"], fraction)
-            elev = elevation_at_km(info["gauges"], cp.distance_from_mouth_km)
+            elev = float(elevation_at_km(info["gauges"], cp.distance_from_mouth_km))
             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"
+            dist_label = f"{cp.distance_from_mouth_km:.2f} km from mouth"
             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)
             point = interpolate_along_chain(nodes, edges, basin_id, fraction)
-            elev = point.elevation_m
+            elev = float(point.elevation_m)
             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"
+            dist_label = f"{point.distance_from_upstream_km:.2f} km from upstream"
             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)
@@ -412,14 +423,14 @@ with gr.Blocks(title="River Network Explorer") as demo:
         gwl, nearest_well_km = None, None
         if gw_df is not None:
             pseudo_point = RiverPoint(basin_id=basin_id, upstream_station="", downstream_station="",
-                                       fraction=0, latitude=lat, longitude=lon, elevation_m=elev or 0,
+                                       fraction=0, latitude=lat, longitude=lon, elevation_m=elev,
                                        distance_from_upstream_km=0, edge_length_km=0)
             gwl, nearest_well_km = groundwater_at_point(pseudo_point, gw_df)
 
         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")
 
         hydro = get_hydrometric(DATA_ROOT)
-        fig_wl, fig_disc, fig_rc = None, None, None
+        fig_wl, fig_disc, fig_rc = default_fig, default_fig, default_fig
         if hydro is not None:
             loader, hydro_df = hydro
             station_df = hydro_df[hydro_df["station_code"] == nearest_station]
@@ -437,14 +448,31 @@ with gr.Blocks(title="River Network Explorer") as demo:
                 except Exception:
                     pass
 
-        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])
-
-    demo.load(update_view, inputs=[view_radio, basin_radio], outputs=[explore_group, validation_group, status_md, plot_output, slider])
-
-# if __name__ == "__main__":
-demo.launch()
+        return {
+            plot_output: fig2,
+            metric_elev: elev,
+            metric_dist: dist_label,
+            metric_gw: gw_text,
+            coords_md: f"Coordinates: {lat:.4f}, {lon:.4f}",
+            plot_waterlevel: fig_wl,
+            plot_discharge: fig_disc,
+            plot_rating: fig_rc
+        }
+
+    all_outputs = [
+        explore_group, validation_group, status_md, plot_output, slider,
+        metric_elev, metric_dist, metric_gw, coords_md, 
+        plot_waterlevel, plot_discharge, plot_rating,
+        m_nodes, m_edges, m_confl, m_gauges, val_caption, val_plot_output, val_snap_caption
+    ]
+
+    view_radio.change(update_view, inputs=[view_radio, basin_radio], outputs=all_outputs)
+    basin_radio.change(update_view, inputs=[view_radio, basin_radio], outputs=all_outputs)
+
+    slider_outputs = [plot_output, metric_elev, metric_dist, metric_gw, coords_md, plot_waterlevel, plot_discharge, plot_rating]
+    slider.change(compute_position, inputs=[basin_radio, slider], outputs=slider_outputs)
+
+    demo.load(update_view, inputs=[view_radio, basin_radio], outputs=all_outputs)
+
+if __name__ == "__main__":
+    demo.launch()