Spaces:
Sleeping
Sleeping
| import pickle | |
| import json | |
| import os | |
| class SafeProgress: | |
| """Wrapper for progress tracking that handles None gracefully""" | |
| def __init__(self, progress_obj=None): | |
| self.progress = progress_obj | |
| def __call__(self, value, desc=""): | |
| if self.progress is not None: | |
| try: | |
| self.progress(value, desc=desc) | |
| except: | |
| print(f"Progress {value}: {desc}") | |
| else: | |
| print(f"Progress {value}: {desc}") | |
| def load_embeddings(embeddings_path): | |
| """Load ingredient embeddings from pickle file""" | |
| print(f"Loading ingredient embeddings from {embeddings_path}") | |
| with open(embeddings_path, "rb") as f: | |
| ingredients_embeddings = pickle.load(f) | |
| print(f"Loaded {len(ingredients_embeddings)} ingredient embeddings") | |
| return ingredients_embeddings | |
| def parse_product_file(file_path): | |
| """Parse a file containing product data and extract product names""" | |
| try: | |
| with open(file_path, 'r') as f: | |
| try: | |
| products_data = json.load(f) | |
| if isinstance(products_data, list): | |
| # Extract product names if it's a list of objects with 'name' field | |
| if all(isinstance(item, dict) for item in products_data): | |
| product_names = [item.get('name', '') for item in products_data if isinstance(item, dict)] | |
| else: | |
| # If it's just a list of strings | |
| product_names = [str(item) for item in products_data if item] | |
| else: | |
| # If it's just a list of product names | |
| product_names = [] | |
| except json.JSONDecodeError: | |
| # If not JSON, try reading as text file with one product per line | |
| f.seek(0) | |
| product_names = [line.strip() for line in f.readlines() if line.strip()] | |
| except Exception as e: | |
| raise Exception(f"Error reading file: {str(e)}") | |
| return product_names | |
| def format_categories_html(product, categories): | |
| """Format categories as HTML with color-coded confidence scores""" | |
| html = f"<div style='margin-bottom: 10px;'><b>{product}</b></div>" | |
| if not categories: | |
| html += "<div style='color: #666; font-style: italic;'>No matching categories found.</div>" | |
| return html | |
| html += "<div style='margin-left: 15px;'>" | |
| for i, (category, score) in enumerate(categories, 1): | |
| # Color code based on confidence | |
| if score >= 0.8: | |
| color = "#1a8a38" # Strong green | |
| elif score >= 0.65: | |
| color = "#4caf50" # Medium green | |
| elif score >= 0.5: | |
| color = "#8bc34a" # Light green | |
| else: | |
| color = "#9e9e9e" # Gray | |
| html += f"<div style='margin-bottom: 5px;'>{i}. <span style='font-weight: 500;'>{category}</span> <span style='color: {color}; font-weight: bold;'>({score:.3f})</span></div>" | |
| html += "</div>" | |
| return html | |