Update app.py

app.py

@@ -3,35 +3,34 @@ from fastapi import FastAPI, HTTPException, Query
 import os
 
 app = FastAPI()
-
-# --- 🔐 SECURITY & CONFIG ---
 VALID_KEYS = {"Satuu5": "Master_Admin"}
 HF_TOKEN = os.getenv("HF_TOKEN") 
 DATA_URL = "https://huggingface.co/datasets/Watchhrr/HITECH_DB/resolve/main/Hi-Tek-DB.zip.001"
 
-@app.get("/")
-def home():
-    return {"status": "Online", "Owner": "Swapnil", "DB_Type": "Virtual_DuckDB"}
-
 @app.get("/search")
-def search(query: str, type: str = "phone", key: str = None):
+def search(query: str, key: str = None):
     if key not in VALID_KEYS:
-        raise HTTPException(status_code=403, detail="Invalid API Key!")
+        raise HTTPException(status_code=403, detail="Invalid Key!")
 
     con = duckdb.connect()
     try:
-        # Har version ke liye fix configuration
         con.execute("INSTALL httpfs; LOAD httpfs;")
         if HF_TOKEN:
-            # Ye latest DuckDB secret method hai jo kabhi fail nahi hota
             con.execute(f"CREATE OR REPLACE SECRET (TYPE HTTP, KEY_ID 'Authorization', SECRET 'Bearer {HF_TOKEN}');")
         
-        # SQL query jo 100GB ke zip parts ko handle karegi
-        # 'all_varchar=True' taaki phone numbers galat read na hon
-        sql = f"SELECT * FROM read_csv_auto('{DATA_URL}', all_varchar=True) WHERE {type} LIKE '%{query}%' LIMIT 5"
+        # --- SMART SEARCH ---
+        # Ye pehle headers check karega automatically
+        df_sample = con.execute(f"SELECT * FROM read_csv_auto('{DATA_URL}') LIMIT 1").df()
+        columns = df_sample.columns
+        
+        # Phone jaisa dikhne wala column dhoond raha hu
+        target_col = [c for c in columns if 'phone' in c.lower() or 'mobile' in c.lower() or 'number' in c.lower()]
+        search_col = target_col[0] if target_col else columns[0] # Agar kuch na mile toh pehla column
+        
+        sql = f"SELECT * FROM read_csv_auto('{DATA_URL}', all_varchar=True) WHERE \"{search_col}\" LIKE '%{query}%' LIMIT 5"
         result = con.execute(sql).df().to_dict(orient="records")
         
-        return {"status": "success", "data": result}
+        return {"status": "success", "searching_in_column": search_col, "data": result}
     except Exception as e:
-        return {"status": "error", "message": "Database scanning... try again in 30 seconds."}
+        return {"status": "error", "message": "Still loading data... wait 1 minute.", "raw_error": str(e)}