Update app.py

app.py

@@ -1,76 +1,36 @@
-from flask import Flask, request
-import pyarrow.parquet as pq
+from fastapi import FastAPI, Query
+from fastapi.middleware.cors import CORSMiddleware
+from datasets import load_dataset
+import os
 
-app = Flask(__name__)
-PARQUET_FILE = "mera_data.parquet"
+os.environ["HF_HOME"] = "/tmp/hf"
 
-def search_data(query):
-    query = str(query).strip().lower()
-    pf = pq.ParquetFile(PARQUET_FILE)
-    for batch in pf.iter_batches(batch_size=10000):
-        for i in range(batch.num_rows):
-            tg_id = str(batch.column("TG_ID")[i].as_py() or "").lower()
-            tg_username = str(batch.column("TG_USERNAME")[i].as_py() or "").lower()
-            if query == tg_id or query == tg_username:
-                return {
-                    "TG_ID": batch.column("TG_ID")[i].as_py(),
-                    "TG_USERNAME": batch.column("TG_USERNAME")[i].as_py(),
-                    "FIRST_NAME": batch.column("FIRST_NAME")[i].as_py(),
-                    "PHONE": batch.column("PHONE")[i].as_py(),
-                }
-    return None
+app = FastAPI()
 
-@app.route("/search", methods=["GET"])
-def search():
-    query = request.args.get("search", "").strip()
-    if not query:
-        r = '{"status":"error","message":"Search value required!","example":"/search?search=Kudesner77"}'
-        return app.response_class(response=r, status=400, mimetype="application/json")
-    result = search_data(query)
-    if result:
-        r = (
-            "{\n"
-            '  "status": "found",\n'
-            f'  "query": "{query}",\n'
-            '  "data": {\n'
-            f'    "TG_ID": "{result["TG_ID"]}",\n'
-            f'    "TG_USERNAME": "{result["TG_USERNAME"]}",\n'
-            f'    "FIRST_NAME": "{result["FIRST_NAME"]}",\n'
-            f'    "PHONE": "{result["PHONE"]}"\n'
-            "  },\n"
-            "\n\n\n\n"
-            '  "owner": "Code X",\n'
-            '  "username": "@C4DX1"\n'
-            "}"
-        )
-        return app.response_class(response=r, status=200, mimetype="application/json")
-    else:
-        r = (
-            "{\n"
-            '  "status": "not found",\n'
-            f'  "query": "{query}",\n'
-            '  "message": "No data found!",\n'
-            "\n\n\n\n"
-            '  "owner": "Code X",\n'
-            '  "username": "@C4DX1"\n'
-            "}"
-        )
-        return app.response_class(response=r, status=404, mimetype="application/json")
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-@app.route("/", methods=["GET"])
-def home():
-    r = (
-        "{\n"
-        '  "status": "online",\n'
-        '  "name": "Code X API",\n'
-        '  "usage": "/search?search=TG_USERNAME or TG_ID",\n'
-        '  "example": "/search?search=Kudesner77",\n'
-        "\n\n\n\n"
-        '  "owner": "Code X",\n'
-        '  "username": "@C4DX1"\n'
-        "}"
-    )
-    return app.response_class(response=r, status=200, mimetype="application/json")
+DATASET = "CodeXDevloper/MergedTgDataset"
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+@app.get("/")
+def root():
+    return {"status": "running", "usage": "/search?uid=385167100"}
+
+@app.get("/search")
+def search_uid(uid: str = Query(...)):
+    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, "data": row}
+            if scanned >= 5000000:
+                break
+        return {"found": False, "uid": uid, "scanned": scanned}
+    except Exception as e:
+        return {"error": str(e)}