Update app.py

app.py

@@ -191,13 +191,15 @@ 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
+def process_example(img_path, prompt_text, neg_prompt, guid_scale, cond_scale, steps, seed_val, up_factor, color_method):
+    # Update UI components and process image
+    output = process_image(
+        img_path, prompt_text, neg_prompt, guid_scale,
+        cond_scale, steps, seed_val, up_factor, color_method
+    )
     
-    # First update the input image and parameters
-    results = [
-        input_img,  # input_image
+    return [
+        img_path,  # input_image
         prompt_text,
         neg_prompt,
         guid_scale,
@@ -206,19 +208,10 @@ def process_example(example_data):
         seed_val,
         up_factor,
         color_method,
-        input_img,  # Show input image in preview
-        gr.update(value=None)  # Clear previous output while processing
+        img_path,  # preview_input
+        output     # output_image
     ]
-    
-    # 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")
     gr.Markdown("Upload an image to enhance its resolution using CCSR.")
@@ -309,18 +302,21 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
     ]
 
     # Add gallery of examples
-    with gr.Row():
-        gr.Examples(
-            examples=examples,
-            inputs=[
-                input_image, prompt, negative_prompt, guidance_scale,
-                conditioning_scale, num_steps, seed, upscale_factor,
-                color_fix_method
-            ],
-            outputs=[preview_input, output_image],
-            fn=process_example,
-            cache_examples=True,
-        )
+    gr.Examples(
+        examples=examples,
+        inputs=[
+            input_image, prompt, negative_prompt, guidance_scale,
+            conditioning_scale, num_steps, seed, upscale_factor,
+            color_fix_method
+        ],
+        outputs=[
+            input_image, prompt, negative_prompt, guidance_scale,
+            conditioning_scale, num_steps, seed, upscale_factor,
+            color_fix_method, preview_input, output_image
+        ],
+        fn=process_example,
+        cache_examples=True,
+    )
 
     # Define submit action
     submit_btn.click(
@@ -370,4 +366,5 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+