Update app.py

app.py

@@ -12,7 +12,9 @@ import torch
 from diffusers import FluxPipeline
 from PIL import Image
 from transformers import pipeline
+
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
 # Hugging Face 토큰 설정
 HF_TOKEN = os.getenv("HF_TOKEN")
 if HF_TOKEN is None:
@@ -100,7 +102,6 @@ def save_image(image):
         print(f"Error in save_image: {str(e)}")
         return None
 
-
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
@@ -208,21 +209,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     def update_seed():
         return get_random_seed()
 
-generate_btn.click(
-    process_and_save_image,
-    inputs=[height, width, steps, scales, prompt, seed],
-    outputs=output
-)
-
-randomize_seed.click(
-    update_seed,
-    outputs=[seed]
-)
+    # Click event handlers inside gr.Blocks context
+    generate_btn.click(
+        process_and_save_image,
+        inputs=[height, width, steps, scales, prompt, seed],
+        outputs=output
+    ).then(
+        update_seed,
+        outputs=[seed]
+    )
 
-generate_btn.click(
-    update_seed,
-    outputs=[seed]
-)
+    randomize_seed.click(
+        update_seed,
+        outputs=[seed]
+    )
 
 if __name__ == "__main__":
     demo.launch(allowed_paths=[PERSISTENT_DIR])