Update giga_App.py

giga_App.py

@@ -9,6 +9,9 @@ import time
 import platform
 import argparse
 
+# Global variable to control batch processing cancellation.
+stop_batch_flag = False
+
 def open_folder():
     open_folder_path = os.path.abspath("outputs")
     if platform.system() == "Windows":
@@ -82,6 +85,10 @@ def process_single_image(input_image_path, reduce_seams):
            f"Upscaling complete in {processing_time:.2f} seconds"]
 
 def process_batch(input_folder, output_folder=None, reduce_seams=False):
+    global stop_batch_flag
+    # Reset the stop flag for each new batch process.
+    stop_batch_flag = False
+
     if not input_folder:
         raise gr.Error("Please provide an input folder path.")
 
@@ -98,6 +105,11 @@ def process_batch(input_folder, output_folder=None, reduce_seams=False):
     yield [results, "Starting batch processing..."]
 
     for filename in input_files:
+        # Check if the stop flag has been set.
+        if stop_batch_flag:
+            yield [results, "Batch processing cancelled by user."]
+            return
+
         input_path = os.path.join(input_folder, filename)
         pil_image = Image.open(input_path)
 
@@ -123,7 +135,12 @@ def process_batch(input_folder, output_folder=None, reduce_seams=False):
 
     yield [results, f"Batch processing complete. {processed_files} images processed."]
 
-title = """<h1 align="center">AuraSR Giga Upscaler V3 by SECourses - Upscales to 4x</h1>
+def stop_batch_process():
+    global stop_batch_flag
+    stop_batch_flag = True
+    return "Stop button clicked. Cancelling batch processing..."
+
+title = """<h1 align="center">AuraSR Giga Upscaler V4 by SECourses - Upscales to 4x</h1>
 <p><center>AuraSR: new open source super-resolution upscaler based on GigaGAN. Works perfect on some images and fails on some images so give it a try</center></p>
 <p><center>Works very fast and very VRAM friendly</center></p>
 <h2 align="center">Latest version on : <a href="https://www.patreon.com/posts/121441873">https://www.patreon.com/posts/121441873</a></h2>
@@ -166,7 +183,9 @@ def create_demo():
                     value=True,
                     info="upscale_4x upscales the image in tiles that do not overlap. This can result in seams. Use upscale_4x_overlapped to reduce seams. This will double the time upscaling by taking an additional pass and averaging the results."
                 )
-            batch_process_btn = gr.Button(value="Process Batch", variant="primary")
+            with gr.Row():
+                batch_process_btn = gr.Button(value="Process Batch", variant="primary")
+                stop_batch_btn = gr.Button(value="Stop Batch Processing", variant="secondary")
             with gr.Column():
                 output_gallery_batch = gr.Gallery(label="Processed Images")
                 progress_text_batch = gr.Markdown("Progress messages will appear here.")
@@ -176,6 +195,11 @@ def create_demo():
                 inputs=[input_folder, output_folder, reduce_seams_batch],
                 outputs=[output_gallery_batch, progress_text_batch]
             )
+            stop_batch_btn.click(
+                fn=stop_batch_process,
+                inputs=[],
+                outputs=[progress_text_batch]
+            )
     
     return demo