compare to chicory-ui

ui.py

@@ -2,6 +2,7 @@ import gradio as gr
 from utils import SafeProgress, format_categories_html
 from embeddings import create_product_embeddings
 from similarity import compute_similarities
+from chicory_api import call_chicory_parser  # Import the Chicory API function
 
 # Global variable for embeddings
 embeddings = {}
@@ -22,6 +23,10 @@ def categorize_products_from_text(product_text, top_n=5, confidence_threshold=0.
     progress_tracker(0.1, desc="Generating product embeddings...")
     products_embeddings = create_product_embeddings(product_names)
     
+    # Call Chicory Parser API
+    progress_tracker(0.5, desc="Calling Chicory Parser API...")
+    chicory_results = call_chicory_parser(product_names)
+    
     # Compute similarities
     progress_tracker(0.6, desc="Computing similarities...")
     all_similarities = compute_similarities(embeddings, products_embeddings)
@@ -36,7 +41,7 @@ def categorize_products_from_text(product_text, top_n=5, confidence_threshold=0.
                                 if score >= confidence_threshold]
         top_similarities = filtered_similarities[:top_n]
         
-        output_html += format_categories_html(product, top_similarities)
+        output_html += format_categories_html(product, top_similarities, chicory_result=chicory_results.get(product))
         output_html += "<hr style='margin: 15px 0; border: 0; border-top: 1px solid #eee;'>"
     
     output_html += "</div>"
@@ -67,6 +72,10 @@ def categorize_products_from_file(file, top_n=5, confidence_threshold=0.5, progr
     progress_tracker(0.2, desc="Generating product embeddings...")
     products_embeddings = create_product_embeddings(product_names)
     
+    # Call Chicory Parser API
+    progress_tracker(0.5, desc="Calling Chicory Parser API...")
+    chicory_results = call_chicory_parser(product_names)
+    
     # Compute similarities
     progress_tracker(0.7, desc="Computing similarities...")
     all_similarities = compute_similarities(embeddings, products_embeddings)
@@ -84,7 +93,7 @@ def categorize_products_from_file(file, top_n=5, confidence_threshold=0.5, progr
                                 if score >= confidence_threshold]
         top_similarities = filtered_similarities[:top_n]
         
-        output_html += format_categories_html(product, top_similarities)
+        output_html += format_categories_html(product, top_similarities, chicory_result=chicory_results.get(product))
         output_html += "<hr style='margin: 15px 0; border: 0; border-top: 1px solid #eee;'>"
     
     output_html += "</div>"