Update app.py

app.py

@@ -126,6 +126,7 @@ def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
             print(error_message)  # For logging
             raise gr.Error(error_message)
 
+        yield debug_resize, None, None, None
         print("Size after cropping", cropped_image.size)
         
         try:
@@ -134,7 +135,7 @@ def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
             error_message = f"Error during duplication: {str(e)}"
             print(error_message)  # For logging
             raise gr.Error(error_message)
-        
+        yield debug_resize, debug_duplicate, None, None
         print("just before getting into pipe")
         # Generate output
         out = pipe(
@@ -150,11 +151,11 @@ def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
         ).images[0]
         
         # First yield for progress
-        yield None, out
+        yield debug_resize, debug_duplicate, out, None
         
         # Process and return final output
         image_2 = safe_crop_output(out)
-        yield image_2, out
+        yield debug_resize, debug_duplicate, out, image_2
         
     except Exception as e:
         error_message = f"Error during generation: {str(e)}"
@@ -189,6 +190,8 @@ with gr.Blocks() as demo:
                 label="Side by side",
                 type="pil"
             )
+            debug_resize = gr.Image()
+            debug_duplicate = gr.Image()
             
     with gr.Row():
         gr.Markdown("""
@@ -204,7 +207,7 @@ with gr.Blocks() as demo:
     generate_btn.click(
         fn=generate,
         inputs=[input_image, prompt_input],
-        outputs=[output_image, output_side],
+        outputs=[debug_resize, debug_duplicate, output_side, output_image],
         api_name="generate"
     )