from fastapi import FastAPI
from pydantic import BaseModel
import joblib

app = FastAPI()

class PredictRequest(BaseModel):
    text: str

# Load model + vectorizer + label encoder
model = joblib.load("svm_model.pkl")
vectorizer = joblib.load("language_tfid.pkl")
label_encoder = joblib.load("label_encoder.pkl")

@app.get("/")
def root():
    return {"message": "Language Identification API"}

@app.post("/predict")
def predict(request: PredictRequest):
    vect = vectorizer.transform([request.text])
    pred = model.predict(vect)
    label = label_encoder.inverse_transform(pred)[0]

    return {"prediction": label}