Update main.py

main.py

@@ -1,28 +1,74 @@
 import os
+import requests
+import uvicorn
 import duckdb
 from fastapi import FastAPI, Query
+from contextlib import asynccontextmanager
 
-app = FastAPI()
 DB_PATH = "insta_hitech.duckdb"
+DB_URL = "https://huggingface.co/datasets/Watchhrr/Insta17m/resolve/main/insta_hitech.duckdb"
 
-@app.on_event("startup")
-def check_db():
-    if os.path.exists(DB_PATH):
-        print(f"✅ Database found! Size: {os.path.getsize(DB_PATH)} bytes")
+# Yahan humne secret variable ka naam use kiya hai jo aapne Settings mein dala tha
+TOKEN = os.getenv("MY_HF_TOKEN") 
+
+# 1. Download Function
+def download_db():
+    if not os.path.exists(DB_PATH):
+        print("📥 Database missing. Downloading from Dataset...")
+        if not TOKEN:
+            print("❌ Error: MY_HF_TOKEN not found in Secrets!")
+            return
+            
+        headers = {"Authorization": f"Bearer {TOKEN}"}
+        try:
+            with requests.get(DB_URL, headers=headers, stream=True) as r:
+                r.raise_for_status()
+                with open(DB_PATH, 'wb') as f:
+                    for chunk in r.iter_content(chunk_size=8192):
+                        f.write(chunk)
+            print("✅ Download Complete!")
+        except Exception as e:
+            print(f"❌ Download Failed: {e}")
     else:
-        print("❌ Database NOT found at startup!")
+        print(f"✅ Database found! Size: {os.path.getsize(DB_PATH)} bytes")
+
+# 2. Lifespan Handler
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    download_db()
+    yield
+    print("Shutting down...")
+
+app = FastAPI(lifespan=lifespan)
 
 @app.get("/search")
 async def search(q: str = Query(None)):
-    if not q: return {"error": "Query missing"}
+    if not q:
+        return {"error": "Query parameter 'q' is missing"}
     
+    if not os.path.exists(DB_PATH):
+        return {"error": "Database file not ready yet. Please wait a few minutes."}
+
     conn = duckdb.connect(DB_PATH, read_only=True)
     try:
-        res = conn.execute("SELECT * FROM users WHERE u = ? OR id = ? OR t = ?", [q, q, q]).df()
-        if res.empty: return {"status": "error", "message": "No Record Found"}
+        # Smart search: username (u), id, and phone (t)
+        query = "SELECT * FROM users WHERE u = ? OR id = ? OR t = ?"
+        res = conn.execute(query, [q, q, q]).df()
+        
+        if res.empty:
+            return {"status": "error", "message": "No Record Found"}
+        
+        data = res.rename(columns={
+            't':'phone', 'u':'username', 'e':'email', 
+            'n':'name', 'id':'user_id', 'a':'extra'
+        }).to_dict(orient='records')
         
-        data = res.rename(columns={'t':'phone', 'u':'username', 'e':'email', 'n':'name', 'id':'user_id'}).to_dict(orient='records')
         return {"status": "success", "data": data}
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
     finally:
         conn.close()
-        
+
+if __name__ == "__main__":
+    uvicorn.run("main:app", host="0.0.0.0", port=7860)
+