Update app.py

app.py

@@ -1,22 +1,44 @@
 import os
+import tempfile
+import requests
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 
 app = FastAPI(title="Fast Mobile Search API")
 
-# DuckDB connection setup aur httpfs extension load karo
+# DuckDB connection
 con = duckdb.connect()
-con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Hugging Face Space ke Secret se token nikalna
+# Hugging Face token environment se
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
-
-# Base Parquet URL
 PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainone/resolve/main/users_data.parquet"
 
-# Agar token maujood hai, toh URL ke sath authenticated request ke liye append karo
-if HF_TOKEN:
-    PARQUET_URL += f"?token={HF_TOKEN}"
+# Global variable to cache the downloaded file path (optional, to avoid re-downloading)
+temp_file_path = None
+
+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
 
 @app.get("/")
 def home():
@@ -25,15 +47,17 @@ def home():
 @app.get("/search")
 async def search_mobile(mobile: str = Query(..., description="Enter the mobile number to search")):
     try:
-        # DuckDB query jo seedhe authenticated URL se data fetch karegi
+        # Download the parquet file (cached after first request)
+        local_path = download_parquet_with_auth()
+        
+        # DuckDB query on local file
         query = f"""
             SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM read_parquet('{PARQUET_URL}') 
+            FROM read_parquet('{local_path}') 
             WHERE mobile = ? 
             LIMIT 1
         """
         
-        # Query execute karo aur result fetch karo
         result = con.execute(query, [str(mobile)]).fetchall()
         
         if not result:
@@ -42,7 +66,6 @@ async def search_mobile(mobile: str = Query(..., description="Enter the mobile n
                 "message": f"Mobile number {mobile} nahi mila."
             }
         
-        # Columns names nikal kar dictionary format me data convert karo
         columns = [desc[0] for desc in con.description]
         row_dict = dict(zip(columns, result[0]))
         
@@ -52,5 +75,6 @@ async def search_mobile(mobile: str = Query(..., description="Enter the mobile n
         }
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Kuch gadbad hui: {str(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}")