Update app.py

app.py

@@ -9,6 +9,7 @@ import torch
 import numpy as np
 from diffusers import DiffusionPipeline
 from transformers import pipeline as hf_pipeline
+import re
 
 ##############################################################################
 # 1) ZeroGPU 환경 처리 + device, dtype 설정
@@ -78,6 +79,29 @@ def contains_korean(text):
             return True
     return False
 
+##############################################################################
+# 입력 텍스트 클린징 함수: 한글, 영문, 숫자, 공백 및 일반적인 문장 부호만 허용
+##############################################################################
+def clean_input_text(text):
+    """
+    한글, 영문, 숫자, 공백 및 . , ! ? - : ; ' " 등의 문장 부호만 남기고 제거합니다.
+    필요에 따라 허용할 문자를 조정할 수 있습니다.
+    """
+    allowed = re.compile(r'[^ㄱ-ㅎ가-힣a-zA-Z0-9\s\.\,\!\?\-\:\;\'\"]')
+    cleaned_text = allowed.sub('', text)
+    return cleaned_text
+
+##############################################################################
+# 입력 텍스트의 예상치 못한 문자 로그 함수 (디버깅용)
+##############################################################################
+def log_unexpected_characters(text):
+    allowed = re.compile(r'[ㄱ-ㅎ가-힣a-zA-Z0-9\s\.\,\!\?\-\:\;\'\"]')
+    unexpected_chars = [char for char in text if not allowed.match(char)]
+    if unexpected_chars:
+        print("예상치 못한 문자 발견:", set(unexpected_chars))
+    else:
+        print("입력 텍스트에 예상치 못한 문자는 없습니다.")
+
 ##############################################################################
 # 이미지 생성 함수
 ##############################################################################
@@ -85,6 +109,11 @@ def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, heig
     original_prompt = prompt
     translated = False
     
+    # 입력 텍스트 클린징 적용
+    prompt = clean_input_text(prompt)
+    # 디버깅: 예상치 못한 문자 로그 출력 (필요 시 활성화)
+    # log_unexpected_characters(prompt)
+    
     if contains_korean(prompt):
         translation = translator(prompt)
         prompt = translation[0]['translation_text']
@@ -492,7 +521,6 @@ with gr.Blocks(
             text_input1 = gr.Textbox(label="키워드 1 (필수)", placeholder="예: 스마트폰")
             text_input2 = gr.Textbox(label="키워드 2 (선택)", placeholder="예: 인공지능")
             text_input3 = gr.Textbox(label="키워드 3 (선택)", placeholder="예: 헬스케어")
-            # 드롭다운 대신 라디오 버튼으로 카테고리 선택 (모든 옵션이 펼쳐짐)
             category_radio = gr.Radio(
                 label="카테고리 선택",
                 choices=list(physical_transformation_categories.keys()),
@@ -520,7 +548,6 @@ with gr.Blocks(
             idea_output = gr.Markdown(label="아이디어 결과")
             generated_image = gr.Image(label="생성된 디자인 이미지", type="pil")
     
-    # 다양한 카테고리가 포함된 예제 (라디오 버튼 입력)
     gr.Examples(
         examples=[
             ["스마트폰", "", "", "공간 이동"],