Create app.py

app.py

@@ -0,0 +1,62 @@
+import os
+import duckdb
+import uvicorn
+from fastapi import FastAPI
+from huggingface_hub import hf_hub_download
+
+app = FastAPI()
+
+# 🚀 Global Path taaki baar-baar download na ho
+DB_PATH = None
+
+def get_db():
+    global DB_PATH
+    if DB_PATH is None or not os.path.exists(DB_PATH):
+        token = os.getenv("HF_TOKEN")
+        if token is None:
+            raise ValueError("HF_TOKEN environment variable not set")
+        DB_PATH = hf_hub_download(
+            repo_id="sheeturt/Telegram",
+            filename="Midnight_Master_v2.db",
+            repo_type="dataset",
+            token=token
+        )
+    return DB_PATH
+
+@app.get("/search")
+@app.post("/search")
+async def search(q: str = None):
+    if not q:
+        return {"status": "error", "message": "Query 'q' missing"}
+
+    try:
+        path = get_db()
+        con = duckdb.connect(database=':memory:')
+        con.execute("INSTALL sqlite;")
+        con.execute("LOAD sqlite;")
+
+        query = f"""
+            SELECT user_id
+            FROM sqlite_scan('{path}', 'telegram_users')
+            WHERE user_id LIKE '{q};%'
+            LIMIT 1
+        """
+        res = con.execute(query).fetchone()
+
+        results = []
+        if res:
+            p = res[0].split(';')
+            results.append({
+                "id": p[0],
+                "t": p[1] if len(p) > 1 else "N/A",
+                "u": p[2] if len(p) > 2 else "N/A",
+                "n": p[3] if len(p) > 3 else "N/A"
+            })
+
+        return {"status": "success", "results": results}
+
+    except Exception as e:
+        return {"status": "error", "details": str(e), "type": type(e).__name__}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)