Update app.py

app.py

@@ -1,62 +1,68 @@
 import os
-import sqlite3
+import duckdb
+import uvicorn
 from fastapi import FastAPI
 from huggingface_hub import hf_hub_download
 
 app = FastAPI(title="Telegram API")
 
-DB_FILE = None
+DB_PATH = None
 
 def get_db():
-    global DB_FILE
+    global DB_PATH
 
-    if DB_FILE and os.path.exists(DB_FILE):
-        return DB_FILE
+    if DB_PATH is None:
+        DB_PATH = hf_hub_download(
+            repo_id="fdsgfuisfydgiuef/Telegram",
+            repo_type="dataset",
+            filename="Midnight_Master_v2.db",
+            token=os.getenv("HF_TOKEN")
+        )
 
-    DB_FILE = hf_hub_download(
-        repo_id="fdsgfuisfydgiuef/Telegram",
-        repo_type="dataset",
-        filename="Midnight_Master_v2.db",
-        token=os.getenv("HF_TOKEN")
-    )
-
-    return DB_FILE
+    return DB_PATH
 
 
 @app.get("/")
 def home():
     return {
         "status": True,
-        "message": "API Running"
+        "message": "Telegram API Running"
     }
 
 
-@app.get("/tables")
-def tables():
-    db = get_db()
-
-    conn = sqlite3.connect(db)
-    cur = conn.cursor()
-
-    cur.execute("SELECT name FROM sqlite_master WHERE type='table'")
-    tables = [x[0] for x in cur.fetchall()]
+@app.get("/search")
+def search(q: str):
 
-    conn.close()
+    db = get_db()
 
-    return {"tables": tables}
+    con = duckdb.connect()
 
+    con.execute("INSTALL sqlite;")
+    con.execute("LOAD sqlite;")
 
-@app.get("/columns/{table}")
-def columns(table: str):
-    db = get_db()
+    try:
+        result = con.execute(f"""
+            SELECT *
+            FROM sqlite_scan('{db}','telegram_users')
+            WHERE user_id LIKE '{q}%'
+            LIMIT 20
+        """).fetchall()
 
-    conn = sqlite3.connect(db)
-    cur = conn.cursor()
+        return {
+            "status": True,
+            "count": len(result),
+            "results": result
+        }
 
-    cur.execute(f"PRAGMA table_info({table})")
+    except Exception as e:
+        return {
+            "status": False,
+            "error": str(e)
+        }
 
-    cols = cur.fetchall()
+    finally:
+        con.close()
 
-    conn.close()
 
-    return cols
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)