Update app.py

app.py

@@ -77,13 +77,13 @@ choices = [
     "4. 시스템을 거부하고 그냥 도망친다."
 ]
 
-def respond(message, history):
+ddef respond(message, history):
     try:
         sel_num = int(message.strip())
         if sel_num < 1 or sel_num > len(choices):
-            return "❌ 올바른 번호를 입력하세요. (1~4)"
+            return gr.ChatMessage(role="assistant", content="❌ 올바른 번호를 입력하세요. (1~4)")
     except ValueError:
-        return "❌ 번호를 입력하세요. (예: 1, 2, 3, 4)"
+        return gr.ChatMessage(role="assistant", content="❌ 번호를 입력하세요. (예: 1, 2, 3, 4)")
 
     user_choice = choices[sel_num - 1]
     result = qa_chain({"query": user_choice})
@@ -99,12 +99,18 @@ def respond(message, history):
     """
 
     response = llm_pipeline(prompt)[0]["generated_text"]
-    return f"[성진우]\n{response}"
 
-# ====== Gradio ChatInterface ======
+    # 사용자 메시지(오른쪽)
+    user_msg = gr.ChatMessage(role="user", content=f"{sel_num}번 선택 ({user_choice})")
+    # 성진우 메시지(왼쪽)
+    sjw_msg = gr.ChatMessage(role="assistant", content=response)
+
+    return [user_msg, sjw_msg]
+
+
 demo = gr.ChatInterface(
     respond,
-    title="성진우 선택 시뮬레이션 (카카오톡 스타일)",
+    title="성진우 선택 시뮬레이션 (카카오톡 스타일, 성진우 왼쪽)",
     description="1~4번 중 하나를 입력하면 성진우의 응답이 대화 형식으로 나타납니다."
 )