generate_cache.py

import os
import sys
import logic

# Configurar rutas
base_dir = os.path.dirname(__file__)
csv_path = os.path.join(base_dir, "data", "furniture_data.csv")
model_path = os.path.join(base_dir, "models", "bert_style_encoder.pth")
cache_path = os.path.join(base_dir, "vectores_cache.pkl")

print("Iniciando generación de caché...")
print(f"CSV: {csv_path}")
print(f"Modelo: {model_path}")

# Inicializar DataManager
dm = logic.DataManager(csv_path, cache_path, model_path)

# Esto forzará la generación y guardado del pickle
df = dm.cargar_datos()

print(f"Caché generado exitosamente en: {cache_path}")
print(f"Total items: {len(df)}")