Add detailed error handling and user-friendly error messages

app.py

@@ -39,43 +39,63 @@ def enhance_image_simple(
 ) -> Image.Image:
     """Enhance architectural image"""
     try:
+        if input_image is None:
+            raise ValueError("Please upload an image first")
+            
         logger.info("Starting enhancement process...")
         
         # Initialize models
-        initialize_models()
+        try:
+            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...")
-        enhanced = enhance_with_diffusion(
-            pipeline_manager,
-            input_image,
-            prompt="professional architectural photography, highly detailed, 8k, photorealistic",
-            strength=strength,
-            guidance_scale=guidance_scale,
-            num_inference_steps=30
-        )
+        try:
+            enhanced = enhance_with_diffusion(
+                pipeline_manager,
+                input_image,
+                prompt="professional architectural photography, highly detailed, 8k, photorealistic",
+                strength=strength,
+                guidance_scale=guidance_scale,
+                num_inference_steps=30
+            )
+        except Exception as e:
+            logger.error(f"AI enhancement failed: {e}", exc_info=True)
+            raise ValueError(f"AI enhancement failed: {str(e)}")
         
         # Step 2: Upscaling
         if use_upscaler:
             logger.info("Upscaling image...")
-            enhanced = upscale_image(enhanced, scale=2)
+            try:
+                enhanced = upscale_image(enhanced, scale=2)
+            except Exception as e:
+                logger.warning(f"Upscaling failed, skipping: {e}")
         
         # Step 3: Post-processing
         if use_postprocess:
             logger.info("Post-processing...")
-            enhanced = postprocess_image(
-                enhanced,
-                clahe=True,
-                sharpen=1.0,
-                color_enhance=1.1
-            )
+            try:
+                enhanced = postprocess_image(
+                    enhanced,
+                    clahe=True,
+                    sharpen=1.0,
+                    color_enhance=1.1
+                )
+            except Exception as e:
+                logger.warning(f"Post-processing failed, skipping: {e}")
         
         logger.info("Enhancement complete!")
         return enhanced
         
+    except ValueError as e:
+        # Re-raise ValueError with user-friendly message
+        raise gr.Error(str(e))
     except Exception as e:
         logger.error(f"Enhancement failed: {e}", exc_info=True)
-        raise
+        raise gr.Error(f"Enhancement failed: {str(e)}")
 
 # Create simple interface
 iface = gr.Interface(