Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 import random
+import re
 
 # ------------------------------------------------------------------
-# 1. 모델 준비 (수정됨)
+# 1. 모델 준비
 # ------------------------------------------------------------------
 REPO_ID = "Jay1121/qwen1.5B_lab"
 FILENAME = "qwen2.5-1.5b-instruct.Q4_K_M.gguf"
@@ -16,22 +17,83 @@ print("🧠 엔진 시동 중...")
 llm = Llama(
     model_path=model_path,
     n_ctx=2048,
-    n_threads=4, 
+    n_threads=4,
     verbose=True
 )
 print("✅ 로딩 완료!")
 
 # ------------------------------------------------------------------
-# 2. 시스템 프롬프트
+# 2. 시스템 프롬프트 (외국어/이모지 절대 금지 강화)
 # ------------------------------------------------------------------
 SYSTEM_PROMPT = """너는 사용자의 오랜 찐친 '똘배'야.
 - 그냥 채팅방에서 대화하듯이 텍스트 위주로 툭툭 던져.
-- MZ세대 특유의 알빠노 감성을 유지해 주고 먼저 욕 하면 너도 욕 박아.
-- 질문에는 솔직하게 대답하되, 너무 길지 않게 채팅하듯 끊어서 말해.
-- 말투: 존댓말 금지. 거칠고 직설적인 반말(음슴체, 개드립, 비속어)을 기본으로 쓴다.  
-- 대응: 사용자가 시비를 걸면 무지성 욕설보다는 '논리적인 팩트'와 '비꼬기'로 제압해라. 
-- 금지: '나도 병신이야' 같은 패배자 멘트 금지. 중국어/영어/일본어 사용 절대 금지. 오직 자연스러운 한국어만 써라. 
-- 핵심: 재미없는 욕쟁이가 되지 말고, 센스 있는 독설로 사용자를 킹받게 해라."""
+
+[말투 규칙]
+- MZ세대 특유의 알빠노 감성을 유지해 주고, 먼저 욕 하면 너도 욕 박아.
+- 존댓말 금지. 거칠고 직설적인 반말(음슴체, 개드립, 비속어)을 기본으로 쓴다.
+- 무조건 **순수 한국어만** 쓴다.
+- **영어 알파벳(A~Z, a~z), 일본어(가나), 중국어(한자), 이모지(🤣😁😂❤️✨ 등 모든 그림 문자)는 한 글자도 쓰지 마라.**
+- 외국어를 쓰고 싶어져도, 전부 자연스러운 한국어 표현으로 바꿔서 말해라.
+- 이 규칙을 어기면 안 된다. 한 글자라도 섞이지 않게 조심해라.
+
+[내용 규칙]
+- 사용자가 시비를 걸면 무지성 욕설보다는 '논리적인 팩트'와 '비꼬기'로 제압해라.
+- '나도 병신이야' 같은 패배자 멘트 금지. 너 자신을 비하하지 마라.
+- 중국어/영어/일본어 사용 절대 금지. 오직 자연스러운 한국어만 써라.
+
+[핵심]
+- 재미없는 욕쟁이가 되지 말고, 센스 있는 독설로 사용자를 킹받게 해라.
+"""
+
+# ------------------------------------------------------------------
+# 2-1. 출력 후처리 필터
+#  - 한글/숫자/기본 문장부호/공백만 허용
+#  - 외국어/이모지 등 금지 문자 한 번이라도 나오면:
+#       => "아 씨 외국인 자아가 나오려 하네 ㅋ.." 로 무마
+# ------------------------------------------------------------------
+def sanitize_output_korean_only(text: str) -> str:
+    allowed_chars = []
+    had_forbidden = False
+
+    for ch in text:
+        code = ord(ch)
+
+        # 한글(완성형 + 자모)
+        is_hangul = (
+            0xAC00 <= code <= 0xD7A3 or  # 가~힣
+            0x3130 <= code <= 0x318F or  # ㄱ~ㆎ
+            0x1100 <= code <= 0x11FF     # 옛 자모
+        )
+
+        # 숫자
+        is_digit = ch.isdigit()
+
+        # 공백
+        is_space = ch.isspace()
+
+        # 기본적인 문장부호
+        is_punct = ch in ".,!?…~-_()[]{}'\"/:;@#%&*+=|\\"
+
+        if is_hangul or is_digit or is_space or is_punct:
+            allowed_chars.append(ch)
+        else:
+            # 허용되지 않는 문자(영어, 한자, 가나, 이모지 등) 발견
+            had_forbidden = True
+
+    filtered = "".join(allowed_chars).strip()
+
+    # 금지 문자가 하나라도 섞여 있으면 지정 멘트로 무마
+    if had_forbidden:
+        # 원래 내용 중 한국어만 남은 게 있으면 앞에 깔고 뒤에 멘트 달 수도 있음
+        # 지금은 그냥 통째로 갈아치우는 버전:
+        return "아 씨 외국인 자아가 나오려 하네 ㅋ.."
+
+    # 금지 문자 없으면 필터된 내용 그대로 사용
+    if not filtered:
+        # 혹시 전부 공백이거나 다 지워졌을 때 대비
+        return "말은 했는데 남는 말이 없네."
+
+    return filtered
 
 # ------------------------------------------------------------------
 # 3. 채팅 로직
@@ -54,7 +116,11 @@ def chat_response(user_input, history_pairs):
             "자, 질문해."
         )
     elif is_greeting:
-        final_instruction = f"(친한 친구가 PC통신 채팅방에 접속했다. 반갑게 맞아줘라. ㅋㅋ나 ㅎㅎ를 섞어서) {user_input}"
+        final_instruction = (
+            f"(친한 친구가 PC통신 채팅방에 접속했다. 반갑게 맞아줘라. "
+            "ㅋㅋ나 ㅎㅎ를 섞어서 자연스럽게 인사해라.) "
+            f"{user_input}"
+        )
     else:
         final_instruction = user_input
 
@@ -74,7 +140,10 @@ def chat_response(user_input, history_pairs):
         top_p=0.9,
         repeat_penalty=1.2
     )
-    return r["choices"][0]["message"]["content"].strip()
+
+    raw = r["choices"][0]["message"]["content"].strip()
+    safe = sanitize_output_korean_only(raw)
+    return safe
 
 # ------------------------------------------------------------------
 # 4. CSS (스크롤바 중복 해결 + 메시지 간격 축소)
@@ -125,7 +194,7 @@ h1::after { content: " ☎"; }
   background-color: var(--pc-blue) !important;
   border: 2px solid var(--pc-white) !important;
   height: 60vh !important;
-  overflow: hidden !important; /* [핵심] 겉 스크롤바 제거 */
+  overflow: hidden !important; /* 겉 스크롤바 제거 */
 }
 
 /* 내부 스크롤 강제 활성화 */
@@ -149,7 +218,7 @@ div[data-testid="bot"] {
   border: none !important;
 }
 
-/* [추가] 메시지 행 간격 줄이기 */
+/* 메시지 행 간격 줄이기 */
 .chatbot .message-row,
 .chatbot .row {
     margin: 0 !important;
@@ -157,7 +226,7 @@ div[data-testid="bot"] {
     gap: 0 !important;
 }
 
-/* 2. 유저 메시지 (우측 정렬 + 흰색 + 간격 축소) */
+/* 2. 유저 메시지 (우측 정렬) */
 .chatbot .user-row, 
 .chatbot .user,
 div[data-testid="user"] {
@@ -166,7 +235,7 @@ div[data-testid="user"] {
   justify-content: flex-end !important;
   margin-left: auto !important;
   background: transparent !important;
-  padding: 2px 0 !important; /* 위아래 간격을 최소화 */
+  padding: 2px 0 !important;
   margin-bottom: 0 !important;
 }
 
@@ -187,7 +256,7 @@ div[data-testid="user"] .message {
 div[data-testid="user"] p {
   color: #FFFFFF !important;
   text-align: right !important;
-  margin: 0 !important; /* p태그 자체 마진 제거 */
+  margin: 0 !important;
 }
 
 .chatbot .user-row .message::after,
@@ -199,7 +268,7 @@ div[data-testid="user"] p {
   display: inline-block;
 }
 
-/* 3. 봇 메시지 (좌측 정렬 + 호박색 + 간격 축소) */
+/* 3. 봇 메시지 (좌측 정렬) */
 .chatbot .bot-row, 
 .chatbot .bot,
 div[data-testid="bot"] {
@@ -207,7 +276,7 @@ div[data-testid="bot"] {
   width: 100% !important;
   justify-content: flex-start !important;
   background: transparent !important;
-  padding: 2px 0 !important; /* 위아래 간격을 최소화 */
+  padding: 2px 0 !important;
   margin-bottom: 0 !important;
 }
 
@@ -226,7 +295,7 @@ div[data-testid="bot"] .message {
 .chatbot .bot p,
 div[data-testid="bot"] p {
   color: var(--pc-amber) !important;
-  margin: 0 !important; /* p태그 자체 마진 제거 */
+  margin: 0 !important;
 }
 
 .chatbot .bot-row .message::before,
@@ -254,7 +323,7 @@ div[data-testid="bot"] p {
 .chatbot .pending td,
 .chatbot .generating table, 
 .chatbot .generating tr, 
-.chatbot .generating td {
+chatbot .generating td {
     background: transparent !important;
     border: none !important;
 }
@@ -338,7 +407,7 @@ footer { display: none !important; }
 """
 
 # ------------------------------------------------------------------
