fixed a bug

ui_expanded_matching.py

@@ -168,6 +168,9 @@ def categorize_products_with_openai_reranking(product_input, is_file=False, use_
                 return product, []
                 
             candidates = all_similarities[product][:embedding_top_n]
+
+            print(f"candidates: {candidates}")
+
             if not candidates:
                 return product, []
             
@@ -175,10 +178,16 @@ def categorize_products_with_openai_reranking(product_input, is_file=False, use_
             expanded_text = expanded_descriptions.get(product, product) if use_expansion else product
             
             try:
+                # FIXED: Filter categories to only include those in the current product's candidates
+                product_category_ids = [cat_id for cat_id, _ in candidates]
+                filtered_categories = {cat_id: category_descriptions[cat_id] 
+                                      for cat_id in product_category_ids 
+                                      if cat_id in category_descriptions}
+                
                 # Pass 0.0 as threshold to get all results - apply threshold at display time
                 category_matches = rank_categories_openai(
                     product=product,
-                    categories=category_descriptions,
+                    categories=filtered_categories,  # Pass only this product's relevant categories
                     expanded_description=expanded_text,
                     client=openai_client,
                     model="gpt-4o-mini",