from fastapi import FastAPI, HTTPException
from app.model.model import predict_language, __version__
from app.schemas import Prediction, TextIn

app = FastAPI()

@app.get("/")
def home():
    return {"message": "Ok", "model_version": __version__}

@app.post("/predict", response_model=Prediction)
def predict(payload: TextIn):
    try:
        language = predict_language(payload.text)
        return {"language": language}
    except Exception as e:
        raise HTTPException(status_code=400, detail=str(e))