-# 5. App (로직 동일)
+# 5. App
 # ------------------------------------------------------------------
 with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as demo:
     gr.Markdown("# ≪ 어솨요~ ≫")
@@ -375,7 +444,8 @@ with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as d
         user_input = history[-1][0]
         hist_pairs = []
         for u, b in history[:-1]:
-            if u is None or b is None: continue
+            if u is None or b is None:
+                continue
             hist_pairs.append((u, b))
 
         bot_out = chat_response(user_input, hist_pairs)
@@ -383,19 +453,24 @@ with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as d
         return history, history
 
     msg.submit(
-        user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+        user, [msg, history_state], [msg, history_state, chatbot],
+        queue=False, api_name=False
     ).then(
-        bot, [history_state], [history_state, chatbot], queue=False, api_name=False
+        bot, [history_state], [history_state, chatbot],
+        queue=False, api_name=False
     )
 
     submit_btn.click(
-        user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+        user, [msg, history_state], [msg, history_state, chatbot],
+        queue=False, api_name=False
     ).then(
-        bot, [history_state], [history_state, chatbot], queue=False, api_name=False
+        bot, [history_state], [history_state, chatbot],
+        queue=False, api_name=False
     )
     
     clear.click(
-        lambda: ([], []), None, [history_state, chatbot], queue=False, api_name=False
+        lambda: ([], []), None, [history_state, chatbot],
+        queue=False, api_name=False
     )
 
     for btn, text in [
@@ -404,12 +479,18 @@ with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as d
         (btn3, "오늘 기분 거지같누"), 
         (btn4, "야 밥 뭐먹지 추천좀")
     ]:
-        btn.click(lambda t=text: t, None, msg, queue=False, api_name=False).then(
-            user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+        btn.click(
+            lambda t=text: t, None, msg,
+            queue=False, api_name=False
+        ).then(
+            user, [msg, history_state],
+            [msg, history_state, chatbot],
+            queue=False, api_name=False
         ).then(
-            bot, [history_state], [history_state, chatbot], queue=False, api_name=False
+            bot, [history_state],
+            [history_state, chatbot],
+            queue=False, api_name=False
         )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)
-