Update app.py

app.py

@@ -1,34 +1,48 @@
-import os
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 
-app = FastAPI()
+app = FastAPI(title="Fast Search API")
+
+# DuckDB connection aur httpfs extension load karo
 con = duckdb.connect()
 con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Performance settings
-con.execute("SET memory_limit = '2GB'")
-con.execute("SET enable_progress_bar = false")
-con.execute("PRAGMA threads=4")
+# Public Parquet URL (Isme koi token ya header nahi chahiye)
+PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainonefile/resolve/main/users_data.parquet"
 
-# Sahi path (sorted file ka)
-HF_PATH = "hf://datasets/tfqdeadlo/Inddatainonefile@refs/convert/parquet/default/train/0000.parquet"
+@app.get("/")
+def home():
+    return {"message": "API online hai! Use /search?mobile=YOUR_NUMBER"}
 
 @app.get("/search")
-async def search_mobile(mobile: str = Query(...)):
+async def search_mobile(mobile: str = Query(..., description="Enter mobile number to search")):
     try:
-        # Sirf needed columns + DuckDB ko sorted column hint dena (optional)
+        # Clean SQL Query bina kisi extra headers ke
         query = f"""
-            SELECT mobile, name, fname, address, alt, circle, id, email
-            FROM read_parquet('{HF_PATH}')
-            WHERE mobile = ?
+            SELECT mobile, name, fname, address, alt, circle, id, email 
+            FROM read_parquet('{PARQUET_URL}') 
+            WHERE mobile = ? 
             LIMIT 1
         """
-        # DuckDB apne aap row group skip kar lega kyunki data sorted hai
-        result = con.execute(query, [mobile]).fetchall()
+        
+        result = con.execute(query, [str(mobile)]).fetchall()
+        
         if not result:
-            return {"status": "error", "message": "Not found"}
+            return {
+                "status": "error", 
+                "message": f"Mobile number {mobile} nahi mila."
+            }
+        
+        # Data ko dictionary format me convert karo
         columns = [desc[0] for desc in con.description]
-        return {"status": "success", "data": dict(zip(columns, result[0]))}
+        row_dict = dict(zip(columns, result[0]))
+        
+        return {
+            "status": "success",
+            "data": row_dict
+        }
+        
     except Exception as e:
-        raise HTTPException(500, detail=str(e))
+        # Safe logging: Taki koi internal server details public na hon
+        raise HTTPException(status_code=500, detail="Server query process nahi kar paya.")
+