Update app.py

app.py

@@ -4,26 +4,12 @@ import duckdb
 from fastapi import FastAPI, Query, HTTPException
 
 app = FastAPI(
-    title="InayatGaming Dataset API",
-    version="1.0.0"
+    title="InayatGaming Authorized Lookup API",
+    version="2.0.0"
 )
 
 API_KEY = os.getenv("API_KEY")
-
-PARQUET_FILES = [
-    "hf://datasets/Inayatgaming/InayatGaming/part1.parquet",
-    "hf://datasets/Inayatgaming/InayatGaming/part2a.parquet",
-    "hf://datasets/Inayatgaming/InayatGaming/part2b_new.parquet",
-]
-
-# Only fields explicitly approved for API output.
-PUBLIC_FIELDS = [
-    "district",
-    "pincode",
-    "state",
-    "town",
-    "source",
-]
+DB_PATH = os.getenv("DB_PATH", "/data/authorized_index.duckdb")
 
 
 def authenticate(api_key: str):
@@ -44,7 +30,7 @@ def authenticate(api_key: str):
 def root():
     return {
         "success": True,
-        "service": "InayatGaming Dataset API"
+        "service": "InayatGaming Authorized Lookup API"
     }
 
 
@@ -58,52 +44,56 @@ def health():
 @app.get("/lookup")
 def lookup(
     api_key: str = Query(...),
-    number: str = Query(..., min_length=1)
+    number: str = Query(...)
 ):
     authenticate(api_key)
 
-    # Normalize the supplied identifier.
     number = number.strip()
 
-    con = duckdb.connect()
+    if not number:
+        raise HTTPException(
+            status_code=400,
+            detail="number is required"
+        )
+
+    con = duckdb.connect(
+        DB_PATH,
+        read_only=True
+    )
 
     try:
-        query = """
+        row = con.execute(
+            """
             SELECT
                 district,
                 pincode,
                 state,
                 town,
                 source
-            FROM read_parquet(
-                ?,
-                union_by_name=true
-            )
-            WHERE phoneNumber = ?
+            FROM authorized_lookup
+            WHERE lookup_key = ?
             LIMIT 1
-        """
-
-        # Query each source file rather than loading the dataset
-        # into Python memory.
-        for parquet_file in PARQUET_FILES:
-            result = con.execute(
-                query,
-                [parquet_file, number]
-            ).fetchone()
-
-            if result:
-                return {
-                    "success": True,
-                    "match": True,
-                    "data": dict(
-                        zip(PUBLIC_FIELDS, result)
-                    )
-                }
+            """,
+            [number]
+        ).fetchone()
+
+        if row is None:
+            return {
+                "success": True,
+                "match": False,
+                "data": None
+            }
 
         return {
             "success": True,
-            "match": False,
-            "data": None
+            "match": True,
+            "data": {
+                "district": row[0],
+                "pincode": row[1],
+                "state": row[2],
+                "town": row[3],
+                "source": row[4]
+            }
         }
 
     finally: