from fastapi import FastAPI
from pydantic import BaseModel
from typing import List
from sentence_transformers import SentenceTransformer
from sklearn.metrics.pairwise import cosine_similarity

app = FastAPI()

model = SentenceTransformer("paraphrase-MiniLM-L3-v2")

class SimilarityRequest(BaseModel):
    ideal_profile: str
    text_blocks: List[str]

@app.post("/similarity")
def compute_similarity(req: SimilarityRequest):
    profile_vec = model.encode(req.ideal_profile)
    text_vecs = model.encode(req.text_blocks)
    similarities = cosine_similarity([profile_vec], text_vecs)[0].tolist()
    return similarities