Update app.py

app.py

@@ -15,7 +15,7 @@ warnings.filterwarnings("ignore")
 from config import Config
 from segmentation import RoomSegmentation
 from style_extractor import StyleExtractor
-from utils import load_image_safe, save_image_safe, create_comparison_image, enhance_image_quality
+from utils import load_image_safe, save_image_safe, create_comparison_image, create_multi_comparison_image, enhance_image_quality
 
 class AdvancedHFSpacesPipeline:
     """Advanced pipeline for Hugging Face Spaces with full capabilities"""
@@ -503,18 +503,40 @@ class AdvancedHFSpacesPipeline:
             # Process images
             result, style_summary, full_style_info = self.advanced_style_transfer(user_room, inspiration_room)
             
-            # Create comparison image
-            comparison = create_comparison_image(
-                user_room, inspiration_room, result, 
-                titles=["Your Room", "Inspiration", "Advanced Result"]
+            # Create multi-comparison image (show all three images)
+            # Convert PIL images to numpy arrays for comparison
+            user_np = np.array(user_room)
+            inspiration_np = np.array(inspiration_room)
+            result_np = np.array(result)
+            
+            # Ensure all images are in RGB format
+            if len(user_np.shape) == 3 and user_np.shape[2] == 3:
+                user_np = cv2.cvtColor(user_np, cv2.COLOR_RGB2BGR)
+            if len(inspiration_np.shape) == 3 and inspiration_np.shape[2] == 3:
+                inspiration_np = cv2.cvtColor(inspiration_np, cv2.COLOR_RGB2BGR)
+            if len(result_np.shape) == 3 and result_np.shape[2] == 3:
+                result_np = cv2.cvtColor(result_np, cv2.COLOR_RGB2BGR)
+            
+            comparison = create_multi_comparison_image(
+                [user_np, inspiration_np, result_np],
+                titles=["Your Room", "Inspiration", "Advanced Result"],
+                title="Advanced Interior Style Transfer"
             )
             
+            # Convert back to PIL format for Gradio
+            comparison = Image.fromarray(cv2.cvtColor(comparison, cv2.COLOR_BGR2RGB))
+            
             return comparison, style_summary, full_style_info
             
         except Exception as e:
             error_msg = f"Advanced processing failed: {str(e)}"
             print(error_msg)
-            return None, error_msg, {}
+            
+            # Fallback: return just the result image
+            try:
+                return result, style_summary, full_style_info
+            except:
+                return None, error_msg, {}
 
 # Initialize advanced pipeline
 pipeline = AdvancedHFSpacesPipeline()
@@ -570,7 +592,7 @@ def create_advanced_interface():
         # Advanced style details
         with gr.Accordion("🔍 Detailed Style Analysis", open=False):
             style_details = gr.JSON(
-                label="Complete Style Information",
+                label="Complete Style Information"
             )
         
         # Processing function
@@ -608,3 +630,4 @@ if __name__ == "__main__":
         share=False,
         show_error=True
     )
+