Update app.py

app.py

@@ -25,7 +25,7 @@ def respond(prompt, system_message, max_tokens, temperature, top_p, selected_mod
     
     # 프롬프트 설정
     messages = [
-        {"role": "system", "content": system_message},
+        {"role": "system", "content": system_message + "\n입력에 대해서만 답변하세요. 추가 질문을 하지 마세요. 입력 내용만 반영하세요."},
         {"role": "user", "content": prompt}
     ]
     
@@ -43,9 +43,8 @@ def respond(prompt, system_message, max_tokens, temperature, top_p, selected_mod
             if stop_event.is_set():
                 break
             if chunk:
-                # 모델이 스스로 질문을 하지 않도록 특정 패턴을 체크
-                if not chunk.strip().endswith("?"):
-                    response += chunk
+                # 모델이 스스로 질문을 하지 않도록 패턴을 체크
+                response += chunk
                 yield [(prompt, response.strip())]  # 실시간으로 부분적인 응답 반환
     
     except Exception as e:
@@ -75,7 +74,7 @@ with gr.Blocks() as demo:
                 top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.90, step=0.05, label="Top-p (핵 샘플링)")
             
             system_message = gr.Textbox(
-                value="너는 나의 최고의 비서이다. 정확하게 답변하라. 추가 질문이나 입력 확장은 하지 않는다.",
+                value="너는 나의 최고의 비서이다. 정확하게 답변하라. 추가 질문을 하지 말고, 사용자의 입력 내용에 대해서만 답변하라.",
                 label="시스템 메시지",
                 lines=5
             )