Update app.py

app.py

@@ -2,10 +2,11 @@ from fastapi import FastAPI, Query
 from fastapi.middleware.cors import CORSMiddleware
 from datasets import load_dataset
 import os
+import time
 
 os.environ["HF_HOME"] = "/tmp/hf"
 
-app = FastAPI()
+app = FastAPI(title="Telegram UID Search API")
 
 app.add_middleware(
     CORSMiddleware,
@@ -18,22 +19,63 @@ DATASET = "CodeXDevloper/MergedTgDataset"
 
 @app.get("/")
 def root():
-    return {"status": "running", "usage": "/search?uid=385167100"}
+    return {
+        "status": "running",
+        "dataset": DATASET,
+        "usage": "/search?uid=385167100"
+    }
 
 @app.get("/search")
-def search_uid(uid: str = Query(...)):
+def search_uid(uid: str = Query(..., description="Telegram UID")):
+    """Poore dataset mein UID dhoondho (25 second timeout ke saath)"""
     try:
         ds = load_dataset(DATASET, split="train", streaming=True)
+        
+        start_time = time.time()
         scanned = 0
-        max_scan = 100000  # sirf 1 lakh rows (fast)
+        max_seconds = 25  # HF Space timeout se pehle ruk jao
         
         for row in ds:
             scanned += 1
+            
+            # UID match karo
             if uid == str(row.get("user_id", "")).strip():
-                return {"found": True, "uid": uid, "scanned": scanned, "data": row}
-            if scanned >= max_scan:
-                break
+                return {
+                    "found": True,
+                    "uid": uid,
+                    "scanned": scanned,
+                    "data": {
+                        "user_id": row.get("user_id"),
+                        "username": row.get("username"),
+                        "first_name": row.get("first_name"),
+                        "last_name": row.get("last_name"),
+                        "phone": row.get("phone"),
+                        "email": row.get("email"),
+                        "status": row.get("status"),
+                        "linked_id": row.get("linked_id"),
+                        "linked_name": row.get("linked_name"),
+                        "linked_handle": row.get("linked_handle"),
+                    }
+                }
+            
+            # Har 5000 rows ke baad time check
+            if scanned % 5000 == 0:
+                if time.time() - start_time > max_seconds:
+                    return {
+                        "found": False,
+                        "uid": uid,
+                        "scanned": scanned,
+                        "time_taken": round(time.time() - start_time, 1),
+                        "message": f"{scanned} rows scan kiye, UID nahi mila. Phir try karo."
+                    }
         
-        return {"found": False, "uid": uid, "scanned": scanned}
+        return {
+            "found": False,
+            "uid": uid,
+            "scanned": scanned,
+            "time_taken": round(time.time() - start_time, 1),
+            "message": "Poore dataset mein nahi mila"
+        }
+    
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": str(e), "uid": uid}