Chanjeans commited on
Commit
5296f5a
·
verified ·
1 Parent(s): b63005a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -1297,7 +1297,14 @@ def chat_or_recommend(req: ChatOrRecommendRequest):
1297
  else:
1298
  # 1) 우울 분석
1299
  depression_score, depression_label = predict_depression(user_input)
1300
-
 
 
 
 
 
 
 
1301
  # 2) 상담 안내 메시지 준비
1302
  counseling_response = ""
1303
  if depression_label == "상담 권장":
@@ -1319,7 +1326,7 @@ def chat_or_recommend(req: ChatOrRecommendRequest):
1319
  recommendations_list = []
1320
 
1321
  # 우울도가 '상담 권장'이 아니고, 사용자가 hobby 정보가 있으면 → 추천
1322
- if depression_label != "상담 권장" and req.hobby:
1323
  user_profile = {
1324
  "extroversion": req.extroversion or "",
1325
  "feeling_thinking": req.feeling_thinking or "",
 
1297
  else:
1298
  # 1) 우울 분석
1299
  depression_score, depression_label = predict_depression(user_input)
1300
+ # 2) 우울도가 '정상'이면 챗봇만 수행
1301
+ if depression_label == "정상":
1302
+ chatbot_msg = chat_response(user_input, mode=mode)
1303
+ return {
1304
+ "mode": "chat",
1305
+ "response": chatbot_msg,
1306
+ "depression_label": depression_label
1307
+ }
1308
  # 2) 상담 안내 메시지 준비
1309
  counseling_response = ""
1310
  if depression_label == "상담 권장":
 
1326
  recommendations_list = []
1327
 
1328
  # 우울도가 '상담 권장'이 아니고, 사용자가 hobby 정보가 있으면 → 추천
1329
+ if depression_label == "관심 필요" and req.hobby:
1330
  user_profile = {
1331
  "extroversion": req.extroversion or "",
1332
  "feeling_thinking": req.feeling_thinking or "",