Update app.py

app.py

@@ -139,6 +139,39 @@ def postprocess(
 
     return processed_image
 
+def run_and_postprocess(prompt, selected_state, enabled, downscale, need_rescale, enable_color_limit, palette_size_color, quantization_methods_color, dither_methods_color, k_means_color, enable_grayscale, palette_size_gray, quantization_methods_gray, dither_methods_gray, k_means_gray, enable_black_and_white, inverse_black_and_white, threshold_black_and_white, enable_custom_palette, palette_image, palette_size_custom, dither_methods_custom):
+    # Run the original image generation
+    original_image = run_lora(prompt, selected_state)
+    
+    # Post-process the image if needed
+    if enabled:
+        processed_image = postprocess(
+            original_image, 
+            enabled, 
+            downscale, 
+            need_rescale, 
+            enable_color_limit, 
+            palette_size_color, 
+            quantization_methods_color, 
+            dither_methods_color, 
+            k_means_color, 
+            enable_grayscale, 
+            palette_size_gray, 
+            quantization_methods_gray, 
+            dither_methods_gray, 
+            k_means_gray, 
+            enable_black_and_white, 
+            inverse_black_and_white, 
+            threshold_black_and_white, 
+            enable_custom_palette, 
+            palette_image, 
+            palette_size_custom, 
+            dither_methods_custom
+        )
+        return processed_image
+    else:
+        return original_image
+        
 with gr.Blocks() as app:
     title = gr.Markdown("# artificialguybr LoRA portfolio")
     description = gr.Markdown("### This is a Pixel Art Generator using SD Loras.")