main.py

import os
os.environ["HF_HOME"] = "/tmp/hf_home"
os.makedirs("/tmp/hf_cache", exist_ok=True)

from fastapi import FastAPI
from pydantic import BaseModel
from app.filter_review import filter_review

app = FastAPI()

@app.get("/")
def read_root():
    return {"message": "Selamat datang di API Filter Ulasan Spam untuk ArTour!"}

class FilterReviewRequest(BaseModel):
    text: str

@app.post("/filter-review")
def filter_spam(request: FilterReviewRequest):
    label, confidence = filter_review(request.text)
    
    binary_label = 1 if label.lower() == "spam" else 0
    
    return {"label": binary_label, "confidence": confidence}