Update app.py

app.py

@@ -65,17 +65,42 @@ 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[].
+    """
     try:
-        items = kys_json.get("data") if isinstance(kys_json, dict) else kys_json
-        if isinstance(items, dict) and "data" in items:
-            items = items["data"]
-        if not items:
+        # 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
+        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()
-        return pd.json_normalize(items)
-    except Exception:
+
+        # 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")
+            })
+
+        # Step 3: Convert to DataFrame
+        return pd.DataFrame(filtered)
+
+    except Exception as e:
+        print("Error parsing KYS JSON:", 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()