Update app.py

app.py

@@ -2,32 +2,33 @@ import os
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 
-app = FastAPI(title="Fast Mobile Search API")
-
+app = FastAPI()
 con = duckdb.connect()
 con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Direct URL - bina token ke (public dataset)
-PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainonefile/resolve/main/users_data.parquet"
+# Performance settings
+con.execute("SET memory_limit = '2GB'")
+con.execute("SET enable_progress_bar = false")
+con.execute("PRAGMA threads=4")
 
-@app.get("/")
-def home():
-    return {"message": "API is running. Use /search?mobile=NUMBER"}
+# Sahi path (sorted file ka)
+HF_PATH = "hf://datasets/tfqdeadlo/Inddatainonefile@refs/convert/parquet/default/train/0000.parquet"
 
 @app.get("/search")
 async def search_mobile(mobile: str = Query(...)):
     try:
+        # Sirf needed columns + DuckDB ko sorted column hint dena (optional)
         query = f"""
-            SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM read_parquet('{PARQUET_URL}') 
-            WHERE mobile = ? 
+            SELECT mobile, name, fname, address, alt, circle, id, email
+            FROM read_parquet('{HF_PATH}')
+            WHERE mobile = ?
             LIMIT 1
         """
-        result = con.execute(query, [str(mobile)]).fetchall()
+        # DuckDB apne aap row group skip kar lega kyunki data sorted hai
+        result = con.execute(query, [mobile]).fetchall()
         if not result:
-            return {"status": "error", "message": f"Mobile {mobile} nahi mila."}
+            return {"status": "error", "message": "Not found"}
         columns = [desc[0] for desc in con.description]
-        row_dict = dict(zip(columns, result[0]))
-        return {"status": "success", "data": row_dict}
+        return {"status": "success", "data": dict(zip(columns, result[0]))}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(500, detail=str(e))