Reset button

app.py

@@ -63,6 +63,39 @@ def update_seed(is_randomize_seed, seed):
         return random.randint(0, max_64_bit_int)
     return seed
 
+def reset():
+    return [
+        None,
+        0,
+        None,
+        None,
+        "Cinematic, High Contrast, highly detailed, taken using a Canon EOS R camera, hyper detailed photo - realistic maximum detail, 32k, Color Grading, ultra HD, extreme meticulous detailing, skin pore detailing, hyper sharpness, perfect without deformations.",
+        "painting, oil painting, illustration, drawing, art, sketch, anime, cartoon, CG Style, 3D render, unreal engine, blurring, aliasing, bokeh, ugly, dirty, messy, worst quality, low quality, frames, watermark, signature, jpeg artifacts, deformed, lowres, over-smooth",
+        1,
+        1024,
+        1,
+        2,
+        50,
+        -1.0,
+        1.,
+        default_setting.s_cfg_Quality if torch.cuda.device_count() > 0 else 1.0,
+        True,
+        random.randint(0, max_64_bit_int),
+        5,
+        1.003,
+        "Wavelet",
+        "fp32",
+        "fp32",
+        1.0,
+        True,
+        False,
+        default_setting.spt_linear_CFG_Quality if torch.cuda.device_count() > 0 else 1.0,
+        0.,
+        "v0-Q",
+        "input",
+        6
+    ]
+
 def check(input_image):
     if input_image is None:
         raise gr.Error("Please provide an image to restore.")
