Update main.py

main.py

@@ -1,60 +1,40 @@
 import os
-import requests
 import uvicorn
 import duckdb
 from fastapi import FastAPI, Query
-from contextlib import asynccontextmanager
 
-DB_PATH = "insta_hitech.duckdb"
-DB_URL = "https://huggingface.co/datasets/Watchhrr/Insta17m/resolve/main/insta_hitech.duckdb"
-TOKEN = os.getenv("MY_HF_TOKEN")
+app = FastAPI()
 
-def download_db():
-    if not os.path.exists(DB_PATH):
-        print("📥 Database missing. Starting download...")
-        headers = {"Authorization": f"Bearer {TOKEN}"} if TOKEN else {}
-        
-        response = requests.get(DB_URL, headers=headers, stream=True)
-        total_size = int(response.headers.get('content-length', 0))
-        
-        if response.status_code == 200:
-            with open(DB_PATH, 'wb') as f:
-                downloaded = 0
-                for chunk in response.iter_content(chunk_size=1024*1024): # 1MB chunks
-                    if chunk:
-                        f.write(chunk)
-                        downloaded += len(chunk)
-                        # Har 50MB par log dikhayega
-                        if (downloaded // (50*1024*1024)) > ((downloaded - len(chunk)) // (50*1024*1024)):
-                            done = int(50 * downloaded / total_size)
-                            print(f"⏳ Progress: [{'=' * done}{' ' * (50-done)}] {downloaded/(1024*1024):.1f}MB / {total_size/(1024*1024):.1f}MB")
-            print("✅ Download Complete!")
-        else:
-            print(f"❌ Download Failed! Status: {response.status_code}. Check your Token/Secret.")
-    else:
-        print(f"✅ DB found. Size: {os.path.getsize(DB_PATH)/(1024*1024):.1f} MB")
-
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    download_db()
-    yield
-
-app = FastAPI(lifespan=lifespan)
+# Mount hone ke baad file is path par milegi
+# /data/ aapka dataset folder ban jayega
+DB_PATH = "/data/insta_hitech.duckdb"
 
 @app.get("/search")
 async def search(q: str = Query(None)):
-    if not q: return {"error": "Query 'q' missing"}
-    if not os.path.exists(DB_PATH): return {"error": "DB still downloading..."}
+    if not q:
+        return {"error": "Query parameter 'q' missing"}
+    
+    if not os.path.exists(DB_PATH):
+        return {"error": f"Database not found at {DB_PATH}. Check if Dataset is mounted."}
     
     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"}
-        data = res.rename(columns={'t':'phone', 'u':'username', 'e':'email', 'n':'name', 'id':'user_id'}).to_dict(orient='records')
+        # Smart search across 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'
+        }).to_dict(orient='records')
+        
         return {"status": "success", "data": data}
     finally:
         conn.close()
 
 if __name__ == "__main__":
-    uvicorn.run("main:app", host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)