from fastapi import FastAPI
import src.paraphrase.Paraphrase as Paraphrase
import src.translate.Translate as Translate

app = FastAPI(docs_url="/")
MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro', 'roen': 'BlackKakapo/opus-mt-ro-en'}

@app.get("/")
def index():
    return {'endpoints': ['/paraphrase', '/translate'], 'mtmodels': MTMODELS}

@app.get("/paraphrase")
def paraphrase(text: str, model: str):
    resultValue, exception = Paraphrase.paraphraseParaphraseMethod(text, model)
    return {"input": text, "translation": resultValue, "exception": exception}

@app.get("/translate")
def translate(text: str, model: str):
    # resultValue, exception = Translate.paraphraseTranslateMethod(text, model)
    resultValue, exception = Translate.gemma(text, model)
    return {"input": text, "translation": resultValue, "exception": exception}