from fastapi import FastAPI
from pydantic import BaseModel
from Similarity import Similarity

app = FastAPI()
similarity_model = Similarity()

# Define the Pydantic model BEFORE using it
class TextPairRequest(BaseModel):
    text: str
    claim: str

@app.post("/get_sim_text")
def get_sim_text_endpoint(request: TextPairRequest):
    try:
        claim_embedding = similarity_model.model.encode(
            request.claim,
            convert_to_tensor=True,
            show_progress_bar=False
        )
        results = similarity_model.get_sim_text(
            request.text,
            claim_embedding
        )
        return {"result": results}
    except Exception as e:
        return {"error": str(e)}