Update app.py

app.py

@@ -1,62 +1,49 @@
-
-from fastapi import FastAPI, Query
+from fastapi import FastAPI
 import duckdb
-import pandas as pd
-from fastapi.middleware.cors import CORSMiddleware
+from huggingface_hub import hf_hub_download
+import os
+
+app = FastAPI()
 
-app = FastAPI(title="Alien X Search API")
+# --- SECURE CONFIG ---
+# Ye 'HF_TOKEN' settings se auto-fetch hoga
+TOKEN = os.getenv("HF_TOKEN") 
+DATASET_REPO = "Watchhrr/Pakistan-Master-Vault"
 
-# Enable CORS for external access
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# Check agar token sahi se mila
+if not TOKEN:
+    print("❌ ERROR: HF_TOKEN settings mein nahi mila!")
 
-# Database Connection
-db = duckdb.connect('AlienX_Vault.db', read_only=True)
+print("📡 Fetching Master DB from Dataset...")
+try:
+    # Dataset se file download ho rahi hai
+    db_path = hf_hub_download(
+        repo_id=DATASET_REPO, 
+        filename="Pakistan.db", 
+        repo_type="dataset", 
+        token=TOKEN
+    )
+    print("✅ Database Loaded Successfully.")
+except Exception as e:
+    print(f"❌ Error downloading DB: {e}")
+    db_path = None
 
 @app.get("/")
 def home():
-    return {
-        "brand": "Alien X",
-        "status": "Engine Online",
-        "endpoints": {
-            "search_number": "/search?pnum=3333769789",
-            "search_cnic": "/search?cnic=42401XXXXXXXX",
-            "search_name": "/search?n=ahmed"
-        }
-    }
+    return {"status": "Alien X Engine: Pakistan Master Vault is ONLINE"}
 
-@app.get("/search")
-def search(
-    pnum: str = Query(None, description="Search by Phone Number (Last 10 digits)"),
-    cnic: str = Query(None, description="Search by CNIC (13 digits)"),
-    n: str = Query(None, description="Search by Name or Address")
-):
-    try:
-        if pnum:
-            # Suffix match taaki 0 ya 92 ka masla na ho
-            query = f"SELECT * FROM vault WHERE pnum LIKE '%{pnum}'"
-        elif cnic:
-            # Exact CNIC match
-            query = f"SELECT * FROM vault WHERE cnic = '{cnic}'"
-        elif n:
-            # Name ya Address mein search (ILIKE is case-insensitive)
-            query = f"SELECT * FROM vault WHERE name ILIKE '%{n}%' OR address ILIKE '%{n}%' LIMIT 100"
-        else:
-            return {"error": "Kam se kam ek parameter (pnum, cnic, ya n) zaruri hai"}
+@app.get("/search/pnum")
+def search_pnum(number: str):
+    if not db_path: return {"error": "DB not loaded"}
+    con = duckdb.connect(db_path)
+    res = con.execute(f"SELECT * FROM vault WHERE pnum = '{number}'").df().to_dict(orient='records')
+    con.close()
+    return {"count": len(res), "data": res}
 
-        results = db.execute(query).df()
-        
-        # Data cleaning for JSON response
-        results = results.fillna("N/A")
-        
-        return {
-            "brand": "Alien X",
-            "count": len(results),
-            "data": results.to_dict(orient='records')
-        }
-    except Exception as e:
-        return {"error": str(e)}
+@app.get("/search/cnic")
+def search_cnic(id_card: str):
+    if not db_path: return {"error": "DB not loaded"}
+    con = duckdb.connect(db_path)
+    res = con.execute(f"SELECT * FROM vault WHERE cnic = '{id_card}'").df().to_dict(orient='records')
+    con.close()
+    return {"count": len(res), "data": res}