feat: Update app.py to enhance image with new options

app.py

@@ -228,30 +228,58 @@ with gr.Blocks(css=css) as demo:
 
     submit_btn = gr.Button("Submit")
     cancel_btn = gr.Button("Cancel")
-    with gr.Row():
-        image_out = gr.Image(label="Image output")
-        image_upscaled = ImageSlider(label="Before / After", type="numpy", show_download_button=False)
     scale_btn = gr.Button("Upscale")
 
+
+    with gr.Row():
+        def clear_output(image_slider):
+            image_slider[0] = None
+            image_slider[1] = None
+            return image_slider
+
+        with gr.Column():
+            generated_image = gr.Image(label="Input Image", type="filepath")
+            enhace_button = gr.Button("Enhance Image")
+
+        with gr.Column():
+            output_slider = ImageSlider(label="Before / After", type="numpy", show_download_button=False)
+
+        with gr.Accordion("Advanced Options", open=False):
+            resolution = gr.Slider(minimum=128, maximum=2048, value=1024, step=128, label="Resolution")
+            num_inference_steps = gr.Slider(minimum=1, maximum=50, value=20, step=1, label="Number of Inference Steps")
+            strength = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.01, label="Strength")
+            hdr = gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="HDR Effect")
+            guidance_scale = gr.Slider(minimum=0, maximum=20, value=6, step=0.5, label="Guidance Scale")
+            controlnet_strength = gr.Slider(minimum=0.0, maximum=2.0, value=0.75, step=0.05, label="ControlNet Strength")
+            scheduler_name = gr.Dropdown(
+                choices=["DDIM", "DPM++ 3M SDE Karras", "DPM++ 3M Karras"],
+                value="DDIM",
+                label="Scheduler"
+            )
+
+
+
+
     selected_index = gr.State(None)
 
     submit_btn.click(
         fn=infer,
         inputs=[selected_index, prompt_in, style_prompt_in, inf_steps, guidance_scale, width, height, seed, lora_weight],
-        outputs=[image_out, last_used_seed, used_prompt]
+        outputs=[generated_image, last_used_seed, used_prompt]
     )
     cancel_btn.click(
         fn=cancel_infer,
         outputs=[]
     )
 
-    def upscale_with_fixed_values(image):
-        return upscale_image(image, 768, 25, 0.4, 0.3, 7.5)
-
-    scale_btn.click(
-        fn=upscale_with_fixed_values,
-        inputs=[image_out],
-        outputs=[image_upscaled]
+    enhace_button.click(
+        fn=clear_output,
+        inputs=[output_slider],
+        outputs=[output_slider]
+    ).then(
+        upscale_image, 
+        [generated_image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name],
+        output_slider
     )