Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
 huggingface_token = os.getenv("HF_TOKEN")
 if not huggingface_token:
-    print("경고: Hugging Face 토큰이 설정되지 않았습니다.")
-    
+    print("Warning: Hugging Face token is not set.")
+
 import gradio as gr
 import json
 import logging
@@ -23,19 +23,15 @@ from gradio_imageslider import ImageSlider
 import numpy as np
 import warnings
 
-
-
-
-
 try:
     translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cpu", token=huggingface_token)
 except Exception as e:
-    print(f"번역기 로드 실패: {str(e)}")
-    # 번역 실패시 원본 텍스트를 그대로 반환하는 대체 함수
+    print(f"Translation model load failed: {str(e)}")
+    # If the translation model fails to load, return the original text
     def translator(text, max_length=512):
         return [{'translation_text': text}]
         
-#Load prompts for randomization
+# Load prompts for randomization
 df = pd.read_csv('prompts.csv', header=None)
 prompt_values = df.values.flatten()
 
@@ -43,20 +39,18 @@ prompt_values = df.values.flatten()
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
-# Initialize the base model
+# Load base FLUX model
 dtype = torch.bfloat16
-
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# 공통 FLUX 모델 로드
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to(device)
 
-# LoRA를 위한 설정
+# Settings for LoRA
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
 good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
 
-# Image-to-Image 파이프라인 설정
+# Set up image-to-image pipeline
 pipe_i2i = AutoPipelineForImage2Image.from_pretrained(
     base_model,
     vae=good_vae,
@@ -276,14 +270,15 @@ def add_custom_lora(custom_lora, selected_indices, current_loras):
                 lora_scale_3,
                 lora_image_1,
                 lora_image_2,
-                lora_image_3
+                lora_image_3,
+                gr.update(visible=True)  # Make "Remove Custom LoRA" button visible
             )
         except Exception as e:
             print(e)
             gr.Warning(str(e))
-            return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+            return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
     else:
-        return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
+        return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
 
 def remove_custom_lora(selected_indices, current_loras):
     if current_loras:
@@ -318,6 +313,8 @@ def remove_custom_lora(selected_indices, current_loras):
         lora3 = current_loras[selected_indices[2]]
         selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}]({lora3['repo']}) ✨"
         lora_image_3 = lora3['image']
+    # If no custom LoRA remains, hide the "Remove Custom LoRA" button
+    remove_button_visibility = gr.update(visible=False) if not any("custom" in lora['repo'] for lora in current_loras) else gr.update(visible=True)
     return (
         current_loras,
         gr.update(value=gallery_items),
@@ -330,7 +327,8 @@ def remove_custom_lora(selected_indices, current_loras):
         lora_scale_3,
         lora_image_1,
         lora_image_2,
-        lora_image_3
+        lora_image_3,
+        remove_button_visibility
     )
 
 @spaces.GPU(duration=75)
@@ -339,7 +337,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     with calculateDuration("Generating image"):
-        # Generate image
+        # Generate image iteratively
         for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt_mash,
             num_inference_steps=steps,
@@ -374,25 +372,23 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
 
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
     try:
-        # 한글 감지 및 번역 부분
+        # Detect and translate Korean text if present
         if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
             try:
                 translated = translator(prompt, max_length=512)[0]['translation_text']
-                print(f"원본 프롬프트: {prompt}")
-                print(f"번역된 프롬프트: {translated}")
+                print(f"Original prompt: {prompt}")
+                print(f"Translated prompt: {translated}")
                 prompt = translated
             except Exception as e:
-                print(f"번역 실패: {str(e)}")
-                # 번역 실패시 원본 프롬프트 사용
-
-
+                print(f"Translation failed: {str(e)}")
+                # Use the original prompt if translation fails
 
         if not selected_indices:
             raise gr.Error("You must select at least one LoRA before proceeding.")
 
         selected_loras = [loras_state[idx] for idx in selected_indices]
 
-        # Build the prompt with trigger words (이 부분은 그대로 유지)
+        # Build the prompt with trigger words
         prepends = []
         appends = []
         for lora in selected_loras:
@@ -451,7 +447,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
 
         print(f"Active adapters after loading: {pipe.get_active_adapters()}")
 
-        # 여기서부터 이미지 생성 로직 (이 부분은 그대로 유지)
+        # Randomize seed if requested
         with calculateDuration("Randomizing seed"):
             if randomize_seed:
                 seed = random.randint(0, MAX_SEED)
@@ -536,6 +532,7 @@ def update_history(new_image, history):
         history.insert(0, new_image)
     return history
 
