Update app.py

app.py

@@ -1107,6 +1107,9 @@ def recommend_api(profile: UserProfile):
 class ChatRequest(BaseModel):
     user_input: str
     mode: str  # `emotion` 또는 `rational`을 명확하게 입력받도록 설정
+RECOMMEND_KEYWORDS = [
+    "추천", "추천해줘", "취미 추천"
+]
 
 @app.post("/chat")
 def chat_api(req: ChatRequest):
@@ -1117,3 +1120,73 @@ def chat_api(req: ChatRequest):
 
     reply = chat_response(user_text, mode=mode)
     return {"response": reply}
+from pydantic import BaseModel
+from typing import Optional
+
+# 기존 ChatRequest 에 user_profile 용 필드를 추가한 모델
+class ChatOrRecommendRequest(BaseModel):
+    user_input: str  # 사용자의 채팅 메시지
+    mode: str        # "emotion" 또는 "rational" (챗봇 모드)
+    
+    # 추천용 프로필 (선택적으로 받을 수 있도록 Optional)
+    extroversion: Optional[str] = None
+    feeling_thinking: Optional[str] = None
+    hobby: Optional[str] = None
+    detail_hobby: Optional[str] = None
+
+@app.post("/chat_or_recommend")
+def chat_or_recommend(req: ChatOrRecommendRequest):
+    user_text = req.user_input
+    mode = req.mode.lower()
+
+    # 1) 추천 키워드 포함 여부 확인
+    if any(keyword in user_text for keyword in RECOMMEND_KEYWORDS):
+        # 사용자가 추천을 원한다고 판단
+        # 프로필 정보가 없는 경우를 대비해서 기본값 혹은 예외처리를 할 수도 있음
+        if not req.hobby:
+            # 만약 프로필이 없으면 예외 처리하거나, 
+            # "프로필(취미/성향)을 입력해주세요" 라고 안내할 수도 있음
+            raise HTTPException(
+                status_code=400, 
+                detail="추천을 위해 hobby, detail_hobby, extroversion, feeling_thinking 정보가 필요합니다."
+            )
+        
+        # (A) 추천 API 로직 호출
+        user_profile = {
+            "extroversion": req.extroversion or "",
+            "feeling_thinking": req.feeling_thinking or "",
+            "hobby": req.hobby or "",
+            "detail_hobby": req.detail_hobby or "",
+        }
+        top_items = recommend_content_based(user_profile, top_n=5)
+
+        results = []
+        for (item, score) in top_items:
+            results.append({
+                "item_id": item["item_id"],
+                "title": item["title"],
+                "desc": item["desc"],
+                "personality": item["personality"],
+                "score": round(score, 4)
+            })
+
+        return {
+            "mode": "recommend",
+            "recommendations": results
+        }
+
+    else:
+        # 2) 추천 키워드가 없으면 챗봇 로직
+        if mode not in ["emotion", "rational"]:
+            raise HTTPException(
+                status_code=400, 
+                detail="mode는 'emotion' 또는 'rational'이어야 합니다."
+            )
+
+        # (B) 챗봇 호출
+        reply = chat_response(user_text, mode=mode)
+        return {
+            "mode": "chat",
+            "response": reply
+        }
+