Update web_search.py

web_search.py

@@ -45,15 +45,13 @@ _UDISE_RE = re.compile(r"(?<!\d)(\d{11})(?!\d)")
 
 def _build_query(school_name: Optional[str], state_name: Optional[str], district: Optional[str]) -> str:
     parts = []
+    parts.append("UDISE code of")
     if school_name:
-        parts.append(school_name.strip())
-    if district:
-        parts.append(f"district {district.strip()}")
+        parts.append(f"School {school_name.strip()}")
     if state_name:
-        parts.append(f"state {state_name.strip()}")
-    # bias to India and ask specifically for UDISE mentions
-    parts.append("India")
-    parts.append("UDISE code OR UDISE number OR Udise")
+        parts.append(f"in state {state_name.strip()}")
+    if district:
+        parts.append(f"of district {district.strip()}")
     return " ".join(parts).strip()
 
 
@@ -66,7 +64,7 @@ def _call_tavily(api_key: Optional[str], query: str):
     try:
         client = TavilyClient(key)
         # SDK may return complex object; keep raw
-        resp = client.search(query=query)
+        resp = client.search(query=query, country="india")
         return {"ok": True, "data": resp}
     except Exception as e:
         return {"ok": False, "error": str(e)}
@@ -108,6 +106,8 @@ def tavily_search_codes(
     Returns:
         List[str] of unique UDISE codes in the order found.
     """
+    if not school_name:
+        return None
     query = _build_query(school_name, state_name, district)
     call = _call_tavily(api_key, query)
     if not call.get("ok"):