Update app.py

app.py

@@ -1,41 +1,48 @@
-import gradio as gr
+from fastapi import FastAPI, Query
+from fastapi.middleware.cors import CORSMiddleware
 from datasets import load_dataset
 
+app = FastAPI(title="Telegram UID Search")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 DATASET = "CodeXDevloper/MergedTgDataset"
 
-def search_uid(uid):
-    if not uid:
-        return {"error": "UID daalo"}
-    
+@app.get("/")
+def root():
+    return {
+        "status": "running",
+        "usage": "/search?uid=385167100"
+    }
+
+@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,
-                    "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"),
+                    "uid": uid,
+                    "scanned": scanned,
+                    "data": row
                 }
+            if scanned >= 5000000:  # 50 lakh rows tak
+                break
         
-        return {"found": False, "message": f"UID '{uid}' nahi mila"}
+        return {
+            "found": False,
+            "uid": uid,
+            "scanned": scanned,
+            "message": f"UID '{uid}' nahi mila"
+        }
     except Exception as e:
-        return {"error": str(e)}
-
-demo = gr.Interface(
-    fn=search_uid,
-    inputs=gr.Textbox(label="UID daalo", placeholder="385167100"),
-    outputs=gr.JSON(label="Result"),
-    title="Telegram UID Search",
-    description="UID daal ke poore dataset mein search karo"
-)
-
-demo.launch()
+        return {"error": str(e)}