Update app.py

app.py

@@ -192,7 +192,7 @@ DEFAULT_VALUES = {
 }
 
 # Define example data
-examples = [
+EXAMPLES = [
     [
         "examples/1.png",  # Input image path
         "clean, texture, high-resolution, 8k",  # Prompt
@@ -250,43 +250,6 @@ examples = [
     ]
 ]
 
-def process_example(img_path, prompt_text, neg_prompt, guid_scale, cond_scale, steps, seed_val, up_factor, color_method):
-    # 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 final result
-    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=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")
@@ -315,43 +278,24 @@ 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")
 
-    # 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=[
-                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")
+    # Add examples
+    gr.Examples(
+        examples=EXAMPLES,
+        inputs=[
+            input_image, prompt, negative_prompt, guidance_scale,
+            conditioning_scale, num_steps, seed, upscale_factor,
+            color_fix_method
+        ],
+        outputs=output_image,
+        fn=process_image,
+        cache_examples=True  # Cache the results for faster loading
+    )
 
+    # Define submit action
     submit_btn.click(
-        fn=process_with_progress,
+        fn=process_image,
         inputs=[
             input_image, prompt, negative_prompt, guidance_scale,
             conditioning_scale, num_steps, seed, upscale_factor,
@@ -360,16 +304,6 @@ with gr.Blocks(title="Controllable Conditional Super-Resolution") as demo:
         outputs=output_image
     )
 
-    # Add event handler for input image change
-    def update_preview(img):
-        return img if img is not None else None
-
-    input_image.change(
-        fn=update_preview,
-        inputs=input_image,
-        outputs=preview_input
-    )
-
     # Define clear action that resets to default values
     def reset_to_defaults():
         return [
@@ -381,9 +315,7 @@ 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"],
-            None,  # preview_input
-            None   # output_image
+            DEFAULT_VALUES["color_fix_method"]
         ]
 
     clear_btn.click(
@@ -392,9 +324,9 @@ 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, preview_input, output_image
+            color_fix_method
         ]
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()