Fix placeholder text and enhance image validation

.chainlit/config.toml

@@ -120,3 +120,4 @@ font_family = "Inter"
 [UI.chat]
 background_image = "assets/PP_backgound.jpg"
 background_image_opacity = 0.3
+chat_input_placeholder = "Upload an image to analyze your shelf layout..."

app.py

@@ -90,13 +90,27 @@ def compress_image(image, max_size=(400, 400)):
 
 def validate_image(image, name="image"):
     """Validate that an image is properly loaded and has valid dimensions."""
-    if image is None:
-        raise ValueError(f"Failed to load {name}")
-    if image.size == 0:
-        raise ValueError(f"{name} is empty")
-    if len(image.shape) != 3:
-        raise ValueError(f"{name} must be a color image (3 channels)")
-    return True
+    try:
+        if image is None:
+            raise ValueError(f"Failed to load {name}")
+        
+        if isinstance(image, np.ndarray):
+            if image.size == 0:
+                raise ValueError(f"{name} is empty")
+            if len(image.shape) != 3:
+                raise ValueError(f"{name} must be a color image (3 channels)")
+            logger.info(f"{name} shape: {image.shape}")
+        elif isinstance(image, Image.Image):
+            if image.size[0] == 0 or image.size[1] == 0:
+                raise ValueError(f"{name} has invalid dimensions")
+            logger.info(f"{name} size: {image.size}")
+        else:
+            raise ValueError(f"{name} must be a numpy array or PIL Image")
+        
+        return True
+    except Exception as e:
+        logger.error(f"Error validating {name}: {str(e)}", exc_info=True)
+        raise
 
 # Load planogram metadata
 try:
@@ -278,7 +292,14 @@ async def main(message: cl.Message):
                 logger.error(f"Failed to read image from path: {img_path}")
                 await processing_msg.update(content="❌ Error: Could not read the uploaded image. Please try again with a different image.")
                 return
-            logger.info(f"Successfully read image with shape: {uploaded_img.shape}")
+            
+            # Validate uploaded image
+            try:
+                validate_image(uploaded_img, "uploaded image")
+            except ValueError as e:
+                logger.error(f"Invalid uploaded image: {str(e)}")
+                await processing_msg.update(content=f"❌ Error: {str(e)}")
+                return
 
             # Load reference image
             logger.info("Loading reference image")
@@ -287,7 +308,14 @@ async def main(message: cl.Message):
                 logger.error("Failed to load reference image")
                 await processing_msg.update(content="❌ Error: Reference planogram image not found.")
                 return
-            logger.info("Successfully loaded reference image")
+            
+            # Validate reference image
+            try:
+                validate_image(ref_img, "reference image")
+            except ValueError as e:
+                logger.error(f"Invalid reference image: {str(e)}")
+                await processing_msg.update(content=f"❌ Error: {str(e)}")
+                return
 
             # Compare images using CLIP
             await processing_msg.update(content="🔄 Comparing with reference planogram...")