перевірка наявності сигнатури при розгортанні на HF

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from sdc_classifier import SDCClassifier
 from dotenv import load_dotenv
 import json
+import os
 
 # Load environment variables
 load_dotenv()
@@ -10,12 +11,40 @@ def main():
     # Ініціалізуємо класифікатор
     classifier = SDCClassifier()
     
-    # Спроба завантажити початкові класи та signatures
+    # Константи файлів
     DEFAULT_CLASSES_FILE = "classes_short.json"
     DEFAULT_SIGNATURES_FILE = "signatures.npz"
+    CACHE_FILE = "embeddings_cache.db"
     
     print("Завантаження початкових класів...")
-    classifier.load_initial_state(DEFAULT_CLASSES_FILE, DEFAULT_SIGNATURES_FILE)
+    
+    # Перевірка наявності кешу
+    if not os.path.exists(CACHE_FILE):
+        print("Кеш ембедінгів не знайдено. Створюємо новий...")
+        
+        # Перевіряємо наявність classes.json
+        if not os.path.exists(DEFAULT_CLASSES_FILE):
+            print(f"ПОМИЛКА: Файл {DEFAULT_CLASSES_FILE} не знайдено!")
+            return
+        
+        # Завантажуємо класи і створюємо нові signatures
+        try:
+            classifier.load_initial_state(DEFAULT_CLASSES_FILE, DEFAULT_SIGNATURES_FILE)
+            result = classifier.initialize_signatures(
+                force_rebuild=True,  # Примусово будуємо нові signatures
+                signatures_file=DEFAULT_SIGNATURES_FILE
+            )
+            print(f"Результат ініціалізації: {result}")
+        except Exception as e:
+            print(f"ПОМИЛКА при ініціалізації: {str(e)}")
+            return
+    else:
+        # Якщо кеш існує, просто завантажуємо початковий стан
+        try:
+            classifier.load_initial_state(DEFAULT_CLASSES_FILE, DEFAULT_SIGNATURES_FILE)
+        except Exception as e:
+            print(f"ПОМИЛКА при завантаженні початкового стану: {str(e)}")
+            return
 
     with gr.Blocks() as demo:
         gr.Markdown("# SDC Classifier з Gradio")

sdc_classifier.py

@@ -24,6 +24,11 @@ class SDCClassifier:
         self.embeddings_mean = None
         self.embeddings_std = None
         
+        # Створення директорії для кешу, якщо потрібно
+        cache_dir = os.path.dirname(cache_path)
+        if cache_dir and not os.path.exists(cache_dir):
+            os.makedirs(cache_dir)
+            
         # Ініціалізація кешу
         from embedding_cache import EmbeddingCache
         self.cache = EmbeddingCache(cache_path)
@@ -61,6 +66,35 @@ class SDCClassifier:
         self.classes_json = self.base_classes_json.copy()
         self.class_signatures = self.base_signatures.copy() if self.base_signatures else None
 
+    def load_initial_state(self, classes_file: str, signatures_file: str) -> str:
+        """
+        Завантаження початкового стану при старті застосунку
+        
+        Args:
+            classes_file: шлях до файлу з класами
+            signatures_file: шлях до файлу з signatures
+            
+        Returns:
+            str: повідомлення про результат завантаження
+        """
+        try:
+            self.base_classes_json = self.load_classes(classes_file)
+            if os.path.exists(signatures_file):
+                self.base_signatures = self.load_signatures(signatures_file)
+            
+            # Встановлюємо поточний стан як базовий
+            self.classes_json = self.base_classes_json.copy()
+            self.class_signatures = self.base_signatures.copy() if self.base_signatures else None
+            
+            return f"Завантажено {len(self.base_classes_json)} базових класів"
+        except Exception as e:
+            return f"Помилка при завантаженні базового стану: {str(e)}"
+            
+    def restore_base_state(self) -> None:
+        """Відновлення базового стану"""
+        self.classes_json = self.base_classes_json.copy()
+        self.class_signatures = self.base_signatures.copy() if self.base_signatures else None
+
     def load_initial_state(self, classes_file: str, signatures_file: str):
         """Завантаження початкового стану при старті застосунку"""
         self.base_classes_json = self.load_classes(classes_file)