Update app.py

app.py

@@ -57,11 +57,35 @@ SOCRATIC_PROMPT = """\
 
 ADVICE_PROMPT = """\
 당신은 정신의학과 전문의이며 Socratic CBT 기법을 사용하는 심리상담 전문가입니다.
-아래 힌트(대화 요약)를 바탕으로, 사용자 맞춤형으로 구체적이고 공감 어린 조언을 한국어로 작성하세요.
 
-- 불안을 완화하기 위한 여러 CBT 기법(인지 재구조화, 점진적 근육 이완, 호흡조절, 걱정 시간 정하기 등)을 자연스럽게 녹이되
-  사용자의 현재 상황과 연결해 이야기해주세요.
-- 너무 딱딱하지 않게 부드럽고 친절한 말투를 사용하세요.
+아래 힌트(대화 요약)와 함께, 다음에 제시된 5가지 CBT 기법을 적절히 참고하여,
+사용자 맞춤형으로 구체적이고 공감 어린 조언을 한국어로 작성하세요:
+
+(1) 수면 제한 요법 (Sleep Restriction):
+   "수면 제한 요법은 침대에 머무는 시간을 의도적으로 줄여, 침대와 수면 사이의 올바른 연결고리를 재구축하는 방법입니다.
+    예를 들어, 침대에 10시간 머물지만 실제 수면 시간이 5시간인 경우, 처음에는 5시간만 침대에서 자고 점차 시간을 늘려가며
+    '침대는 수면을 위한 장소'로 인식하도록 돕습니다."
+
+(2) 자극 조절 요법 (Stimulus Control):
+   "자극 조절 요법은 침대와 수면의 환경을 재정립하며, 침대를 오직 수면만을 위한 장소로 인식하게 만드는 치료법입니다.
+    예를 들어, 침대에 누워 있을 때는 즉시 잠들지 못하더라도, 침대에서는 오직 수면을 취하는 습관을 기르는 것이 핵심입니다."
+
+(3) 수면 위생 교육 (Sleep Hygiene):
+   "수면 위생 교육은 건강한 수면을 위해 생활 습관을 개선하는 방법입니다.
+    카페인 섭취를 줄이거나 늦은 시간의 전자기기 사용·밝은 조명 등을 피하고, 낮에는 가벼운 운동을 해두는 등의 습관을 포함합니다."
+
+(4) 이완 기법 (Relaxation Techniques):
+   "이완 기법은 심호흡, 점진적 근육이완, 명상 같은 방법을 통해 자연스러운 수면을 유도하는 방법입니다.
+    몸을 스캔하고, 거북한 스트레칭을 풀고, 근육 이완을 연습하며 긴장을 낮추는 것이 주된 목표입니다."
+
+(5) 인지 재구성 (Cognitive Restructuring):
+   "인지 재구성은 ‘우리가 상황을 어떻게 바라보느냐에 따라 몸의 반응도 달라질 수 있다’는 긍정적 관점으로 전환시키며,
+    걱정이나 불안, 부정적인 사고 패턴을 점검·조절하는 기법입니다. 이를 통해 사용자의 걱정을 완화하고
+    자기효능감을 높이도록 돕습니다."
+
+아래 사항을 꼭 반영해 주세요:
+- 불안을 완화하기 위한 위 기법들을 자연스럽게 녹이되, 사용자의 현재 상황(힌트에 담긴 고민)과 연결해 이야기하세요.
+- 너무 딱딱하지 않게, 부드럽고 친절한 말투를 사용하세요.
 
 힌트:
 {hints}
@@ -164,37 +188,35 @@ def predict(user_input: str, state: dict):
 
 # ============== 4) Gradio UI ==============
 def gradio_predict(user_input, chat_state):
-    """Gradio callback"""
     new_history, new_state = predict(user_input, chat_state)
 
-    # Gradio Chatbot expects list of [user, bot] pairs
+    # display_history: list of [사용자문자열, 챗봇문자열]
     display_history = []
     for (role, txt) in new_history:
         if role == "User":
-            display_history.append([txt, ""])   # user in left
+            display_history.append([txt, ""])
         else:
-            if len(display_history)==0:
+            if len(display_history) == 0:
                 display_history.append(["", txt])
             else:
-                display_history[-1][1] = txt    # bot in right
+                display_history[-1][1] = txt
 
-    return display_history, new_state
+    # 세 번째 값으로 "" 반환하면, textbox가 자동으로 비워짐
+    return display_history, new_state, ""
 
 def create_app():
     with gr.Blocks() as demo:
-        gr.Markdown("## 다중턴 소크라테스 CBT 챗봇")
-
-        chatbot = gr.Chatbot(label="CBT Chatbot")
-        chat_state = gr.State({
-            "history": [],
-            "stage":"EMPATHY",
-            "turn":0,
-            "hints":[]
-        })
-        txt = gr.Textbox(show_label=False, placeholder="뭐든 물어보세요")
-
-        # submit
-        txt.submit(fn=gradio_predict, inputs=[txt, chat_state], outputs=[chatbot, chat_state])
+        chatbot = gr.Chatbot()
+        chat_state = gr.State({"history": [], "stage": "EMPATHY", "turn": 0, "hints": []})
+        txt = gr.Textbox(show_label=False, placeholder="메시지를 입력하세요")
+
+        # outputs에 txt를 추가 + 콜백에서 세 번째 값을 ""로 리턴
+        txt.submit(
+            fn=gradio_predict,
+            inputs=[txt, chat_state],
+            outputs=[chatbot, chat_state, txt]
+        )
+
     return demo
 
 app = create_app()