Update app-backup.py

app-backup.py

@@ -9,32 +9,26 @@ import torch
 import numpy as np
 from diffusers import DiffusionPipeline
 from transformers import pipeline as hf_pipeline
+import re
 
 ##############################################################################
 # 1) ZeroGPU 환경 처리 + device, dtype 설정
 ##############################################################################
-##############################################################################
-# 1) ZeroGPU 환경 처리 + device, dtype 설정
-##############################################################################
-# ZeroGPU 초기화 시도
 try:
     import zerogpu
     zerogpu.init()
     print("ZeroGPU initialized successfully")
     device = "cuda" if torch.cuda.is_available() else "cpu"
 except ImportError:
-    # ZeroGPU가 설치되지 않은 경우
     print("ZeroGPU package not installed, continuing without it")
     if os.getenv("ZERO_GPU"):
         print("ZeroGPU environment variable is set but zerogpu package is not installed.")
     device = "cuda" if torch.cuda.is_available() else "cpu"
 except Exception as e:
-    # ZeroGPU 초기화 중 다른 오류가 발생한 경우
     print(f"Error initializing ZeroGPU: {e}")
     print("Continuing without ZeroGPU")
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# GPU일 때만 bfloat16, 그 외에는 float32
 dtype = torch.bfloat16 if device == "cuda" else torch.float32
 
 print(f"Using device: {device}, dtype: {dtype}")
@@ -57,7 +51,6 @@ try:
     print("Models loaded successfully")
 except Exception as e:
     print(f"Error loading models: {e}")
-    # 모델 로드 에러 처리를 위한 더미 함수들
     def dummy_translator(text):
         return [{'translation_text': text}]
     
@@ -87,23 +80,48 @@ def contains_korean(text):
     return False
 
 ##############################################################################
-# 이미지 생성 함수
+# 입력 텍스트 클린징 함수: 한글, 영문, 숫자, 공백 및 일반적인 문장 부호만 허용
 ##############################################################################
-def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, height=1024, num_inference_steps=4):
+def clean_input_text(text):
     """
-    생성된 확장 아이디어 텍스트(prompt)를 입력받아,
-    필요시 한국어를 영어로 번역한 후 DiffusionPipeline으로 이미지를 생성합니다.
+    한글, 영문, 숫자, 공백 및 . , ! ? - : ; ' " 등의 문장 부호만 남기고 제거합니다.
+    필요에 따라 허용할 문자를 조정할 수 있습니다.
     """
+    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("입력 텍스트에 예상치 못한 문자는 없습니다.")
+
+##############################################################################
+# 이미지 생성 함수
+##############################################################################
+def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, height=1024, num_inference_steps=4):
     original_prompt = prompt
     translated = False
     
-    # 한국어가 포함되어 있으면 영어로 번역
+    # 입력 텍스트 클린징 적용
+    prompt = clean_input_text(prompt)
+    
+    # 사전 처리: 입력 텍스트가 너무 길면 미리 자르기 (예: 1000자 이상이면 1000자로 잘라냄)
+    if len(prompt) > 1000:
+        prompt = prompt[:1000]
+    
     if contains_korean(prompt):
-        translation = translator(prompt)
+        # 번역 호출 시 max_length와 truncation 옵션을 추가하여 길이 초과 문제를 방지
+        translation = translator(prompt, max_length=400, truncation=True)
         prompt = translation[0]['translation_text']
         translated = True
 
-    # 랜덤 시드 설정
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
@@ -120,6 +138,7 @@ def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, heig
     
     return image
 
+
 ##############################################################################
 # 로깅 설정
 ##############################################################################
