Update app.py

app.py

@@ -30,7 +30,6 @@ except Exception as e:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
 dtype = torch.bfloat16 if device == "cuda" else torch.float32
-
 print(f"Using device: {device}, dtype: {dtype}")
 
 ##############################################################################
@@ -42,18 +41,15 @@ try:
         model="Helsinki-NLP/opus-mt-ko-en",
         device=0 if device == "cuda" else -1
     )
-
     pipe = DiffusionPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-schnell",
         torch_dtype=dtype
     ).to(device)
-    
     print("Models loaded successfully")
 except Exception as e:
     print(f"Error loading models: {e}")
     def dummy_translator(text):
         return [{'translation_text': text}]
-    
     class DummyPipe:
         def __call__(self, **kwargs):
             from PIL import Image
@@ -63,7 +59,6 @@ except Exception as e:
                 def __init__(self, img):
                     self.images = [img]
             return DummyResult(dummy_img)
-    
     translator = dummy_translator
     pipe = DummyPipe()
 
@@ -102,25 +97,19 @@ def log_unexpected_characters(text):
 def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, height=1024, num_inference_steps=4):
     original_prompt = prompt
     translated = False
-    
     # Clean the input text
     prompt = clean_input_text(prompt)
-    
     # Pre-process: if input is too long, trim to 1000 characters
     if len(prompt) > 1000:
         prompt = prompt[:1000]
-    
     if contains_korean(prompt):
         # When calling translation, add max_length and truncation options to avoid length issues
         translation = translator(prompt, max_length=400, truncation=True)
         prompt = translation[0]['translation_text']
         translated = True
-
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    
     generator = torch.Generator(device=device).manual_seed(seed)
-    
     image = pipe(
         prompt=prompt,
         width=width,
@@ -129,7 +118,6 @@ def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, heig
         generator=generator,
         guidance_scale=0.0
     ).images[0]
-    
     return image
 
 ##############################################################################
