Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, Query
 from fastapi.middleware.cors import CORSMiddleware
 from datasets import load_dataset
+import os
 
 app = FastAPI(title="Telegram UID Search")
 
@@ -15,34 +16,14 @@ DATASET = "CodeXDevloper/MergedTgDataset"
 
 @app.get("/")
 def root():
-    return {
-        "status": "running",
-        "usage": "/search?uid=385167100"
-    }
+    return {"status": "running"}
 
 @app.get("/search")
-def search_uid(uid: str = Query(..., description="Telegram UID")):
-    try:
-        ds = load_dataset(DATASET, split="train", streaming=True)
-        
-        scanned = 0
-        for row in ds:
-            scanned += 1
-            if uid == str(row.get("user_id", "")).strip():
-                return {
-                    "found": True,
-                    "uid": uid,
-                    "scanned": scanned,
-                    "data": row
-                }
-            if scanned >= 5000000:  # 50 lakh rows tak
-                break
-        
-        return {
-            "found": False,
-            "uid": uid,
-            "scanned": scanned,
-            "message": f"UID '{uid}' nahi mila"
-        }
-    except Exception as e:
-        return {"error": str(e)}
+def search_uid(uid: str = Query(...)):
+    # Lazy loading — dataset sirf search ke time load hoga
+    ds = load_dataset(DATASET, split="train", streaming=True)
+    
+    for row in ds:
+        if uid == str(row.get("user_id", "")):
+            return {"found": True, "data": row}
+    return {"found": False}