from fastapi import FastAPI
from pydantic import BaseModel
from fastapi.responses import JSONResponse
from Paraphrase import paraphrase_paragraph

app = FastAPI()

class Sentence(BaseModel):
    sentence: str


@app.get("/")
def index():
    return "OK"

@app.post("/rephrase")
def rephrase(sentence: Sentence):
    try:
        rephrase_text = paraphrase_paragraph(sentence.sentence)
        return JSONResponse(status_code=200, content={"rephrased_sentence": rephrase_text})
    except Exception as e:
        return JSONResponse(status_code=422,content="Unable to rephrase")