Update app.py

app.py

@@ -1,80 +1,49 @@
 import os
-import tempfile
-import requests
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 
 app = FastAPI(title="Fast Mobile Search API")
 
-# DuckDB connection
+# DuckDB connection with httpfs extension
 con = duckdb.connect()
+con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Hugging Face token environment se
+# Hugging Face token from Space secrets
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
-PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainone/resolve/main/users_data.parquet"
+if not HF_TOKEN:
+    raise Exception("HF_TOKEN environment variable not set. Add it in Space secrets.")
 
-# Global variable to cache the downloaded file path (optional, to avoid re-downloading)
-temp_file_path = None
+# Set HTTP headers for all outgoing httpfs requests
+con.execute(f"SET http_headers = 'Authorization: Bearer {HF_TOKEN}'")
 
-def download_parquet_with_auth():
-    global temp_file_path
-    if temp_file_path and os.path.exists(temp_file_path):
-        return temp_file_path
-    
-    if not HF_TOKEN:
-        raise Exception("HF_TOKEN environment variable not set. Please add your token in Space secrets.")
-    
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    response = requests.get(PARQUET_URL, headers=headers, stream=True)
-    
-    if response.status_code != 200:
-        raise Exception(f"Download failed with status {response.status_code}: {response.text[:200]}")
-    
-    # Save to a temporary file
-    fd, path = tempfile.mkstemp(suffix=".parquet")
-    os.close(fd)
-    with open(path, "wb") as f:
-        for chunk in response.iter_content(chunk_size=8192):
-            f.write(chunk)
-    
-    temp_file_path = path
-    return path
+# Parquet URL (without ?token)
+PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainone/resolve/main/users_data.parquet"
 
 @app.get("/")
 def home():
-    return {"message": "API is running perfectly. Use /search?mobile=YOUR_NUMBER"}
+    return {"message": "API is running. Use /search?mobile=NUMBER"}
 
 @app.get("/search")
-async def search_mobile(mobile: str = Query(..., description="Enter the mobile number to search")):
+async def search_mobile(mobile: str = Query(..., description="Mobile number to search")):
     try:
-        # Download the parquet file (cached after first request)
-        local_path = download_parquet_with_auth()
-        
-        # DuckDB query on local file
+        # DuckDB will read remote parquet directly (no download)
+        # Predicate pushdown: sirf matching rows fetch hongi
         query = f"""
             SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM read_parquet('{local_path}') 
+            FROM read_parquet('{PARQUET_URL}') 
             WHERE mobile = ? 
             LIMIT 1
         """
-        
         result = con.execute(query, [str(mobile)]).fetchall()
         
         if not result:
-            return {
-                "status": "error", 
-                "message": f"Mobile number {mobile} nahi mila."
-            }
+            return {"status": "error", "message": f"Mobile {mobile} nahi mila."}
         
         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": row_dict}
         
     except Exception as e:
-        # Sanitize error message to hide token
-        error_msg = str(e).replace(HF_TOKEN, "[HIDDEN_TOKEN]") if HF_TOKEN else str(e)
-        raise HTTPException(status_code=500, detail=f"Kuch gadbad hui: {error_msg}")
+        # Token hide karo error se
+        error_msg = str(e).replace(HF_TOKEN, "[HIDDEN]")
+        raise HTTPException(status_code=500, detail=f"Error: {error_msg}")