@@ -167,12 +186,17 @@ def choose_alternative(transformation):
 # 물리적 변화 카테고리 사전 (총 15개)
 ##############################################################################
 physical_transformation_categories = {
-    "공간 이동": [
-        "앞/뒤 이동", "좌/우 이동", "위/아래 이동", "세로축 회전(고개 끄덕임)",
-        "가로축 회전(고개 젓기)", "길이축 회전(옆으로 기울임)", "원 운동", "나선형 이동",
-        "관성에 의한 미끄러짐", "회전축 변화", "불규칙 회전", "흔들림 운동", "포물선 이동",
-        "무중력 부유", "수면 위 부유", "점프/도약", "슬라이딩", "롤링", "자유 낙하",
-        "왕복 운동", "탄성 튕김", "관통", "회피 움직임", "지그재그 이동", "스윙 운동"
+    "센서 기능": [
+        "시각 센서/감지", "청각 센서/감지", "촉각 센서/감지", "미각 센서/감지", "후각 센서/감지",
+        "온도 센서/감지", "습도 센서/감지", "압력 센서/감지", "가속도 센서/감지", "회전 센서/감지",
+        "근접 센서/감지", "위치 센서/감지", "운동 센서/감지", "가스 센서/감지", "적외선 센서/감지",
+        "자외선 센서/감지", "방사선 센서/감지", "자기장 센서/감지", "전기장 센서/감지", "화학물질 센서/감지",
+        "생체신호 센서/감지", "진동 센서/감지", "소음 센서/감지", "빛 세기 센서/감지", "빛 파장 센서/감지",
+        "기울기 센서/감지", "pH 센서/감지", "전류 센서/감지", "전압 센서/감지", "이미지 센서/감지",
+        "거리 센서/감지", "깊이 센서/감지", "중력 센서/감지", "속도 센서/감지", "흐름 센서/감지",
+        "수위 센서/감지", "탁도 센서/감지", "염도 센서/감지", "금속 감지", "압전 센서/감지",
+        "광전 센서/감지", "열전대 센서/감지", "홀 효과 센서/감지", "초음파 센서/감지", "레이더 센서/감지",
+        "라이다 센서/감지", "터치 센서/감지", "제스처 센서/감지", "심박 센서/감지", "혈압 센서/감지"
     ],
     "크기와 형태 변화": [
         "부피 늘어남/줄어듦", "길이 늘어남/줄어듦", "너비 늘어남/줄어듦", "높이 늘어남/줄어듦",
@@ -190,7 +214,10 @@ physical_transformation_categories = {
         "광택/윤기 변화", "색조/채도 변화", "발광/형광", "빛 산란 효과",
         "빛 흡수 변화", "반투명 효과", "그림자 효과 변화", "자외선 반응 변화",
         "야광 효과"
-    ],
+    ],    
+
+
+
     "물질의 상태 변화": [
         "고체/액체/기체 전환", "결정화/용해", "산화/부식", "딱딱해짐/부드러워짐",
         "특수 상태 전환", "무정형/결정형 전환", "성분 분리", "미세 입자 형성/분해",
@@ -199,14 +226,7 @@ physical_transformation_categories = {
         "건조/습윤", "팽윤/수축", "동결/해동", "풍화/침식", "충전/방전",
         "결합/분리", "발효/부패"
     ],
-    "열 관련 변화": [
-        "온도 상승/하강", "열에 의한 팽창/수축", "열 전달/차단", "압력 상승/하강",
-        "열 변화에 따른 자화", "무질서도 변화", "열전기 현상", "자기장에 의한 열 변화",
-        "상태변화 중 열 저장/방출", "열 스트레스 발생/해소", "급격한 온도 변화 영향",
-        "복사열에 의한 냉각/가열", "발열/흡열", "열 분포 변화", "열 반사/흡수",
-        "냉각 응축", "열 활성화", "열 변색", "열 팽창 계수 변화", "열 안정성 변화",
-        "내열성/내한성", "자기발열", "열적 평형/불균형", "열적 변형", "열 분산/집중"
-    ],
+
     "움직임 특성 변화": [
         "가속/감속", "일정 속도 유지", "진동/진동 감소", "부딪힘/튕김",
         "회전 속도 증가/감소", "회전 방향 변화", "불규칙 움직임", "멈췄다 미끄러지는 현상",
@@ -223,22 +243,12 @@ physical_transformation_categories = {
         "외부 구조 변화", "중심축 이동", "균형점 변화", "계층 구조 변화", "지지 구조 변화",
         "응력 분산 구조", "충격 흡수 구조", "그리드/매트릭스 구조 변화", "상호 연결성 변화"
     ],
-    "전기 및 자기 변화": [
-        "자성 생성/소멸", "전하량 증가/감소", "전기장 생성/소멸", "자기장 생성/소멸",
-        "초전도 상태 전환", "강유전체 특성 변화", "양자 상태 변화", "플라즈마 상태 형성/소멸",
-        "스핀파 전달", "빛에 의한 전기 발생", "압력에 의한 전기 발생", "자기장 속 전류 변화",
-        "전기 저항 변화", "전기 전도성 변화", "정전기 발생/방전", "전자기 유도",
-        "전자기파 방출/흡수", "전기 용량 변화", "자기 이력 현상", "전기적 분극",
-        "전자 흐름 방향 변화", "전기적 공명", "전기적 차폐/노출", "자기 차폐/노출",
-        "자기장 방향 정렬"
-    ],
-    "화학적 변화": [
-        "표면 코팅 변화", "물질 성분 변화", "화학 반응 변화", "촉매 작용 시작/중단",
-        "빛에 의한 화학 반응", "전기에 의한 화학 반응", "단분자막 형성", "분자 수준 계산 변화",
-        "자연 모방 표면 변화", "환경 반응형 물질 변화", "주기적 화학 반응", "산화", "환원",
-        "고분자화", "물 분해", "화합", "방사선 영향", "산-염기 반응", "중화 반응",
-        "이온화", "화학적 흡착/탈착", "촉매 효율 변화", "효소 활성 변화", "발색 반응",
-        "pH 변화", "화학적 평형 이동", "결합 형성/분해", "용해도 변화"
+    "공간 이동": [
+        "앞/뒤 이동", "좌/우 이동", "위/아래 이동", "세로축 회전(고개 끄덕임)",
+        "가로축 회전(고개 젓기)", "길이축 회전(옆으로 기울임)", "원 운동", "나선형 이동",
+        "관성에 의한 미끄러짐", "회전축 변화", "불규칙 회전", "흔들림 운동", "포물선 이동",
+        "무중력 부유", "수면 위 부유", "점프/도약", "슬라이딩", "롤링", "자유 낙하",
+        "왕복 운동", "탄성 튕김", "관통", "회피 움직임", "지그재그 이동", "스윙 운동"
     ],
     "시간 관련 변화": [
         "노화/풍화", "마모/부식", "색 바램/변색", "손상/회복", "수명 주기 변화",
@@ -264,6 +274,32 @@ physical_transformation_categories = {
         "소리 지향성", "음향 왜곡", "비트 생성", "하모닉스 생성", "주파수 변조",
         "음향 충격파", "음향 필터링"
     ],
+    "열 관련 변화": [
+        "온도 상승/하강", "열에 의한 팽창/수축", "열 전달/차단", "압력 상승/하강",
+        "열 변화에 따른 자화", "무질서도 변화", "열전기 현상", "자기장에 의한 열 변화",
+        "상태변화 중 열 저장/방출", "열 스트레스 발생/해소", "급격한 온도 변화 영향",
+        "복사열에 의한 냉각/가열", "발열/흡열", "열 분포 변화", "열 반사/흡수",
+        "냉각 응축", "열 활성화", "열 변색", "열 팽창 계수 변화", "열 안정성 변화",
+        "내열성/내한성", "자기발열", "열적 평형/불균형", "열적 변형", "열 분산/집중"
+    ],
+    "전기 및 자기 변화": [
+        "자성 생성/소멸", "전하량 증가/감소", "전기장 생성/소멸", "자기장 생성/소멸",
+        "초전도 상태 전환", "강유전체 특성 변화", "양자 상태 변화", "플라즈마 상태 형성/소멸",
+        "스핀파 전달", "빛에 의한 전기 발생", "압력에 의한 전기 발생", "자기장 속 전류 변화",
+        "전기 저항 변화", "전기 전도성 변화", "정전기 발생/방전", "전자기 유도",
+        "전자기파 방출/흡수", "전기 용량 변화", "자기 이력 현상", "전기적 분극",
+        "전자 흐름 방향 변화", "전기적 공명", "전기적 차폐/노출", "자기 차폐/노출",
+        "자기장 방향 정렬"
+    ],
+    "화학적 변화": [
+        "표면 코팅 변화", "물질 성분 변화", "화학 반응 변화", "촉매 작용 시작/중단",
+        "빛에 의한 화학 반응", "전기에 의한 화학 반응", "단분자막 형성", "분자 수준 계산 변화",
+        "자연 모방 표면 변화", "환경 반응형 물질 변화", "주기적 화학 반응", "산화", "환원",
+        "고분자화", "물 분해", "화합", "방사선 영향", "산-염기 반응", "중화 반응",
+        "이온화", "화학적 흡착/탈착", "촉매 효율 변화", "효소 활성 변화", "발색 반응",
+        "pH 변화", "화학적 평형 이동", "결합 형성/분해", "용해도 변화"
+    ],
+
     "생물학적 변화": [
         "생장/위축", "세포 분열/사멸", "생물 발광", "신진대사 변화", "면역 반응",
         "호르몬 분비", "신경 반응", "유전적 발현", "적응/진화", "생체리듬 변화",
@@ -278,19 +314,8 @@ physical_transformation_categories = {
         "방사선 반응", "진동 감지", "pH 반응", "용매 반응", "기체 교환",
         "환경 오염 반응", "날씨 반응", "계절 변화 반응", "일주기 반응", "생태계 상호작용",
         "공생/경쟁 반응", "포식/피식 관계", "군집 형성", "영역 설정", "이주/정착 패턴"
-    ],
-    "센서 기능": [
-        "시각 센서/감지", "청각 센서/감지", "촉각 센서/감지", "미각 센서/감지", "후각 센서/감지",
-        "온도 센서/감지", "습도 센서/감지", "압력 센서/감지", "가속도 센서/감지", "회전 센서/감지",
-        "근접 센서/감지", "위치 센서/감지", "운동 센서/감지", "가스 센서/감지", "적외선 센서/감지",
-        "자외선 센서/감지", "방사선 센서/감지", "자기장 센서/감지", "전기장 센서/감지", "화학물질 센서/감지",
-        "생체신호 센서/감지", "진동 센서/감지", "소음 센서/감지", "빛 세기 센서/감지", "빛 파장 센서/감지",
-        "기울기 센서/감지", "pH 센서/감지", "전류 센서/감지", "전압 센서/감지", "이미지 센서/감지",
-        "거리 센서/감지", "깊��� 센서/감지", "중력 센서/감지", "속도 센서/감지", "흐름 센서/감지",
-        "수위 센서/감지", "탁도 센서/감지", "염도 센서/감지", "금속 감지", "압전 센서/감지",
-        "광전 센서/감지", "열전대 센서/감지", "홀 효과 센서/감지", "초음파 센서/감지", "레이더 센서/감지",
-        "라이다 센서/감지", "터치 센서/감지", "제스처 센서/감지", "심박 센서/감지", "혈압 센서/감지"
     ]
+
 }
 
 ##############################################################################
@@ -420,7 +445,6 @@ def process_inputs(text1, text2, text3, selected_category, progress=gr.Progress(
     time.sleep(0.3)
     progress(0.1, desc="창의적인 아이디어 생성 시작...")
     
-    # 카테고리에 해당하는 아이디어 생성
     results = generate_transformations(text1, text2, text3, selected_category)
     
     progress(0.8, desc="결과 포맷팅 중...")
@@ -497,7 +521,7 @@ with gr.Blocks(
     </style>
     """)
     
-    gr.Markdown("# 🚀 키워드 기반 창의적 변화 아이디어 및 디자인 생성기")
+    gr.Markdown("# 🚀 아이디어 트랜스포머")
     gr.Markdown("입력한 **키워드**(최대 3개)와 **카테고리**를 바탕으로, 창의적인 모델/컨셉/형상 변화 아이디어를 생성하고, 해당 확장 아이디어를 프롬프트로 하여 디자인 이미지를 생성합니다.")
     
     warning = gr.Markdown(get_warning_message())
@@ -507,7 +531,7 @@ 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_dropdown = gr.Dropdown(
+            category_radio = gr.Radio(
                 label="카테고리 선택",
                 choices=list(physical_transformation_categories.keys()),
                 value=list(physical_transformation_categories.keys())[0],
@@ -534,34 +558,30 @@ with gr.Blocks(
             idea_output = gr.Markdown(label="아이디어 결과")
             generated_image = gr.Image(label="생성된 디자인 이미지", type="pil")
     
-    # 예제
     gr.Examples(
         examples=[
-            ["스마트폰", "", "", list(physical_transformation_categories.keys())[0]],
-            ["자동차", "", "", list(physical_transformation_categories.keys())[0]],
-            ["자동차", "인공지능", "", list(physical_transformation_categories.keys())[0]],
-            ["드론", "인공지능", "", list(physical_transformation_categories.keys())[0]],
-            ["운동화", "웨어러블", "건강", list(physical_transformation_categories.keys())[0]],
+            ["스마트폰", "", "", "센서 기능"],
+            ["자동차", "", "", "크기와 형태 변화"],
+            ["자동차", "인공지능", "", "표면 및 외관 변화"],
+            ["드론", "인공지능", "", "물질의 상태 변화"],
+            ["운동화", "웨어러블", "건강", "구조적 변화"],
         ],
-        inputs=[text_input1, text_input2, text_input3, category_dropdown],
+        inputs=[text_input1, text_input2, text_input3, category_radio],
     )
     
-    # 처리중 아이콘 보이기
     def show_processing_indicator():
         return gr.update(visible=True)
     
-    # 처리중 아이콘 숨기기
     def hide_processing_indicator():
         return gr.update(visible=False)
     
-    # 버튼 클릭 시 처리 로직
     submit_button.click(
         fn=show_processing_indicator,
         inputs=None,
         outputs=processing_indicator
     ).then(
         fn=process_all,
-        inputs=[text_input1, text_input2, text_input3, category_dropdown],
+        inputs=[text_input1, text_input2, text_input3, category_radio],
         outputs=[idea_output, generated_image]
     ).then(
         fn=hide_processing_indicator,
@@ -569,6 +589,5 @@ with gr.Blocks(
         outputs=processing_indicator
     )
 
-# 메인 실행
 if __name__ == "__main__":
     demo.launch(debug=True)