@@ -257,257 +245,6 @@ physical_transformation_categories = {
         "광도 변화"
     ],
     "소리와 진동 효과": [
-        "소리 발생/소멸", "소리 높낮이 변화", "소리 크기 변화", "음색 변화",
-        "공명/반공명", "음향 진동", "초음파/저음파 발생", "음향 집중/분산",
-        "음향 반사/흡수", "음향 도플러 효과", "음파 간섭", "음향 공진",
-        "진동 패턴 변화", "타악 효과", "음향 피드백", "음향 차폐/증폭",
-        "소리 지향성", "음향 왜곡", "비트 생성", "하모닉스 생성", "주파수 변조",
-        "음향 충격파", "음향 필터링"
-    ],
-    "소리와 진동 효과": [
-        "소리 발생/소멸",
-        "음 높낮이 변화",
-        "음량 변화",
-        "음색 변화",
-        "공명/반공명",
-        "음향 진동",
-        "초음파/저음파 발생",
-        "소리 집중/분산",
-        "음향 반사/흡수",
-        "음향 도플러 효과",
-        "음파 간섭",
-        "음향 공진",
-        "진동 패턴 변화",
-        "타악 효과",
-        "음향 피드백",
-        "음향 차폐/증폭",
-        "소리 지향성",
-        "소리 왜곡",
-        "비트 생성",
-        "배음 생성",
-        "주파수 변조",
-        "음향 충격파",
-        "음향 필터링"
-    ],
-    "열 관련 변화": [
-        "온도 상승/하강",
-        "열 팽창/수축",
-        "열 전달/차단",
-        "압력 상승/하강",
-        "열 변화에 따른 자화",
-        "엔트로피 변화",
-        "열전기 효과",
-        "자기장에 의한 열 변화",
-        "상태 변화 중 열 저장/방출",
-        "열 스트레스 발생/해소",
-        "급격한 온도 변화 영향",
-        "복사 냉각/가열",
-        "발열/흡열",
-        "열 분포 변화",
-        "열 반사/흡수",
-        "냉각 응축",
-        "열 활성화",
-        "열 변색",
-        "열 팽창 계수 변화",
-        "열 안정성 변화",
-        "내열성/내한성",
-        "자가 발열",
-        "열적 평형/불균형",
-        "열적 변형",
-        "열 분산/집중"
-    ],
-    "전기 및 자기 변화": [
-        "자성 생성/소멸",
-        "전하량 증가/감소",
-        "전기장 생성/소멸",
-        "자기장 생성/소멸",
-        "초전도 상태 전환",
-        "강유전체 특성 변화",
-        "양자 상태 변화",
-        "플라즈마 형성/소멸",
-        "스핀파 전달",
-        "빛에 의한 전기 발생",
-        "압력에 의한 전기 발생",
-        "자기장 내 전류 변화",
-        "전기 저항 변화",
-        "전기 전도성 변화",
-        "정��기 발생/방전",
-        "전자기 유도",
-        "전자기파 방출/흡수",
-        "전기 용량 변화",
-        "자기 이력 현상",
-        "전기적 분극",
-        "전자 흐름 방향 변화",
-        "전기적 공명",
-        "전기적 차폐/노출",
-        "자기 차폐/노출",
-        "자기장 정렬"
-    ],
-    "화학적 변화": [
-        "표면 코팅 변화",
-        "물질 성분 변화",
-        "화학 반응 변화",
-        "촉매 작용 시작/중단",
-        "빛에 의한 화학 반응",
-        "전기에 의한 화학 반응",
-        "단분자막 형성",
-        "분자 수준 구조 변화",
-        "생체 모방 표면 변화",
-        "환경 반응형 물질 변화",
-        "주기적 화학 반응",
-        "산화",
-        "환원",
-        "고분자화",
-        "물 분해",
-        "화합",
-        "방사선 영향",
-        "산-염기 반응",
-        "중화 반응",
-        "이온화",
-        "화학적 흡착/탈착",
-        "촉매 효율 변화",
-        "효소 활성 변화",
-        "발색 반응",
-        "pH 변화",
-        "화학적 평형 이동",
-        "결합 형성/분해",
-        "용해도 변화"
-    ],
-    "생물학적 변화": [
-        "성장/위축",
-        "세포 분열/사멸",
-        "생물 발광",
-        "신진대사 변화",
-        "면역 반응",
-        "호르몬 분비",
-        "신경 반응",
-        "유전적 발현",
-        "적응/진화",
-        "생체리듬 변화",
-        "재생/치유",
-        "노화/성숙",
-        "생체 모방 변화",
-        "바이오필름 형성",
-        "생물학적 분해",
-        "효소 활성화/비활성화",
-        "생물학적 신호 전달",
-        "스트레스 반응",
-        "체온 조절",
-        "생물학적 시계 변화",
-        "세포외 기질 변화",
-        "생체 역학적 반응",
-        "세포 운동성",
-        "세포 극성 변화",
-        "영양 상태 변화"
-    ],
-    "환경 상호작용": [
-        "온도 반응",
-        "습도 반응",
-        "기압 반응",
-        "중력 반응",
-        "자기장 반응",
-        "빛 반응",
-        "소리 반응",
-        "화학 물질 감지",
-        "기계적 자극 감지",
-        "전기 자극 반응",
-        "방사선 반응",
-        "진동 감지",
-        "pH 반응",
-        "용매 반응",
-        "기체 교환",
-        "환경 오염 반응",
-        "날씨 반응",
-        "계절 반응",
-        "일주기 반응",
-        "생태계 상호작용",
-        "공생/경쟁 반응",
-        "포식/피식 관계",
-        "군집 형성",
-        "영역 설정",
-        "이주/정착 패턴"
-    ]
-}
-
-# English version
-physical_transformation_categories_en = {
-    "Sensor Functions": [
-        "Visual sensor/detection", "Auditory sensor/detection", "Tactile sensor/detection", "Taste sensor/detection", "Olfactory sensor/detection",
-        "Temperature sensor/detection", "Humidity sensor/detection", "Pressure sensor/detection", "Acceleration sensor/detection", "Rotational sensor/detection",
-        "Proximity sensor/detection", "Position sensor/detection", "Motion sensor/detection", "Gas sensor/detection", "Infrared sensor/detection",
-        "Ultraviolet sensor/detection", "Radiation sensor/detection", "Magnetic sensor/detection", "Electric field sensor/detection", "Chemical sensor/detection",
-        "Biosignal sensor/detection", "Vibration sensor/detection", "Noise sensor/detection", "Light intensity sensor/detection", "Light wavelength sensor/detection",
-        "Tilt sensor/detection", "pH sensor/detection", "Current sensor/detection", "Voltage sensor/detection", "Image sensor/detection",
-        "Distance sensor/detection", "Depth sensor/detection", "Gravity sensor/detection", "Speed sensor/detection", "Flow sensor/detection",
-        "Water level sensor/detection", "Turbidity sensor/detection", "Salinity sensor/detection", "Metal detection", "Piezoelectric sensor/detection",
-        "Photovoltaic sensor/detection", "Thermocouple sensor/detection", "Hall effect sensor/detection", "Ultrasonic sensor/detection", "Radar sensor/detection",
-        "Lidar sensor/detection", "Touch sensor/detection", "Gesture sensor/detection", "Heart rate sensor/detection", "Blood pressure sensor/detection"
-    ],
-    "Size and Shape Change": [
-        "Volume increase/decrease", "Length increase/decrease", "Width increase/decrease", "Height increase/decrease",
-        "Density change", "Weight increase/decrease", "Shape deformation", "State change", "Uneven deformation",
-        "Complex shape deformation", "Twisting/entwining", "Non-uniform expansion/contraction", "Rounded/sharpened edges",
-        "Cracking/splitting", "Fragmentation", "Water resistance", "Dust resistance", "Denting/recovery",
-        "Folding/unfolding", "Compression/expansion", "Stretching/contraction", "Wrinkling/flattening", "Crushing/hardening",
-        "Rolling/unrolling", "Bending/curving"
-    ],
-    "Surface and Appearance Change": [
-        "Color change", "Texture change", "Transparency change", "Glossy/matte change",
-        "Light reflection variation", "Pattern change", "Angle-dependent color change", "Light-induced color change",
-        "Temperature-dependent color change", "Holographic effect", "Angle-specific light reflection", "Surface shape alteration",
-        "Nano-scale surface structure change", "Self-cleaning effect", "Stain/pattern formation", "Blurriness/clarity change",
-        "Luster/shine change", "Hue/saturation change", "Luminescence/fluorescence", "Light scattering effect",
-        "Light absorption change", "Translucency effect", "Shadow effect change", "UV response change",
-        "Glow effect"
-    ],
-    "Material State Change": [
-        "Solid/liquid/gas transition", "Crystallization/dissolution", "Oxidation/corrosion", "Hardening/softening",
-        "Special state transition", "Amorphous/crystalline transition", "Component separation", "Particle formation/disintegration",
-        "Gel formation/dissolution", "Metastable state change", "Molecular self-assembly/disintegration", "Delayed state change",
-        "Melting", "Solidification", "Evaporation/condensation", "Sublimation/deposition", "Precipitation/suspension", "Dispersion/aggregation",
-        "Drying/moistening", "Swelling/shrinkage", "Freezing/thawing", "Weathering/erosion", "Charging/discharging",
-        "Bonding/separation", "Fermentation/decay"
-    ],
-    "Movement Characteristics Change": [
-        "Acceleration/deceleration", "Maintaining constant speed", "Vibration/vibration reduction", "Collision/bouncing",
-        "Increase/decrease in rotational speed", "Change in rotational direction", "Irregular movement", "Stop-and-slide phenomenon",
-        "Resonance/anti-resonance", "Resistance/lift change in fluid", "Change in movement resistance", "Complex vibrational movement",
-        "Movement in special fluid", "Rotational-translational movement", "Inertial stoppage", "Shock absorption",
-        "Shock transfer", "Conservation of momentum", "Friction change", "Overcoming inertia", "Unstable equilibrium",
-        "Dynamic stability", "Damping of oscillation", "Path predictability", "Evasive movement"
-    ],
-    "Structural Change": [
-        "Addition/removal of components", "Assembly/disassembly", "Folding/unfolding", "Deformation/recovery", "Optimal structural change",
-        "Self-rearrangement", "Natural pattern formation/disappearance", "Regular pattern change", "Modular transformation",
-        "Increased structural complexity", "Memory of original shape effect", "Shape change over time", "Partial removal",
-        "Partial replacement", "Bonding", "Separation", "Division/integration", "Overlaying", "Internal structure change",
-        "External structure change", "Shift of center axis", "Balance point change", "Hierarchical structure change", "Support structure change",
-        "Stress distribution structure", "Shock absorption structure", "Grid/matrix structure change", "Interconnectivity change"
-    ],
-    "Spatial Movement": [
-        "Forward/backward movement", "Left/right movement", "Up/down movement", "Vertical axis rotation (nodding)",
-        "Horizontal axis rotation (shaking head)", "Longitudinal axis rotation (tilting sideways)", "Circular motion", "Spiral movement",
-        "Slipping due to inertia", "Change of rotation axis", "Irregular rotation", "Shaking movement", "Parabolic motion",
-        "Zero-gravity floating", "Floating on water surface", "Jump/leap", "Sliding", "Rolling", "Free fall",
-        "Reciprocating motion", "Elastic bouncing", "Penetration", "Evasive movement", "Zigzag movement", "Swinging movement"
-    ],
-    "Time-Related Change": [
-        "Aging/weathering", "Wear/corrosion", "Fading/discoloration", "Damage/recovery", "Lifecycle change",
-        "Adaptation through user interaction", "Learning-based shape optimization", "Property change over time",
-        "Collective memory effect", "Cultural significance change", "Delayed response", "History-dependent change",
-        "Gradual time change", "Evolutionary change", "Periodic regeneration", "Seasonal adaptation",
-        "Circadian rhythm change", "Lifecycle stage", "Growth/decline", "Self-repair/regeneration",
-        "Natural cycle adaptation", "Persistence/transience", "Memory effect", "Delayed effect", "Cumulative effect"
-    ],
-    "Light and Visual Effects": [
-        "Illumination/shutdown", "Light transmission/blocking", "Light scattering/concentration", "Color spectrum change", "Light diffraction",
-        "Light interference", "Hologram creation", "Laser effect", "Light polarization", "Fluorescence/phosphorescence",
-        "UV/IR emission", "Optical illusion", "Light refraction", "Shadow creation/removal",
-        "Chromatic aberration", "Rainbow effect", "Glow effect", "Flash effect", "Lighting pattern",
-        "Beam effect", "Light filter effect", "Change in light direction", "Projection effect", "Light detection/response",
-        "Luminance change"
-    ],
-    "Sound and Vibration Effects": [
         "Sound generation/cessation", "Pitch change", "Volume change", "Timbre change",
         "Resonance/antiresonance", "Acoustic vibration", "Ultrasonic/infrasonic emission", "Sound concentration/distribution",
         "Sound reflection/absorption", "Acoustic Doppler effect", "Sound wave interference", "Acoustic resonance",
@@ -696,10 +433,8 @@ def process_inputs_lang(text1, text2, text3, selected_category, categories_dict,
     text1 = text1.strip() if text1 else None
     text2 = text2.strip() if text2 else None
     text3 = text3.strip() if text3 else None
-    
     if not text1:
         return "오류: 최소 하나의 키워드를 입력해주세요." if lang=="ko" else "Error: Please enter at least one keyword."
-    
     if lang == "ko":
         progress(0.05, desc="아이디어 생성 준비 중...")
         time.sleep(0.3)
@@ -708,9 +443,7 @@ def process_inputs_lang(text1, text2, text3, selected_category, categories_dict,
         progress(0.05, desc="Preparing idea generation...")
         time.sleep(0.3)
         progress(0.1, desc="Generating creative idea...")
-    
     results = generate_transformations_lang(text1, text2, text3, selected_category, categories_dict, lang)
-    
     if lang == "ko":
         progress(0.8, desc="결과 포맷팅 중...")
         formatted = format_results_lang(results, lang)
@@ -741,6 +474,41 @@ def get_warning_message_lang(lang="ko"):
         return "⚠️ 환경 변수 GEMINI_API_KEY가 설정되지 않았습니다. Gemini API 키를 설정하세요." if lang=="ko" else "⚠️ The GEMINI_API_KEY environment variable is not set. Please set your Gemini API key."
     return ""
 
+##############################################################################
+# Add "Business Ideas" category to each dictionary
+##############################################################################
+physical_transformation_categories["비즈니스 아이디어"] = [
+    "시장 재정의/신규 시장 개척",
+    "비즈니스 모델 혁신/디지털 전환",
+    "고객 경험 혁신/서비스 혁신",
+    "협력 및 파트너십 강화/생태계 구축",
+    "글로벌 확장/지역화 전략",
+    "운영 효율성 증대/원가 절감",
+    "브랜드 리포지셔닝/이미지 전환",
+    "지속 가능한 성장/사회적 가치 창출",
+    "데이터 기반 의사결정/AI 도입",
+    "신기술 융합/혁신 투자"
+]
+
+physical_transformation_categories_en["Business Ideas"] = [
+    "Market redefinition/new market creation",
+    "Business model innovation/digital transformation",
+    "Customer experience and service innovation",
+    "Enhanced partnerships and ecosystem building",
+    "Global expansion and localization strategy",
+    "Operational efficiency improvement/cost reduction",
+    "Brand repositioning/image transformation",
+    "Sustainable growth and social value creation",
+    "Data-driven decision making/AI adoption",
+    "Convergence of new technologies/innovative investments"
+]
+
+##############################################################################
+# Helper function for caching examples in the English tab
+##############################################################################
+def process_all_lang_example(text1, text2, text3, selected_category):
+    # 고정된 state값(physical_transformation_categories_en, "en")으로 호출
+    return process_all_lang(text1, text2, text3, selected_category, physical_transformation_categories_en, "en")
 
 ##############################################################################
 # Gradio UI with Two Tabs: English (Main Home) and Korean
@@ -749,7 +517,6 @@ with gr.Blocks(
     title="Idea Transformer",
     theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", neutral_hue="neutral")
 ) as demo:
-    
     gr.HTML("""
     <style>
     body { 
@@ -792,9 +559,7 @@ with gr.Blocks(
         with gr.Tab(label="English"):
             gr.Markdown("# 🚀 Idea Transformer")
             gr.Markdown("Based on up to **three keywords** and a **selected category**, this tool generates a creative transformation idea and a design image using the expanded idea as a prompt. https://discord.gg/openfreeai")
-            
             warning_en = gr.Markdown(get_warning_message_lang("en"))
-            
             with gr.Row():
                 with gr.Column(scale=1):
                     text_input1_en = gr.Textbox(label="Keyword 1 (required)", placeholder="e.g., Smartphone")
@@ -807,7 +572,6 @@ with gr.Blocks(
                         info="Select a category."
                     )
                     status_msg_en = gr.Markdown("💡 Click the 'Generate Idea' button to create an idea and design image based on the selected category.")
-                    
                     processing_indicator_en = gr.HTML("""
                     <div style="display: flex; justify-content: center; align-items: center; margin: 10px 0;">
                         <div style="border: 5px solid #f3f3f3; border-top: 5px solid #3498db; border-radius: 50%; width: 30px; height: 30px; animation: spin 2s linear infinite;"></div>
@@ -820,13 +584,10 @@ with gr.Blocks(
                     }
                     </style>
                     """, visible=False)
-                    
                     submit_button_en = gr.Button("Generate Idea", variant="primary")
-                
                 with gr.Column(scale=2):
                     idea_output_en = gr.Markdown(label="Idea Output")
                     generated_image_en = gr.Image(label="Generated Design Image", type="pil")
-            
             gr.Examples(
                 examples=[
                     ["Smartphone", "", "", "Sensor Functions"],
@@ -836,14 +597,14 @@ with gr.Blocks(
                     ["Sneakers", "Wearable", "Health", "Structural Change"],
                 ],
                 inputs=[text_input1_en, text_input2_en, text_input3_en, category_radio_en],
+                fn=process_all_lang_example,
+                outputs=[idea_output_en, generated_image_en],
+                cache_examples=True
             )
-            
             def show_processing_indicator_en():
                 return gr.update(visible=True)
-            
             def hide_processing_indicator_en():
                 return gr.update(visible=False)
-            
             submit_button_en.click(
                 fn=show_processing_indicator_en,
                 inputs=None,
@@ -857,13 +618,10 @@ with gr.Blocks(
                 inputs=None,
                 outputs=processing_indicator_en
             )
-        
         with gr.Tab(label="한국어"):
             gr.Markdown("# 🚀 아이디어 트랜스포머")
             gr.Markdown("입력한 **키워드**(최대 3개)와 **카테고리**를 바탕으로, 창의적인 모델/컨셉/형상 변화 아이디어를 생성하고, 해당 확장 아이디어를 프롬프트로 하여 디자인 이미지를 생성합니다. https://discord.gg/openfreeai")
-            
             warning_ko = gr.Markdown(get_warning_message_lang("ko"))
-            
             with gr.Row():
                 with gr.Column(scale=1):
                     text_input1_ko = gr.Textbox(label="키워드 1 (필수)", placeholder="예: 스마트폰")
@@ -876,7 +634,6 @@ with gr.Blocks(
                         info="출력할 카테고리를 선택하세요."
                     )
                     status_msg_ko = gr.Markdown("💡 '아이디어 생성하기' 버튼을 클릭하면 선택한 카테고리에 해당하는 아이디어와 디자인 이미지가 생성됩니다.")
-                    
                     processing_indicator_ko = gr.HTML("""
                     <div style="display: flex; justify-content: center; align-items: center; margin: 10px 0;">
                         <div style="border: 5px solid #f3f3f3; border-top: 5px solid #3498db; border-radius: 50%; width: 30px; height: 30px; animation: spin 2s linear infinite;"></div>
@@ -889,13 +646,10 @@ with gr.Blocks(
                     }
                     </style>
                     """, visible=False)
-                    
                     submit_button_ko = gr.Button("아이디어 생성하기", variant="primary")
-                
                 with gr.Column(scale=2):
                     idea_output_ko = gr.Markdown(label="아이디어 결과")
                     generated_image_ko = gr.Image(label="생성된 디자인 이미지", type="pil")
-            
             gr.Examples(
                 examples=[
                     ["스마트폰", "", "", "센서 기능"],
@@ -904,15 +658,12 @@ with gr.Blocks(
                     ["드론", "인공지능", "", "물질의 상태 변화"],
                     ["운동화", "웨어러블", "건강", "구조적 변화"],
                 ],
-                inputs=[text_input1_ko, text_input2_ko, text_input3_ko, category_radio_ko],
+                inputs=[text_input1_ko, text_input2_ko, text_input3_ko, category_radio_ko]
             )
-            
             def show_processing_indicator_ko():
                 return gr.update(visible=True)
-            
             def hide_processing_indicator_ko():
                 return gr.update(visible=False)
-            
             submit_button_ko.click(
                 fn=show_processing_indicator_ko,
                 inputs=None,