Update app.py

app.py

@@ -8,6 +8,23 @@ import json
 from tavily import TavilyClient
 
 KYS_SAMPLE = "https://kys.udiseplus.gov.in/webapp/api/search-schools?searchType=3&searchParam={udise}"
+
+# List of valid state codes for UDISE (first 2 digits of UDISE code)
+VALID_UDISE_STATE_CODES = [
+    '11', '12', '13', '14', '15', '16', '17', '18', '19',  # Northern
+    '20', '21', '22', '23', '24', '25', '26', '27', '28',  # Western
+    '29', '30', '31', '32', '33', '34', '35', '36',        # Southern
+    '37', '38', '39', '40', '41', '42', '43', '44',        # Eastern
+    '45', '46', '47', '48', '49', '50', '51', '52'         # North-Eastern & UTs
+]
+
+def is_valid_udise(code):
+    """Check if a string is a valid UDISE code."""
+    if not (code and code.isdigit() and len(code) == 11):
+        return False
+    # Optional: Validate state code (first 2 digits)
+    return code[:2] in VALID_UDISE_STATE_CODES
+
 STATES = [
     "Arunachal_pradesh",
     "Assam",
@@ -70,13 +87,16 @@ def extract_udise_candidates_from_search(search_json):
     if not isinstance(results, list):
         return []
     
-    # Patterns to match UDISE codes in text
+    # Compile patterns once for better performance
     patterns = [
-        r'UDISE\s*[Cc]ode\s*:?\s*(\d{11})',  # Matches "UDISE Code: 12345678901"
-        r'Udise\s+School\s+Code\s*:?\s*(\d{11})',  # Matches "Udise School Code: 12345678901"
-        r'\b(\d{11})\b'  # Matches any 11-digit number as fallback
+        r'UDISE\s*[Cc]ode\s*:?\s*(\d{11})(?![0-9])',  # Matches "UDISE Code: 12345678901"
+        r'Udise\s+School\s+Code\s*:?\s*(\d{11})(?![0-9])',  # Matches "Udise School Code: 12345678901"
+        r'(?<![0-9])(\d{11})(?![0-9])'  # Matches exactly 11 digits not part of a longer number
     ]
     
+    # Compile patterns for better performance
+    compiled_patterns = [re.compile(pattern) for pattern in patterns]
+    
     for result in results:
         if not isinstance(result, dict):
             continue
@@ -88,14 +108,13 @@ def extract_udise_candidates_from_search(search_json):
                 continue
                 
             # Try each pattern
-            for pattern in patterns:
-                import re
-                matches = re.findall(pattern, text)
+            for pattern in compiled_patterns:
+                matches = pattern.finditer(text)
                 for match in matches:
-                    if isinstance(match, tuple):  # If pattern has groups
-                        match = match[0]  # Take the first group
-                    if match.isdigit() and len(match) == 11:  # UDISE codes are 11 digits
-                        found.add(match)
+                    # Get the first group if it exists, otherwise the whole match
+                    udise_code = match.group(1) if len(match.groups()) > 0 else match.group(0)
+                    if udise_code and is_valid_udise(udise_code):
+                        found.add(udise_code)
     
     return sorted(list(found))