Update app.py

app.py

@@ -151,28 +151,38 @@ def generate_image(
     progress: gr.Progress = gr.Progress()
 ):
     try:
-        # 프롬프트 번역 추가
-        prompt = translate_to_english(prompt)
+        print(f"\nStarting image generation with prompt: {prompt}")
+        
+        # 프롬프트 번역
+        translated_prompt = translate_to_english(prompt)
+        print(f"Translated prompt: {translated_prompt}")
         
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
+        print(f"Using seed: {seed}")
         
         generator = torch.Generator(device=device).manual_seed(seed)
         
+        print("Starting inference...")
         with torch.inference_mode():
             image = pipe(
-                prompt=prompt,
+                prompt=translated_prompt,
                 width=width,
                 height=height,
                 num_inference_steps=num_inference_steps,
                 guidance_scale=guidance_scale,
                 generator=generator,
             ).images[0]
+        print("Image generation completed successfully")
+        
+        filepath = save_generated_image(image, translated_prompt)
+        print(f"Image saved to: {filepath}")
         
-        filepath = save_generated_image(image, prompt)
         return image, seed
         
     except Exception as e:
+        print(f"Error in generate_image: {str(e)}")
+        print(f"Full error details: {repr(e)}")
         raise gr.Error(f"Image generation failed: {str(e)}")
     finally:
         clear_memory()