import sqlite3, re, os
from flask import Flask, request, jsonify
from huggingface_hub import hf_hub_download

app = Flask(__name__)

DB_PATH = hf_hub_download(
    repo_id="Bruhletme/Paytm",
    filename="paytm.db",
    repo_type="dataset",
    token=os.environ.get("HF_TOKEN")
)

def query_db(q, a=()):
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    cur = conn.cursor()
    cur.execute(q, a)
    r = cur.fetchall()
    conn.close()
    return [dict(x) for x in r]

@app.route("/num")
def num():
    num = (request.args.get("info") or "").strip()
    num = re.sub(r"\D","",num)[-10:]
    if len(num)!=10:
        return jsonify({"error":"invalid"}),400
    d = query_db("SELECT * FROM users WHERE mobile=? LIMIT 1",(num,))
    return jsonify({"found":bool(d),"data":d[0] if d else None})

@app.route("/")
def home():
    return {"status":"ok"}

app.run(host="0.0.0.0", port=7860)