Update app.py

app.py

@@ -191,6 +191,33 @@ DEFAULT_VALUES = {
     "color_fix_method": "adain"
 }
 
+def process_example(example_data):
+    # Unpack example data
+    input_img, prompt_text, neg_prompt, guid_scale, cond_scale, steps, seed_val, up_factor, color_method = example_data
+    
+    # First update the input image and parameters
+    results = [
+        input_img,  # input_image
+        prompt_text,
+        neg_prompt,
+        guid_scale,
+        cond_scale,
+        steps,
+        seed_val,
+        up_factor,
+        color_method,
+        input_img,  # Show input image in preview
+        gr.update(value=None)  # Clear previous output while processing
+    ]
+    
+    # Then process the image
+    output = process_image(
+        input_img, prompt_text, neg_prompt, guid_scale,
+        cond_scale, steps, seed_val, up_factor, color_method
+    )
+    
+    return results[:-2] + [input_img, output]  # Return all updated values plus input/output images
+
 # Create interface components
 with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
     gr.Markdown("## Controllable Conditional Super-Resolution")
@@ -219,6 +246,7 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
                 submit_btn = gr.Button("Submit", variant="primary")
 
         with gr.Column():
+            preview_input = gr.Image(label="Preview Input")
             output_image = gr.Image(label="Generated Image")
 
     # Define example data
@@ -257,7 +285,7 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
             "wavelet"
         ],
         [
-            "examples/9D03D7F206775949.png.png",
+            "examples/9D03D7F206775949.png",
             "clean, texture, high-resolution, 8k",
             "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
             3.0,
@@ -268,7 +296,7 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
             "wavelet"
         ],
         [
-            "examples/3.png",
+            "examples/3.jpeg",
             "clean, texture, high-resolution, 8k",
             "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
             2.5,
@@ -289,9 +317,9 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
                 conditioning_scale, num_steps, seed, upscale_factor,
                 color_fix_method
             ],
-            outputs=output_image,
-            fn=process_image,
-            run_on_click=True  # This makes it run inference automatically when clicked
+            outputs=[preview_input, output_image],
+            fn=process_example,
+            cache_examples=True,
         )
 
     # Define submit action
@@ -306,14 +334,13 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
     )
 
     # Add event handler for input image change
+    def update_preview(img):
+        return img
+
     input_image.change(
-        fn=process_image,
-        inputs=[
-            input_image, prompt, negative_prompt, guidance_scale,
-            conditioning_scale, num_steps, seed, upscale_factor,
-            color_fix_method
-        ],
-        outputs=output_image
+        fn=update_preview,
+        inputs=input_image,
+        outputs=preview_input
     )
 
     # Define clear action that resets to default values
@@ -327,7 +354,9 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
             DEFAULT_VALUES["num_steps"],
             DEFAULT_VALUES["seed"],
             DEFAULT_VALUES["upscale_factor"],
-            DEFAULT_VALUES["color_fix_method"]
+            DEFAULT_VALUES["color_fix_method"],
+            None,  # preview_input
+            None   # output_image
         ]
 
     clear_btn.click(
@@ -336,7 +365,7 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
         outputs=[
             input_image, prompt, negative_prompt, guidance_scale,
             conditioning_scale, num_steps, seed, upscale_factor,
-            color_fix_method
+            color_fix_method, preview_input, output_image
         ]
     )