Add progress bar and fix postprocess parameters

app.py

@@ -35,24 +35,26 @@ def enhance_image_simple(
     strength: float,
     guidance_scale: float,
     use_upscaler: bool,
-    use_postprocess: bool
+    use_postprocess: bool,
+    progress=gr.Progress()
 ) -> Image.Image:
     """Enhance architectural image"""
     try:
         if input_image is None:
             raise ValueError("Please upload an image first")
             
-        logger.info("Starting enhancement process...")
+        progress(0, desc="Starting enhancement...")
         
         # Initialize models
         try:
+            progress(0.1, desc="Initializing AI models...")
             initialize_models()
         except Exception as e:
             logger.error(f"Model initialization failed: {e}", exc_info=True)
             raise ValueError(f"Failed to load AI models: {str(e)}")
         
         # Step 1: AI Enhancement
-        logger.info("Applying AI enhancement...")
+        progress(0.3, desc="Applying AI enhancement (this may take several minutes on CPU)...")
         try:
             enhanced = pipeline_manager.enhance(
                 image=input_image,
@@ -60,7 +62,7 @@ def enhance_image_simple(
                 negative_prompt="blurry, low quality, distorted, ugly, bad architecture",
                 strength=strength,
                 guidance_scale=guidance_scale,
-                num_inference_steps=30
+                num_inference_steps=20  # Reduced for faster processing
             )
         except Exception as e:
             logger.error(f"AI enhancement failed: {e}", exc_info=True)
@@ -68,7 +70,7 @@ def enhance_image_simple(
         
         # Step 2: Upscaling
         if use_upscaler:
-            logger.info("Upscaling image...")
+            progress(0.7, desc="Upscaling image...")
             try:
                 enhanced = upscale_image(enhanced, scale=2)
             except Exception as e:
@@ -76,18 +78,20 @@ def enhance_image_simple(
         
         # Step 3: Post-processing
         if use_postprocess:
-            logger.info("Post-processing...")
+            progress(0.9, desc="Applying final touches...")
             try:
                 enhanced = postprocess_image(
                     enhanced,
-                    clahe=True,
-                    sharpen=1.0,
-                    color_enhance=1.1
+                    apply_contrast=True,
+                    apply_sharpening=True,
+                    apply_color_grading=True,
+                    apply_grain=False,
+                    apply_vignette_effect=False
                 )
             except Exception as e:
                 logger.warning(f"Post-processing failed, skipping: {e}")
         
-        logger.info("Enhancement complete!")
+        progress(1.0, desc="Complete!")
         return enhanced
         
     except ValueError as e: