Update app.py

app.py

@@ -140,37 +140,47 @@ 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):
+    # Debug: Starting the function
+    logging.debug("Starting run_and_postprocess function.")
+    
     # Run the original image generation
     original_image = run_lora(prompt, selected_state)
     
-    # Post-process the image if needed
+    # Debug: Confirming that the original image was generated
+    logging.debug("Original image generated.")
+    
+    # Post-process the image based on user input
+    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
+    )
+    
+    # Debug: Confirming that post-processing was applied
     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
+        logging.debug("Post-processing applied.")
     else:
-        return original_image
+        logging.debug("Post-processing not applied.")
+    
+    return processed_image if enabled else original_image
         
 with gr.Blocks() as app:
     title = gr.Markdown("# artificialguybr LoRA portfolio")