from fastapi import FastAPI
from pydantic import BaseModel
from app import GradioScamDetector

app = FastAPI()
detector = GradioScamDetector()

class PredictRequest(BaseModel):
    data: list[str]

@app.post("/predict")
async def predict(request: PredictRequest):
    message = request.data[0]
    result_text, confidence, _, _ = detector.predict_message(message)
    return {
        "result": result_text,
        "confidence": confidence
    }