+# Custom theme configuration
 custom_theme = gr.themes.Base(
     primary_hue="blue",
     secondary_hue="purple",
@@ -574,7 +571,7 @@ custom_theme = gr.themes.Base(
 )
 
 css = '''
-/* 기본 버튼 및 컴포넌트 스타일 */
+/* Basic button and component styles */
 #gen_btn {
     height: 100%
 }
@@ -596,11 +593,11 @@ css = '''
 
 #lora_list {
     background: var(--block-background-fill);
-    padding: 0 1em .3em; 
+    padding: 0 1em 0.3em; 
     font-size: 90%
 }
 
-/* 커스텀 LoRA 카드 스타일 */
+/* Custom LoRA card styles */
 .custom_lora_card {
     margin-bottom: 1em
 }
@@ -608,19 +605,19 @@ css = '''
 .card_internal {
     display: flex;
     height: 100px;
-    margin-top: .5em
+    margin-top: 0.5em
 }
 
 .card_internal img {
     margin-right: 1em
 }
 
-/* 유틸리티 클래스 */
+/* Utility classes */
 .styler {
     --form-gap-width: 0px !important
 }
 
-/* 프로그레스 바 스타일 */
+/* Progress bar styles */
 #progress {
     height: 30px;
     width: 90% !important;
@@ -647,7 +644,7 @@ css = '''
     transition: width 0.5s ease-in-out
 }
 
-/* 컴포넌트 특정 스타일 */
+/* Component-specific styles */
 #component-8, .button_total {
     height: 100%; 
     align-self: stretch;
@@ -674,7 +671,7 @@ css = '''
     align-self: stretch;
 }
 
-/* 갤러리 메인 스타일 */
+/* Gallery main styles */
 #lora_gallery {
     margin: 20px 0;
     padding: 10px;
@@ -687,7 +684,7 @@ css = '''
     display: block !important;
 }
 
-/* 갤러리 그리드 스타일 */
+/* Gallery grid styles */
 #gallery {
     display: grid !important;
     grid-template-columns: repeat(10, 1fr) !important;
@@ -699,7 +696,7 @@ css = '''
     max-width: 100% !important;
 }
 
-/* 갤러리 아이템 스타일 */
+/* Gallery item styles */
 .gallery-item {
     position: relative !important;
     width: 100% !important;
@@ -718,7 +715,7 @@ css = '''
     border-radius: 12px !important;
 }
 
-/* 갤러리 그리드 래퍼 */
+/* Gallery grid wrapper */
 .wrap, .svelte-w6dy5e {
     display: grid !important;
     grid-template-columns: repeat(10, 1fr) !important;
@@ -727,7 +724,7 @@ css = '''
     max-width: 100% !important;
 }
 
-/* 컨테이너 공통 스타일 */
+/* Common container styles */
 .container, .content, .block, .contain {
     width: 100% !important;
     max-width: 100% !important;
@@ -741,7 +738,7 @@ css = '''
     padding: 0 !important;
 }
 
