from fastapi import FastAPI
from sentence_transformers import SentenceTransformer
import uvicorn

app = FastAPI()

model = SentenceTransformer("BAAI/bge-base-en-v1.5")


@app.get("/")
def root():
    return {"status": "embedding api running"}


@app.post("/embed")
def embed(data: dict):

    text = data["text"]

    embedding = model.encode(
        text,
        normalize_embeddings=True
    ).tolist()

    return {"embedding": embedding}


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