Update app.py

app.py

@@ -8,9 +8,6 @@ translator = translation_pipeline("translation", model="Helsinki-NLP/opus-mt-ko-
 # Diffusers 모델 로드
 from diffusers import DiffusionPipeline
 
-# -- Stable Diffusion 계열 파이프라인 설정 --
-#  (모델 예시: black-forest-labs/FLUX.1-schnell -> 마인드맵용으로 커스텀된 모델이지만,
-#   여기서는 "스토리보드" 스타일 프롬프트도 시도 가능)
 model_id = "black-forest-labs/FLUX.1-schnell"
 
 pipe = DiffusionPipeline.from_pretrained(
@@ -18,10 +15,7 @@ pipe = DiffusionPipeline.from_pretrained(
     torch_dtype=torch.float32
 ).to("cpu")  # CPU 사용
 
-# 한글 프롬프트를 영어로 변환하기 위한 헬퍼 함수
 def translate_prompt_if_korean(prompt_text: str) -> str:
-    # 간단히, 문자열 내에 한글이 포함되어 있는지 확인 후 번역
-    # (영어 입력일 경우 번역을 스킵)
     if any("가" <= ch <= "힣" for ch in prompt_text):
         result = translator(prompt_text)
         return result[0]['translation_text']
@@ -53,7 +47,6 @@ def generate_storyboard(
         ).images[0]
     return result
 
-
 # --- 비주얼 & 세련된 UI를 위한 CSS ---
 custom_css = """
 #title {
@@ -94,7 +87,6 @@ custom_css = """
 }
 """
 
-# --- Gradio 인터페이스 구성 ---
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("<div id='title'>Gini Storyboard</div>")
     gr.Markdown("<div id='subtitle'>Generate a hand-drawn style storyboard in black & white film noir or any style you wish!</div>")
@@ -152,7 +144,8 @@ with gr.Blocks(css=custom_css) as demo:
 
     # 예제 프롬프트
     gr.Markdown("### Example Prompt")
-    with gr.Box(elem_classes="example-box"):
+    # ↓ Box가 아닌 Group으로 변경
+    with gr.Group(elem_classes="example-box"):
         example_text = (
             "A hand-drawn storyboard style, film noir theme, black and white.\n"
             "SCENE 1: A detective enters a dark alley [Frame 1]\n"
@@ -162,25 +155,23 @@ with gr.Blocks(css=custom_css) as demo:
         gr.Markdown(f"```\n{example_text}\n```")
         example_button = gr.Button("Use Example")
 
-    # 예제 버튼 클릭 시 프롬프트에 반영
     def load_example():
         return example_text
     example_button.click(fn=load_example, outputs=[prompt])
 
-    # 버튼 클릭 & 프롬프트 Enter 이벤트 처리
     run_button.click(
         fn=generate_storyboard,
         inputs=[prompt, width, height, num_inference_steps, guidance_scale, seed],
         outputs=[result]
     )
 
+    # 엔터로 제출 시에도 실행
     prompt.submit(
         fn=generate_storyboard,
         inputs=[prompt, width, height, num_inference_steps, guidance_scale, seed],
         outputs=[result]
     )
 
-# 실행
 if __name__ == "__main__":
     demo.queue()
     demo.launch(server_name="0.0.0.0", server_port=7860, share=False)