Update app.py

app.py

@@ -8,7 +8,30 @@ import json
 from tavily import TavilyClient
 
 KYS_SAMPLE = "https://kys.udiseplus.gov.in/webapp/api/search-schools?searchType=3&searchParam={udise}"
-STATES = ["Arunachal Pradesh", "Assam", "Karnataka", "Kerala", "Tamil Nadu", "Maharashtra"]
+states = [
+    "Arunachal_pradesh",
+    "Assam",
+    "Bihar",
+    "Chhattisgarh",
+    "Delhi",
+    "Jharkhand",
+    "Karnataka",
+    "Madhya pradesh",
+    "Manipur",
+    "Meghalaya",
+    "Mizoram",
+    "Nagaland",
+    "Odisha",
+    "Puducherry",
+    "Rajasthan",
+    "Sikkim",
+    "Tamil nadu",
+    "Telangana",
+    "Tripura",
+    "Uttar pradesh",
+    "Uttarakhand"
+]
+
 
 
 def call_kys_by_udise(udise_code):
@@ -66,41 +89,37 @@ def json_to_table(obj):
 
 def to_table_from_kys(kys_json):
     """
-    Convert KYS JSON to a simplified pandas DataFrame
-    showing only a few important fields from content[].
+    Convert KYS JSON wrapper into a simplified pandas DataFrame showing only
+    selected fields from the `content` list.
     """
     try:
-        # Step 1: Extract the list under data.content
-        items = (kys_json.get("data", {}) or {}).get("data", {})
-        content = items.get("content") if isinstance(items, dict) else None
+        content = None
+        if isinstance(kys_json, dict):
+            inner = kys_json.get("data") if kys_json.get("data") is not None else None
+            if isinstance(inner, dict) and isinstance(inner.get("content"), list):
+                content = inner.get("content")
+            elif isinstance(inner, dict) and isinstance(inner.get("data"), dict) and isinstance(inner.get("data").get("content"), list):
+                content = inner.get("data").get("content")
+            elif isinstance(kys_json.get("content"), list):
+                content = kys_json.get("content")
         if not content:
-            # fallback if structure slightly differs
-            content = kys_json.get("content") or []
-
-        if not isinstance(content, list) or not content:
             return pd.DataFrame()
-
-        # Step 2: Filter only required columns
-        filtered = []
-        for rec in content:
-            filtered.append({
-                "School Name": rec.get("schoolName"),
-                "School ID": rec.get("schoolId"),
-                "Pincode": rec.get("pincode"),
-                "State": rec.get("stateName"),
-                "District": rec.get("districtName"),
-                "Management Type": rec.get("schMgmtType")
+        rows = []
+        for r in content:
+            rows.append({
+                "School Name": r.get("schoolName"),
+                "School ID": r.get("schoolId"),
+                "Pincode": r.get("pincode"),
+                "State": r.get("stateName"),
+                "District": r.get("districtName"),
+                "Management Type": r.get("schMgmtType")
             })
-
-        # Step 3: Convert to DataFrame
-        return pd.DataFrame(filtered)
-
+        return pd.DataFrame(rows)
     except Exception as e:
-        print("Error parsing KYS JSON:", e)
+        print("to_table_from_kys error:", e)
         return pd.DataFrame()
 
 
-
 def search_workflow(school_name, state_name, search_key, use_search, use_kys):
     out = {"kys": None, "search": None, "suggestions": []}
     payload_text = f"{school_name or ''} {state_name or ''} UDISE code".strip()
@@ -124,16 +143,18 @@ def search_workflow(school_name, state_name, search_key, use_search, use_kys):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-# Provide your API key in the textbox.
+# Search + KYS Lookup (Hugging Face Space)
+
+This version uses the Search SDK internally. Provide your API key in the textbox.
 
-Enter a school name and select the state; the app calls Search 
+Enter a school name (or UDISE code) and select the state; the app calls Search SDK
 with the combined query `<school> <state> UDISE code`, then optionally calls KYS.
 """
     )
 
     with gr.Row():
-        inp = gr.Textbox(label="School name", placeholder="e.g. GOVT SEC SCHOOL DARLONG", lines=1)
-        state_dropdown = gr.Dropdown(choices=STATES, label="State", value=STATES[0] if STATES else "", interactive=True)
+        inp = gr.Textbox(label="School name or UDISE code", placeholder="e.g. GOVT SEC SCHOOL DARLONG or 12345678901", lines=1)
+        state_dropdown = gr.Dropdown(choices=STATES, label="State", value=STATES[0] if STATES else "", interactive=True, allow_custom_value=True)
 
     search_key = gr.Textbox(label="Search API Key (required)", placeholder="api-key...", lines=1)
     save_key_toggle = gr.Checkbox(value=False, label="Save key in session (keeps key between interactions)")
@@ -145,7 +166,10 @@ with the combined query `<school> <state> UDISE code`, then optionally calls KYS
 
     run = gr.Button("Search")
 
-    output_json = gr.JSON(label="Raw Search Output (JSON)")
+    # By default hide raw JSON outputs; users can toggle visibility with `show_raw_checkbox`
+    show_raw_checkbox = gr.Checkbox(value=False, label="Show raw JSON outputs")
+
+    output_json = gr.JSON(label="Raw Search Output (JSON)", visible=False)
     search_table = gr.DataFrame(headers=None, label="Search results (table)")
 
     gr.Markdown("### UDISE candidates found in Search results")
@@ -153,7 +177,7 @@ with the combined query `<school> <state> UDISE code`, then optionally calls KYS
     udise_input = gr.Textbox(label="UDISE to lookup (editable)", placeholder="Pick a candidate or type a UDISE code...", lines=1)
     lookup_btn = gr.Button("Lookup UDISE (Call KYS)")
 
-    kys_output_json = gr.JSON(label="KYS Raw Output")
+    kys_output_json = gr.JSON(label="KYS Raw Output", visible=False)
     kys_table = gr.DataFrame(headers=None, label="KYS results (table)")
 
     saved_key_state = gr.State("")
@@ -170,9 +194,10 @@ with the combined query `<school> <state> UDISE code`, then optionally calls KYS
         if save_key:
             new_saved_key = key or saved_key
             textbox_value = new_saved_key
-        return res.get("search"), tbl, suggestions, new_saved_key, textbox_value
+        # return raw JSON objects (even if hidden) and the table + suggestions
+        return res.get("search"), tbl, suggestions, new_saved_key, textbox_value, res.get("kys")
 
-    run.click(on_run, inputs=[inp, state_dropdown, search_key, save_key_toggle, use_search, use_kys, saved_key_state], outputs=[output_json, search_table, suggestions_dropdown, saved_key_state, search_key])
+    run.click(on_run, inputs=[inp, state_dropdown, search_key, save_key_toggle, use_search, use_kys, saved_key_state], outputs=[output_json, search_table, suggestions_dropdown, saved_key_state, search_key, kys_output_json])
 
     def on_select_suggestion(choice):
         return choice or ""
@@ -195,6 +220,12 @@ with the combined query `<school> <state> UDISE code`, then optionally calls KYS
 
     clear_key_btn.click(on_clear_key, inputs=[saved_key_state], outputs=[saved_key_state, search_key])
 
+    # Toggle visibility handler for raw JSON outputs
+    def toggle_raw(visible: bool):
+        return gr.update(visible=visible), gr.update(visible=visible)
+
+    show_raw_checkbox.change(toggle_raw, inputs=[show_raw_checkbox], outputs=[output_json, kys_output_json])
+
     gr.Markdown(
         """
 ---