from flask import Flask, request, jsonify
import pyarrow.parquet as pq

app = Flask(__name__)

PARQUET_FILE = "mera_data.parquet"

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(),
                    "FIRST_NAME": batch.column("FIRST_NAME")[i].as_py(),
                    "TG_USERNAME": batch.column("TG_USERNAME")[i].as_py(),
                    "PHONE": batch.column("PHONE")[i].as_py(),
                }
    return None

@app.route("/search", methods=["GET"])
def search():
    query = request.args.get("search", "").strip()
    
    if not query:
        return jsonify({
            "status": "error",
            "message": "Search value required!",
            "example": "/search?search=Kudesner77",
            "owner": "Code X",
            "username": "@C4DX1"
        }), 400
    
    result = search_data(query)
    
    if result:
        return jsonify({
            "status": "found",
            "query": query,
            "data": result,
            "owner": "Code X",
            "username": "@C4DX1"
        })
    else:
        return jsonify({
            "status": "not found",
            "query": query,
            "message": "No data found!",
            "owner": "Code X",
            "username": "@C4DX1"
        }), 404

@app.route("/", methods=["GET"])
def home():
    return jsonify({
        "status": "online",
        "name": "Code X API",
        "owner": "Code X",
        "username": "@C4DX1",
        "usage": "/search?search=TG_USERNAME or TG_ID",
        "example": "/search?search=Kudesner77"
    })

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)