Update backend/jewelry_recomm_service.py

backend/jewelry_recomm_service.py

@@ -31,34 +31,35 @@ class JewelryRecommenderService:
         # Initialize pipeline components
         self.image_processor = ImageProcessor(self.model)
         self.recommender = RecommenderEngine(self.index, self.metadata)
-    
-    def get_recommendations(self, image, num_recommendations=None,skip_exact_match=True):
+
+    def get_recommendations(self, image, num_recommendations=None, skip_exact_match=True):
         """Get recommendations for a query image.
         
         Args:
             image: Query image (various formats)
             num_recommendations (int, optional): Number of recommendations
+            skip_exact_match (bool): Whether to skip the first/exact match
             
         Returns:
             list: Recommendation results
         """
-            
         if not self.index or not self.metadata:
             return [{"error": "Index/metadata not loaded"}]
-        if embedding is None:
-            return [{"error": "Failed to process image"}]
-
-
+        
+        if image is None:
+            return [{"error": "Invalid image input"}]
+        
         num_recommendations = num_recommendations or Config.DEFAULT_NUM_RECOMMENDATIONS
         
         # Extract embedding from the image
         embedding = self.image_processor.extract_embedding(image)
         if embedding is None:
-            return[{"error":"Failed to process image"}]
+            return [{"error": "Failed to process image"}]
         
         # Get similar items based on the embedding
         recommendations = self.recommender.find_similar_items(
-            embedding, num_recommendations
+            embedding, num_recommendations, skip_exact_match
         )
         
-        return recommendations
+        return recommendations
+