@@ -208,7 +241,7 @@ def restore_in_Xmin(
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
-        return [noisy_image, denoise_image], [denoise_image], None, None
+        return [noisy_image, denoise_image], [denoise_image], None, gr.update(visible=True)
 
     if model_select != model.current_model:
         print('load ' + model_select)
@@ -358,6 +391,7 @@ def restore(
     results = [x_samples[i] for i in range(num_samples)]
 
     # All the results have the same size
+    input_height, input_width, input_channel = np.array(input_image).shape
     result_height, result_width, result_channel = np.array(results[0]).shape
 
     print('<<== restore')
@@ -378,9 +412,20 @@ def restore(
     " pixels large and " + str(result_height) + \
     " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels."
     print(information)
+    try:
+        print("Estimated minutes: " + str(math.log(result_width * result_height * input_width * input_height * edm_steps * num_samples)))
+    except Exception as e:
+        print('Exception of Estimation')
+    try:
+        unique_name = str(uuid.uuid4()) + "." + output_format
+        image_copy = Image.fromarray(np.array(results[0]))
+        image_copy.save(unique_name)
+        print(unique_name)
+    except Exception as e:
+        print('Exception printing the path: ' + str(e))
 
     # Only one image can be shown in the slider
-    return [noisy_image] + [results[0]], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = results), gr.update(value = information, visible = True)
+    return [noisy_image] + [results[0]], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = results), gr.update(value = information, visible = True), gr.update(visible=True)
 
 def load_and_reset(param_setting):
     print('load_and_reset ==>>')
@@ -425,6 +470,9 @@ def log_information(result_gallery):
 
 def on_select_result(result_slider, result_gallery, evt: gr.SelectData):
     print('on_select_result')
+    if result_gallery is not None:
+        for i, result in enumerate(result_gallery):
+            print(result[0])
     return [result_slider[0], result_gallery[evt.index][0]]
 
 title_html = """
@@ -510,7 +558,7 @@ with gr.Blocks() as interface:
                 model_select = gr.Radio([["💃 Quality (v0-Q)", "v0-Q"], ["🎯 Fidelity (v0-F)", "v0-F"]], label="Model Selection", info="Pretrained model", value="v0-Q",
                                         interactive=True)
             with gr.Column():
-                color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", info="AdaIn=Improve following a style, Wavelet=For JPEG artifacts", value="Wavelet",
+                color_fix_type = gr.Radio([["None", "None"], ["AdaIn (improve as a photo)", "AdaIn"], ["Wavelet (for JPEG artifacts)", "Wavelet"]], label="Color-Fix Type", info="AdaIn=Improve following a style, Wavelet=For JPEG artifacts", value="Wavelet",
                                           interactive=True)
         s_cfg = gr.Slider(label="Text Guidance Scale", info="lower=follow the image, higher=follow the prompt", minimum=1.0, maximum=15.0,
                           value=default_setting.s_cfg_Quality if torch.cuda.device_count() > 0 else 1.0, step=0.1)
@@ -536,15 +584,16 @@ with gr.Blocks() as interface:
         randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different")
         seed = gr.Slider(label="Seed", minimum=0, maximum=max_64_bit_int, step=1, randomize=True)
         with gr.Group():
-            param_setting = gr.Radio(["Quality", "Fidelity"], interactive=True, label="Presetting", value="Quality")
+            param_setting = gr.Radio(["Quality", "Fidelity"], interactive=True, label="Presetting", value = "Quality")
             restart_button = gr.Button(value="Apply presetting")
 
-    with gr.Group():
-        diffusion_button = gr.Button(value="🚀 Upscale/Restore", variant = "primary", elem_id="process_button")
+    with gr.Column():
+        diffusion_button = gr.Button(value="🚀 Upscale/Restore", variant = "primary", elem_id = "process_button")
+        reset_btn = gr.Button(value="🧹 Reinit page", variant="stop", elem_id="reset_button", visible = False)
 
-        restore_information = gr.HTML(value="Restart the process to get another result.", visible = False)
-        result_slider = ImageSlider(label='Comparator', show_label=False, elem_id="slider1", show_download_button = False)
-        result_gallery = gr.Gallery(label='Downloadable results', show_label=True, elem_id="gallery1")
+        restore_information = gr.HTML(value = "Restart the process to get another result.", visible = False)
+        result_slider = ImageSlider(label = 'Comparator', show_label = False, elem_id = "slider1", show_download_button = False)
+        result_gallery = gr.Gallery(label = 'Downloadable results', show_label = True, elem_id = "gallery1")
 
     gr.Examples(
         run_on_click = True,
@@ -583,7 +632,8 @@ with gr.Blocks() as interface:
 	    outputs = [
             result_slider,
             result_gallery,
-            restore_information
+            restore_information,
+            reset_btn
         ],
         examples = [
                 [
@@ -775,11 +825,13 @@ with gr.Blocks() as interface:
     ], outputs = [
         result_slider,
         result_gallery,
-        restore_information
-    ]).then(fn = log_information, inputs = [
+        restore_information,
+        reset_btn
+    ]).success(fn = log_information, inputs = [
         result_gallery
     ], outputs = [], queue = False, show_progress = False)
 
+    result_gallery.change(on_select_result, [result_slider, result_gallery], result_slider)
     result_gallery.select(on_select_result, [result_slider, result_gallery], result_slider)
 
     restart_button.click(fn = load_and_reset, inputs = [
@@ -800,5 +852,37 @@ with gr.Blocks() as interface:
         spt_linear_s_stage2,
         model_select
     ])
+
+    reset_btn.click(fn = reset, inputs = [], outputs = [
+            input_image,
+            rotation,
+            denoise_image,
+            prompt,
+            a_prompt,
+            n_prompt,
+            num_samples,
+            min_size,
+            downscale,
+            upscale,
+            edm_steps,
+            s_stage1,
+            s_stage2,
+            s_cfg,
+            randomize_seed,
+            seed,
+            s_churn,
+            s_noise,
+            color_fix_type,
+            diff_dtype,
+            ae_dtype,
+            gamma_correction,
+            linear_CFG,
+            linear_s_stage2,
+            spt_linear_CFG,
+            spt_linear_s_stage2,
+            model_select,
+            output_format,
+            allocation
+        ], queue = False, show_progress = False)
         
     interface.queue(10).launch()