Update app.py

app.py

@@ -19,9 +19,17 @@ def chat_with_model(message, history):
     if not loading_done:
         return history + [[message, "⚠️ 모델이 아직 로딩 중입니다. 잠시만 기다려 주세요."]]
     
-    prompt = f"다음 문장을 분석하고, 무례하거나 공격적인 표현이 있으면 알려주고 더 예의 있게 바꿔줘:\n\n{message}"
+    prompt = f"""
+    ### Instruction: 다음 문장을 분석해서 무례하거나 공격적인 표현이 있는지 판단하고, 있다면 더 예의 있게 고쳐줘.
+    
+    ### Input:
+    {message}
+    
+    ### Response:
+    """
     response = chat_model(prompt, max_new_tokens=200)[0]['generated_text']
-    return history + [[message, response]]
+    response_only = response[len(prompt):].strip()
+    return history + [[message, response_only]]
 
 # 상태 텍스트 반환 함수 (매번 새로 읽어옴)
 def get_status():