Update app.py

app.py

@@ -308,7 +308,7 @@ def run_process(
     # Build final instructions & call twice -> 2 final images
     instruction = build_multi_change_instruction(r1, final_text1, r2, final_text2, r3, final_text3)
     final_imgs = change_text_in_image_two_times(_random_image, instruction)
-    # Return only the 2 final images (don't show the random image)
+    # Return only the 2 final images
     return [final_imgs[0], final_imgs[1]]
 
 #######################################
@@ -359,7 +359,8 @@ with gr.Blocks(title="Eevery Text Imaginator: FLUX") as demo:
 
     with gr.Row():
         with gr.Column():
-            with gr.Box():
+            # Instead of gr.Box(), we can just group them or leave them as is
+            with gr.Group():  
                 prompt_input = gr.Textbox(
                     lines=3,
                     label="Prompt (Korean or English)",
@@ -397,7 +398,6 @@ with gr.Blocks(title="Eevery Text Imaginator: FLUX") as demo:
             final_image_output1 = gr.Image(label="Final Image #1", type="pil")
             final_image_output2 = gr.Image(label="Final Image #2", type="pil")
 
-    # We only display the 2 final images, not the initial random image
     run_btn.click(
         fn=run_process,
         inputs=[