from fastapi import FastAPI
import duckdb

app = FastAPI()

HF_PATH = "hf://datasets/tfqdeadlo/Inddata"

@app.get("/")
def home():
    return {"status": "API running"}

@app.get("/search")
def search(mobile: str):
    try:
        prefix = mobile[:3]

        query = f"""
        SELECT *
        FROM read_parquet('{HF_PATH}/{prefix}/*.parquet')
        WHERE mobile = '{mobile}'
        LIMIT 10
        """

        df = duckdb.query(query).to_df()

        return {
            "found": len(df),
            "data": df.to_dict(orient="records")
        }

    except Exception as e:
        return {"error": str(e)}