Update app.py

app.py

@@ -1,24 +1,30 @@
 import os
+import json
 from fastapi import FastAPI, HTTPException, Query
 import duckdb
 
 app = FastAPI(title="Fast Mobile Search API")
 
-# DuckDB connection with httpfs extension
+# DuckDB connection
 con = duckdb.connect()
 con.execute("INSTALL httpfs; LOAD httpfs;")
 
-# Hugging Face token from Space secrets
+# Hugging Face token
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 if not HF_TOKEN:
     raise Exception("HF_TOKEN environment variable not set. Add it in Space secrets.")
 
-# Remote parquet URL
-PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainone/resolve/main/users_data.parquet"
+# Create secret with proper JSON headers
+headers_json = json.dumps({"Authorization": f"Bearer {HF_TOKEN}"})
+con.execute(f"""
+    CREATE OR REPLACE SECRET hf_token (
+        TYPE HTTP,
+        HEADERS '{headers_json}'
+    );
+""")
 
-# DuckDB map syntax for HTTP headers
-# Isse token automatically authorisation header mein bheja jayega
-HEADERS_MAP = f"map({{'Authorization': 'Bearer {HF_TOKEN}'}})"
+# Parquet URL (without ?token)
+PARQUET_URL = "https://huggingface.co/datasets/tfqdeadlo/Inddatainone/resolve/main/users_data.parquet"
 
 @app.get("/")
 def home():
@@ -27,24 +33,22 @@ def home():
 @app.get("/search")
 async def search_mobile(mobile: str = Query(..., description="Mobile number to search")):
     try:
-        # Direct remote parquet read with authentication headers
-        # DuckDB sirf matching rows fetch karega, poori file nahi
+        # Secret automatically adds auth header to any httpfs request
         query = f"""
             SELECT mobile, name, fname, address, alt, circle, id, email 
-            FROM read_parquet('{PARQUET_URL}', http_headers={HEADERS_MAP}) 
+            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}
         
     except Exception as e:
-        # Token ko error message se hata do
-        error_msg = str(e).replace(HF_TOKEN, "[HIDDEN_TOKEN]")
+        error_msg = str(e).replace(HF_TOKEN, "[HIDDEN]")
         raise HTTPException(status_code=500, detail=f"Error: {error_msg}")