Update app.py

app.py

@@ -4,11 +4,11 @@ import sys
 import importlib.util
 import time
 
-# 필요한 환경 변수 설정
+# Set environment variables
 os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "1"
 os.environ["TRANSFORMERS_COMPILER_DISABLED"] = "1"
 
-# 필요한 모듈 설치 함수
+# Function to install required packages
 def install_required_packages():
     required_packages = [
         "warmup_scheduler",
@@ -129,10 +129,6 @@ import gradio as gr
 import spaces
 from huggingface_hub import hf_hub_url
 from huggingface_hub import hf_hub_download
-from transformers import pipeline
-
-# Initialize the translation pipeline
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -164,26 +160,6 @@ def load_message(height, width, seed, prompt, args, stage_a_tiled):
     args.stage_a_tiled = stage_a_tiled
     return args
 
-def is_korean(text):
-    return any('\uac00' <= char <= '\ud7a3' for char in text)
-
-def translate_if_korean(text, translation_display=None):
-    if is_korean(text):
-        translated = translator(text, max_length=512)[0]['translation_text']
-        print(f"Translated from Korean: {text} -> {translated}")
-        
-        # If translation_display is provided, update it to show the translation
-        if translation_display is not None:
-            translation_display.update(value=f"Original (Korean): {text}\nTranslation (English): {translated}", visible=True)
-            
-        return translated
-    
-    # If not Korean, hide the translation display if it exists
-    if translation_display is not None:
-        translation_display.update(visible=False)
-    
-    return text
-
 @spaces.GPU(duration=120)
 def get_image(height, width, seed, prompt, cfg, timesteps, stage_a_tiled):
     global args
@@ -242,13 +218,6 @@ def get_image(height, width, seed, prompt, cfg, timesteps, stage_a_tiled):
                     
     return imgs[0]
 
-def process_and_generate(height, width, seed, prompt, cfg, timesteps, stage_a_tiled, translation_display):
-    # First translate if needed and show the translation
-    translated_prompt = translate_if_korean(prompt, translation_display)
-    
-    # Then generate the image
-    return get_image(height, width, seed, translated_prompt, cfg, timesteps, stage_a_tiled)
-
 css = """
 footer {
     visibility: hidden;
@@ -347,17 +316,6 @@ input[type="text"]:focus {
     box-shadow: 0 5px 15px rgba(0, 0, 0, 0.2);
 }
 
-/* Language indicator */
-.language-indicator {
-    background-color: #e6f7ff;
-    color: #0072b5;
-    border-radius: 20px;
-    padding: 2px 10px;
-    font-size: 0.8rem;
-    margin-left: 10px;
-    display: inline-block;
-}
-
 /* Loading animation */
 @keyframes pulse {
     0% { opacity: 0.6; }
@@ -384,7 +342,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                     prompt = gr.Textbox(
                         label="Text Prompt",
                         max_lines=2,
-                        placeholder="Enter your prompt in Korean or English (한국어나 영어로 프롬프트를 입력하세요)",
+                        placeholder="Enter your prompt (e.g., 'A majestic mountain landscape with snow')",
                         elem_id="prompt-input"
                     )
                 
@@ -392,17 +350,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                     generate_button = gr.Button("Generate Image", variant="primary", elem_id="generate-btn")
                     clear_button = gr.Button("Clear", elem_id="clear-btn")
         
-        # Language detection indicator
-        with gr.Row(visible=False) as language_indicator:
-            gr.Markdown('<div class="language-indicator">Korean detected - will translate to English</div>')
-        
-        # Translation display component
-        with gr.Row(visible=False) as translation_display:
-            translation_text = gr.Textbox(
-                label="Translation Information",
-                interactive=False
-            )
-        
         # Loading indicator
         with gr.Row(visible=False) as loading_indicator:
             gr.Markdown('<div class="loading">Generating your ultra high resolution image... This may take a minute...</div>')
@@ -476,15 +423,15 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                 gr.TabItem("Nature", gr.Examples(
                     examples=[
                         "A detailed view of a blooming magnolia tree, with large, white flowers and dark green leaves, set against a clear blue sky.",
-                        "눈 덮인 산맥의 장엄한 전경, 푸른 하늘을 배경으로 한 고요한 호수가 있는 모습",
+                        "A majestic view of snow-covered mountains with a calm lake against a blue sky background",
                     ],
                     inputs=[prompt],
                     outputs=[output_img],
                 )),
                 gr.TabItem("Animals", gr.Examples(
                     examples=[
-                        "스웨터를 입은 악어",
-                        "골든 리트리버 강아지가 푸른 잔디밭에서 빨간 공을 쫓는 귀여운 모습",
+                        "A crocodile wearing a sweater",
+                        "A cute golden retriever puppy chasing a red ball on a green lawn",
                     ],
                     inputs=[prompt],
                     outputs=[output_img],
@@ -524,20 +471,13 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
         preset_square.click(set_square, outputs=[width, height])
         preset_ultrawide.click(set_ultrawide, outputs=[width, height])
         
-        # Function to show Korean detection indicator
-        def check_korean(text):
-            if any('\uac00' <= char <= '\ud7a3' for char in text):
-                return gr.update(visible=True)
-            return gr.update(visible=False)
-            
         # Connect events
-        prompt.change(check_korean, inputs=[prompt], outputs=[language_indicator])
         generate_button.click(
             lambda: gr.update(visible=True),
             outputs=[loading_indicator]
         ).then(
-            process_and_generate,
-            inputs=[height, width, seed, prompt, cfg, timesteps, stage_a_tiled, translation_display],
+            get_image,
+            inputs=[height, width, seed, prompt, cfg, timesteps, stage_a_tiled],
             outputs=[output_img]
         ).then(
             lambda: gr.update(visible=False),