Update app.py

app.py

@@ -2,14 +2,14 @@ import os
 import pandas as pd
 import gradio as gr
 from huggingface_hub import hf_hub_download
+import json
+from functools import lru_cache
 
 # ----------------------------------------------------
 # Imports
 # ----------------------------------------------------
 from searchschool import (
     DEFAULT_STATE_KEY,
-    load_master_for_state,
-    update_blocks,
     search_candidates,
     on_search_web,
 )
@@ -28,16 +28,20 @@ if not ADMIN_PASSWORD:
 # ----------------------------------------------------
 # Load states ONCE
 # ----------------------------------------------------
-def load_state_choices_once():
+@lru_cache(maxsize=1)
+def load_state_hierarchy():
     path = hf_hub_download(
         repo_id=HF_SCHOOLS_DATASET,
         repo_type="dataset",
-        filename=MASTER_ALL_STATES_FILE,
+        filename="state_district_block.json",
     )
-    df = pd.read_csv(path, dtype=str)
-    return sorted(df[MASTER_STATE_COL].dropna().str.upper().unique().tolist())
+    with open(path, "r", encoding="utf-8") as f:
+        return json.load(f)
+
+STATE_HIER = load_state_hierarchy()
+STATE_CHOICES = sorted(STATE_HIER.keys())
+
 
-STATE_CHOICES = load_state_choices_once()
 
 # ----------------------------------------------------
 # UI RENDER HELPERS
@@ -82,6 +86,27 @@ def render_results_md(df, show_score=True):
     return md
 
 
+def on_state_change(state):
+    if not state:
+        return (
+            gr.update(choices=[], value=None),
+            gr.update(choices=[], value=None),
+        )
+
+    districts = sorted(STATE_HIER.get(state, {}).keys())
+
+    return (
+        gr.update(choices=districts, value=None),
+        gr.update(choices=[], value=None),
+    )
+
+def on_district_change(state, district):
+    if not state or not district:
+        return gr.update(choices=[], value=None)
+
+    blocks = STATE_HIER.get(state, {}).get(district, [])
+
+    return gr.update(choices=blocks, value=None)
 
 
 
@@ -163,17 +188,18 @@ with gr.Blocks(title="State School Fuzzy Matcher") as demo:
 
     # ================= Wiring =================
     state_dd.change(
-        fn=load_master_for_state,
+        fn=on_state_change,
         inputs=state_dd,
         outputs=[district_dd, block_dd],
     )
-
+    
     district_dd.change(
-        fn=update_blocks,
-        inputs=district_dd,
+        fn=on_district_change,
+        inputs=[state_dd, district_dd],
         outputs=block_dd,
     )
 
+
     search_btn.click(
     fn=rapid_search_and_render,
     inputs=[school_input, state_dd, district_dd, block_dd],