server.py

# server.py
from fastapi import FastAPI, Body, Path
from fastapi.middleware.cors import CORSMiddleware
from recommendation_model import RecommendationModel, generate_comparison_questions
from fastapi.responses import JSONResponse
from data_processor import preprocess_input_data    # 데이터 전처리 함수
from utils import get_formatted_recommendation_response

app = FastAPI()


# CORS 설정 (필요한 도메인만 지정 가능)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 일단 모든 도메인 허용
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# DB 설정
DB_CONNECTION = "sqlite:///sqlite_recommend.db"

# 모델 인스턴스 생성
recommend_model = RecommendationModel(model_name="gpt-4o-mini")

@app.post("/ai/fortune")
def get_fortune(data: dict = Body(..., example={
    "question": "오늘 점심을 뭘 먹을까? 짜장면 vs 짬뽕?",
    "gpt_mbti": {"MBTI": "INFJ"},
    "user_info": {
        "birth": "1997-02-07",
        "birth_time": "23:05",
        "gender": "여성",
        "name": "엘리스",
        "isLunar": True
    }
})):
    # 최초 데이터 정제해서
    clean_data = preprocess_input_data(data)
    # 모델에 프롬프트 전달하고 응답 받고
    ai_response = recommend_model.get_recommendation(clean_data, session_id="default_session")
    # 반환 데이터를 한번 더 정제해서 전달 (fe, be와 의논 필요)
    final_response = get_formatted_recommendation_response(ai_response, session_id="default_session")
    return JSONResponse(content=final_response)


@app.get("/ai/choice")
def get_choices():
    response = generate_comparison_questions()
    print(response)
    return JSONResponse(content=response)


# @app.get("/ai/chat-history/{session_id}")
# def get_chat_history(session_id: str = Path(..., 
#     title="Session ID", description="조회할 대화 세션 ID", example="default_session"
# )):
#     """
#     특정 session_id의 대화 기록을 role과 함께 JSON 형태로 반환하는 API 엔드포인트
#     """
#     from utils import get_chat_history_with_roles
#     chat_history = get_chat_history_with_roles(session_id, DB_CONNECTION)
    
#     return JSONResponse(content={"session_id": session_id, "messages": chat_history})


if __name__ == "__main__":
    import uvicorn
    uvicorn.run("server:app", host="0.0.0.0", port=8000, reload=True)