Update app.py

app.py

@@ -25,12 +25,15 @@ def search_uid(uid: str = Query(...)):
     try:
         ds = load_dataset(DATASET, split="train", streaming=True)
         scanned = 0
+        max_scan = 100000  # sirf 1 lakh rows (fast)
+        
         for row in ds:
             scanned += 1
             if uid == str(row.get("user_id", "")).strip():
-                return {"found": True, "uid": uid, "data": row}
-            if scanned >= 5000000:
+                return {"found": True, "uid": uid, "scanned": scanned, "data": row}
+            if scanned >= max_scan:
                 break
+        
         return {"found": False, "uid": uid, "scanned": scanned}
     except Exception as e:
         return {"error": str(e)}