from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from inference import outcome
import warnings
warnings.filterwarnings("ignore")

app=FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.get('/')
def index():
    return {"message":"Welcome to Hinglish Translator"}

@app.post('/predict')
def predict(input: str):
    return {"result":outcome(input)}