app.py

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from model.emotion_model import EmotionAnalysis
import uvicorn

app = FastAPI()


class DiaryInput(BaseModel):
    content: str


class EmotionResponse(BaseModel):
    emotion: dict[str, int]


@app.get("/api/health")
def health():
    return {"status": "ok"}


@app.post("/api/predict", response_model=EmotionResponse)
def predict(diary: DiaryInput):
    try:
        emotion_model = EmotionAnalysis()
        emotion_result = emotion_model.analyze_emotion(diary.content)
        return EmotionResponse(emotion=emotion_result)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


if __name__ == "__main__":
    uvicorn.run(app, host="127.0.0.1", port=8000)