from fastapi import FastAPI
from pydantic import BaseModel
import joblib

app = FastAPI()

class PredictRequest(BaseModel):
    text: str

# Load models
model = joblib.load("bangla_model.pkl")
vectorizer = joblib.load("bangla_vectorizer.pkl")
label_encoder = joblib.load("bangla_label_encoder.pkl")

@app.get("/")
def root():
    return {"message": "Bangla model API is running!"}

@app.post("/predict")
def predict(request: PredictRequest):
    vect = vectorizer.transform([request.text])
    pred = model.predict(vect)
    label = label_encoder.inverse_transform(pred)
    # Convert numpy type to Python native type
    if label[0]==1:
        return {"prediction": "Hate"}
    else:
        return {"prediction": "Not Hate"}