main.py

from fastapi import FastAPI
from app.train import train_model
from app.test import test_model
from app.validate import validate_model
from app.predict import predict_text
from pydantic import BaseModel

app = FastAPI()

class TextRequest(BaseModel):
    text: str

@app.post("/train")
def train_api():
    result = train_model()
    return {"message": "Training complete", "details": result}

@app.post("/test")
def test_api():
    metrics = test_model()
    return {"message": "Testing complete", "metrics": metrics}

@app.post("/validate")
def validate_api():
    metrics = validate_model()
    return {"message": "Validation complete", "metrics": metrics}

@app.post("/predict")
def predict_api(request: TextRequest):
    prediction = predict_text(request.text)
    return {"prediction": prediction}