Update app.py

app.py

@@ -214,7 +214,7 @@ def generate_txt2img(prompt, negative_prompt, num_steps, guidance_scale, width,
             negative_prompt or "", seed, num_steps, guidance_scale
         )
         
-        return png_path, status
+        return result.images[0], png_path, status
         
     except Exception as e:
         return None, f"Generation failed: {str(e)}"
@@ -291,7 +291,7 @@ def generate_img2img(input_image, prompt, negative_prompt, num_steps, guidance_s
             negative_prompt or "", seed, num_steps, guidance_scale, strength
         )
         
-        return png_path, status
+        return result.images[0], png_path, status
         
     except Exception as e:
         return None, f"Transformation failed: {str(e)}"
@@ -315,7 +315,8 @@ def set_example_prompt():
     return random.choice(EXAMPLE_PROMPTS)
 
 # Simplified negative prompt for better performance
-DEFAULT_NEGATIVE = """(low quality:1.3), (worst quality:1.3), (bad quality:1.2), blurry, noisy, ugly, deformed, 
+DEFAULT_NEGATIVE = """
+(low quality:1.3), (worst quality:1.3), (bad quality:1.2), blurry, noisy, ugly, deformed, 
 (text, watermark:1.4), (extra limbs:1.3), (bad hands:1.3), (bad anatomy:1.2)
 """
 
@@ -374,7 +375,8 @@ with gr.Blocks(
                     txt2img_btn = gr.Button("🎨 Generate", variant="primary", size="lg")
                 
                 with gr.Column():
-                    txt2img_output = gr.File(label="Generated PNG with Metadata", file_types=[".png"])
+                    txt2img_preview = gr.Image(label="Preview", height=400)
+                    txt2img_output = gr.File(label="📥 Download PNG with Metadata", file_types=[".png"])
                     txt2img_status = gr.Textbox(label="Generation Info", interactive=False, lines=6)
         
         with gr.TabItem("🖼️ Image to Image"):
@@ -420,7 +422,8 @@ with gr.Blocks(
                     img2img_btn = gr.Button("🖼️ Transform", variant="primary", size="lg")
                 
                 with gr.Column():
-                    img2img_output = gr.File(label="Generated PNG with Metadata", file_types=[".png"])
+                    img2img_preview = gr.Image(label="Preview", height=400)
+                    img2img_output = gr.File(label="📥 Download PNG with Metadata", file_types=[".png"])
                     img2img_status = gr.Textbox(label="Generation Info", interactive=False, lines=6)
     
     # Event handlers
@@ -428,14 +431,14 @@ with gr.Blocks(
         fn=generate_txt2img,
         inputs=[txt2img_prompt, txt2img_negative, txt2img_steps, txt2img_guidance, 
                 txt2img_width, txt2img_height, txt2img_seed, txt2img_quality_tags],
-        outputs=[txt2img_output, txt2img_status]
+        outputs=[txt2img_preview, txt2img_output, txt2img_status]
     )
     
     img2img_btn.click(
         fn=generate_img2img,
         inputs=[img2img_input, img2img_prompt, img2img_negative, img2img_steps, img2img_guidance, 
                 img2img_strength, img2img_seed, img2img_quality_tags],
-        outputs=[img2img_output, img2img_status]
+        outputs=[img2img_preview, img2img_output, img2img_status]
     )
     
     # Example prompt buttons