Update app.py

app.py

@@ -8,16 +8,16 @@ import logging
 import time
 from tqdm import tqdm
 
-# Import additional modules
-from pixelart import Script as PixelArtScript
-from postprocessing_pixelart import ScriptPostprocessingUpscale
-from utils import downscale_image, limit_colors, resize_image, convert_to_grayscale, convert_to_black_and_white
-
 # Placeholder class for processed images
 class SomeClass:
     def __init__(self):
         self.images = []
 
+# Import your existing modules
+from pixelart import Script as PixelArtScript
+from postprocessing_pixelart import ScriptPostprocessingUpscale
+from utils import downscale_image, limit_colors, resize_image, convert_to_grayscale, convert_to_black_and_white
+
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
@@ -65,6 +65,23 @@ def run_lora(prompt, selected_state, pixel_art_options, postprocess_options, pro
         
         return original_image, refined_image
 
+def apply_post_processing(image, image_processing_options):
+    processed_image = image.copy()
+    
+    if image_processing_options['downscale'] > 1:
+        processed_image = downscale_image(processed_image, image_processing_options['downscale'])
+        
+    if image_processing_options['limit_colors']:
+        processed_image = limit_colors(processed_image)
+        
+    if image_processing_options['grayscale']:
+        processed_image = convert_to_grayscale(processed_image)
+        
+    if image_processing_options['black_and_white']:
+        processed_image = convert_to_black_and_white(processed_image)
+        
+    return processed_image
+
 with gr.Blocks() as app:
     title = gr.Markdown("# artificialguybr LoRA portfolio")
     description = gr.Markdown("### This is a Pixel Art Generator using SD Loras.")
@@ -87,12 +104,31 @@ with gr.Blocks() as app:
                 button = gr.Button("Run")
             
             result = gr.Image(interactive=False, label="Generated Image")
-            refined_result = gr.Image(interactive=False, label="Refined Generated Image")  # New Output
+            refined_result = gr.Image(interactive=False, label="Refined Generated Image")
+            
+            # New Output for Post-Processed Image
+            post_processed_result = gr.Image(interactive=False, label="Post-Processed Image")
             
             # New UI elements for pixel art options
             with gr.Row():
                 pixel_art_options = PixelArtScript().ui(True)
                 postprocess_options = ScriptPostprocessingUpscale().ui()
+                
+            # New UI elements for image processing options
+            with gr.Row():
+                downscale = gr.Slider(minimum=1, maximum=10, step=1, label="Downscale")
+                limit_colors = gr.Checkbox(label="Limit Colors")
+                grayscale = gr.Checkbox(label="Grayscale")
+                black_and_white = gr.Checkbox(label="Black and White")
+                
+            image_processing_options = {
+                'downscale': downscale,
+                'limit_colors': limit_colors,
+                'grayscale': grayscale,
+                'black_and_white': black_and_white
+            }
+            
+            post_process_button = gr.Button("Apply Post-Processing")
     
     gallery.select(
         update_selection,
@@ -105,10 +141,10 @@ with gr.Blocks() as app:
         outputs=[result, refined_result]
     )
     
-    button.click(
-        fn=run_lora,
-        inputs=[prompt, selected_state, pixel_art_options, postprocess_options],
-        outputs=[result, refined_result]
+    post_process_button.click(
+        fn=apply_post_processing,
+        inputs=[refined_result, image_processing_options],
+        outputs=[post_processed_result]
     )
 
 app.queue(max_size=20, concurrency_count=5)