-/* 버튼 스타일 */
+/* Button styles */
 .button_total {
     box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
     transition: all 0.3s ease;
@@ -752,7 +749,7 @@ css = '''
     box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05);
 }
 
-/* 입력 필드 스타일 */
+/* Input field styles */
 input, textarea {
     box-shadow: inset 0 2px 4px 0 rgba(0, 0, 0, 0.06);
     transition: all 0.3s ease;
@@ -762,14 +759,14 @@ input:focus, textarea:focus {
     box-shadow: 0 0 0 3px rgba(66, 153, 225, 0.5);
 }
 
-/* 컴포넌트 border-radius */
+/* Component border-radius */
 .gradio-container .input, 
 .gradio-container .button,
 .gradio-container .block {
     border-radius: 12px;
 }
 
-/* 스크롤바 스타일 */
+/* Scrollbar styles */
 #gallery::-webkit-scrollbar {
     width: 8px;
 }
@@ -788,57 +785,57 @@ input:focus, textarea:focus {
     background: #555;
 }
 
-/* Flex 컨테이너 */
+/* Flex container */
 .flex {
     width: 100% !important;
     max-width: 100% !important;
     display: flex !important;
 }
 
-/* Svelte 특정 클래스 */
+/* Svelte specific classes */
 .svelte-1p9xokt {
     width: 100% !important;
     max-width: 100% !important;
 }
 
-/* Footer 숨김 */
+/* Hide Footer */
 #footer {
     visibility: hidden;
 }
 
-/* 결과 이미지 및 컨테이너 스타일 */
+/* Generated image and container styles */
 #result_column, #result_column > div {
     display: flex !important;
     flex-direction: column !important;
-    align-items: flex-start !important;  /* center에서 flex-start로 변경 */
+    align-items: flex-start !important;
     width: 100% !important;
-    margin: 0 !important;  /* auto에서 0으로 변경 */
+    margin: 0 !important;
 }
 
 .generated-image, .generated-image > div {
     display: flex !important;
-    justify-content: flex-start !important;  /* center에서 flex-start로 변경 */
-    align-items: flex-start !important;  /* center에서 flex-start로 변경 */
+    justify-content: flex-start !important;
+    align-items: flex-start !important;
     width: 90% !important;
     max-width: 768px !important;
-    margin: 0 !important;  /* auto에서 0으로 변경 */
-    margin-left: 20px !important;  /* 왼쪽 여백 추가 */
+    margin: 0 !important;
+    margin-left: 20px !important;
 }
 
 .generated-image img {
-    margin: 0 !important;  /* auto에서 0으로 변경 */
+    margin: 0 !important;
     display: block !important;
     max-width: 100% !important;
 }
 
-/* 히스토리 갤러리도 좌측 정렬로 변경 */
+/* History gallery left alignment */
 .history-gallery {
     display: flex !important;
-    justify-content: flex-start !important;  /* center에서 flex-start로 변경 */
+    justify-content: flex-start !important;
     width: 90% !important;
     max-width: 90% !important;
-    margin: 0 !important;  /* auto에서 0으로 변경 */
-    margin-left: 20px !important;  /* 왼쪽 여백 추가 */
+    margin: 0 !important;
+    margin-left: 20px !important;
 }
 '''
 
@@ -848,9 +845,11 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
 
     gr.Markdown(
         """
-        # GiniGen: 멀티 Lora(이미지 학습) 통합 생성 모델
-        ### 사용 안내:
-        갤러리에서 원하는 모델을 선택(최대 3개까지) < 프롬프트에 한글 또는 영문으로 원하는 내용을 입력 < Generate 버튼 실행  
+        # GiniGen: Multi-LoRA (Image Training) Integrated Generation Model
+        ### Instructions:
+        Select a model from the gallery (up to 3 models) &nbsp;&nbsp;&nbsp;|&nbsp;&nbsp;&nbsp;
+        Enter your prompt in Korean or English &nbsp;&nbsp;&nbsp;|&nbsp;&nbsp;&nbsp;
+        Click the **Generate** button
         """
     )
 
@@ -922,7 +921,7 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
         # Result and Progress Area
         with gr.Column(elem_id="result_column"):
             progress_bar = gr.Markdown(elem_id="progress", visible=False)
-            with gr.Column(elem_id="result_box"):  # Box를 Column으로 변경
+            with gr.Column(elem_id="result_box"):
                 result = gr.Image(
                     label="Generated Image", 
                     interactive=False,
@@ -940,12 +939,11 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
                     elem_classes=["history-gallery"]
                 )
 
-
         # Advanced Settings
         with gr.Row():
             with gr.Accordion("Advanced Settings", open=False):
                 with gr.Row():
-                    input_image = gr.Image(label="Input image", type="filepath")
+                    input_image = gr.Image(label="Input Image", type="filepath")
                     image_strength = gr.Slider(label="Denoise Strength", info="Lower means more image influence", minimum=0.1, maximum=1.0, step=0.01, value=0.75)
                 with gr.Column():
                     with gr.Row():
@@ -955,7 +953,7 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
                         width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
                         height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
                     with gr.Row():
-                        randomize_seed = gr.Checkbox(True, label="Randomize seed")
+                        randomize_seed = gr.Checkbox(True, label="Randomize Seed")
                         seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
 
         # Custom LoRA Section
@@ -1013,7 +1011,7 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
             inputs=[custom_lora, selected_indices, loras_state],
             outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, 
                 selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, 
-                lora_image_1, lora_image_2, lora_image_3]
+                lora_image_1, lora_image_2, lora_image_3, remove_custom_lora_button]
         )
 
         remove_custom_lora_button.click(
@@ -1021,7 +1019,7 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
             inputs=[selected_indices, loras_state],
             outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, 
                 selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, 
-                lora_image_1, lora_image_2, lora_image_3]
+                lora_image_1, lora_image_2, lora_image_3, remove_custom_lora_button]
         )
 
         gr.on(
@@ -1039,4 +1037,4 @@ with gr.Blocks(theme=custom_theme, css=css, delete_cache=(60, 3600)) as app:
 
 if __name__ == "__main__":
     app.queue(max_size=20)
-    app.launch(debug=True)
+    app.launch(debug=True)