from fastapi import FastAPI
import duckdb

app = FastAPI()

DATASET_PATH = "hf://datasets/CutehackX/hitek-data-bucket/*.parquet"

@app.get("/")
def home():
    return {"message": "API is running! URL ke aage /search/MOBILE_NUMBER lagayein."}

@app.get("/search/{mobile_number}")
def search_data(mobile_number: str):
    query = f"SELECT * FROM '{DATASET_PATH}' WHERE mobile = '{mobile_number}' LIMIT 1"
    try:
        result = duckdb.query(query).to_df().to_dict(orient="records")
        if result:
            data = result[0]
            return {
                "status": "success",
                "result": {
                    "Mobile Number": data.get("mobile", "N/A"),
                    "Full Name": data.get("name", "N/A"),
                    "Father Name": data.get("fname", "N/A"),
                    "Address": data.get("address", "N/A"),
                    "Circle/State": data.get("circle", "N/A"),
                    "Alt Number": data.get("alt", "N/A"),
                    "Email": data.get("email", "N/A")
                }
            }
        else:
            return {"status": "not found", "message": "Ye number database me nahi mila"}
    except Exception as e:
        return {"status": "error", "message": str(e)}