from fastapi import FastAPI, HTTPException
from huggingface_hub import hf_hub_download
import pandas as pd

DATASET_REPO = "credi-net/CrediPred"
FILENAME = "mlpInfer_dec2024_pc1_embeddinggemma-300m_GNN-RNI.parquet"

app = FastAPI(title="CrediNet API")

lookup = {}

@app.on_event("startup")
def load_data():
    global lookup
    path = hf_hub_download(
        repo_id=DATASET_REPO,
        filename=FILENAME,
        repo_type="dataset"
    )
    df = pd.read_parquet(path, columns=["domain", "pc1_score"])
    lookup = dict(zip(df["domain"], df["pc1_score"]))
    print(f"Loaded {len(lookup):,} rows")

@app.get("/health")
def health():
    return {"status": "ok", "rows": len(lookup)}

@app.get("/by_domain/{domain}")
def get_score_score(domain: str):
    score = lookup.get(domain)
    if score is None:
        raise HTTPException(status_code=404, detail="Domain not found")
    return {"domain": domain, "pc1_score": float(score)}