Update app.py

app.py

@@ -191,14 +191,89 @@ DEFAULT_VALUES = {
     "color_fix_method": "adain"
 }
 
+# Define example data
+examples = [
+    [
+        "examples/1.png",  # Input image path
+        "clean, texture, high-resolution, 8k",  # Prompt
+        "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",  # Negative prompt
+        3.0,  # Guidance scale
+        1.0,  # Conditioning scale
+        6,    # Num steps
+        42,   # Seed
+        4,    # Upscale factor
+        "wavelet"  # Color fix method
+    ],
+    [
+        "examples/22.png",
+        "clean, texture, high-resolution, 8k",
+        "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
+        3.0,
+        1.0,
+        6,
+        123,
+        4,
+        "wavelet"
+    ],
+    [
+        "examples/4.png",
+        "clean, texture, high-resolution, 8k",
+        "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
+        3.0,
+        1.0,
+        6,
+        123,
+        4,
+        "wavelet"
+    ],
+    [
+        "examples/9D03D7F206775949.png",
+        "clean, texture, high-resolution, 8k",
+        "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
+        3.0,
+        1.0,
+        6,
+        123,
+        4,
+        "wavelet"
+    ],
+    [
+        "examples/3.jpeg",
+        "clean, texture, high-resolution, 8k",
+        "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
+        2.5,
+        1.0,
+        6,
+        456,
+        4,
+        "wavelet"
+    ]
+]
+
 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
+    # First update the input fields and preview
+    yield [
+        img_path,  # input_image
+        prompt_text,
+        neg_prompt,
+        guid_scale,
+        cond_scale,
+        steps,
+        seed_val,
+        up_factor,
+        color_method,
+        img_path,  # preview_input
+        gr.update(value=None, label="Processing...")  # output_image
+    ]
+    
+    # Process the image
     output = process_image(
         img_path, prompt_text, neg_prompt, guid_scale,
         cond_scale, steps, seed_val, up_factor, color_method
     )
     
-    return [
+    # Return final result
+    yield [
         img_path,  # input_image
         prompt_text,
         neg_prompt,
@@ -209,9 +284,10 @@ def process_example(img_path, prompt_text, neg_prompt, guid_scale, cond_scale, s
         up_factor,
         color_method,
         img_path,  # preview_input
-        output     # output_image
+        gr.update(value=output, label="Generated Image")  # output_image
     ]
 
+# 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.")
@@ -242,85 +318,40 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
             preview_input = gr.Image(label="Preview Input")
             output_image = gr.Image(label="Generated Image")
 
-    # Define example data
-    examples = [
-        [
-            "examples/1.png",  # Input image path
-            "clean, texture, high-resolution, 8k",  # Prompt
-            "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",  # Negative prompt
-            3.0,  # Guidance scale
-            1.0,  # Conditioning scale
-            6,    # Num steps
-            42,   # Seed
-            4,    # Upscale factor
-            "wavelet"  # Color fix method
-        ],
-        [
-            "examples/22.png",
-            "clean, texture, high-resolution, 8k",
-            "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
-            3.0,
-            1.0,
-            6,
-            123,
-            4,
-            "wavelet"
-        ],
-        [
-            "examples/4.png",
-            "clean, texture, high-resolution, 8k",
-            "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
-            3.0,
-            1.0,
-            6,
-            123,
-            4,
-            "wavelet"
-        ],
-        [
-            "examples/9D03D7F206775949.png",
-            "clean, texture, high-resolution, 8k",
-            "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
-            3.0,
-            1.0,
-            6,
-            123,
-            4,
-            "wavelet"
-        ],
-        [
-            "examples/3.jpeg",
-            "clean, texture, high-resolution, 8k",
-            "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
-            2.5,
-            1.0,
-            6,
-            456,
-            4,
-            "wavelet"
-        ]
-    ]
-
     # Add gallery of examples
-    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,
-    )
+    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=[
+                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 with progress
+    def process_with_progress(img, prompt_text, neg_prompt, guid_scale, cond_scale, steps, seed_val, up_factor, color_method):
+        # Update UI to show processing state
+        yield gr.update(value=None, label="Processing...")
+        
+        # Process image
+        result = process_image(
+            img, prompt_text, neg_prompt, guid_scale,
+            cond_scale, steps, seed_val, up_factor, color_method
+        )
+        
+        # Return final result
+        yield gr.update(value=result, label="Generated Image")
 
-    # Define submit action
     submit_btn.click(
-        fn=process_image,
+        fn=process_with_progress,
         inputs=[
             input_image, prompt, negative_prompt, guidance_scale,
             conditioning_scale, num_steps, seed, upscale_factor,
@@ -331,7 +362,7 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
 
     # Add event handler for input image change
     def update_preview(img):
-        return img
+        return img if img is not None else None
 
     input_image.change(
         fn=update_preview,
@@ -367,4 +398,3 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
 
 if __name__ == "__main__":
     demo.launch()
-