ініціалізація сигнатури при першому запуску

app.py

@@ -7,26 +7,53 @@ import os
 # Load environment variables
 load_dotenv()
 
-def main():
-    # Ініціалізуємо класифікатор
-    classifier = SDCClassifier()
+def initialize_environment():
+    """Ініціалізація середовища при першому запуску"""
+    DEFAULT_CLASSES_FILE = "classes.json"
+    DEFAULT_SIGNATURES_FILE = "signatures.npz"
+    CACHE_FILE = "embeddings_cache.db"
     
+    # Перевіряємо наявність необхідних файлів
+    if not os.path.exists(DEFAULT_CLASSES_FILE):
+        print(f"ПОМИЛКА: Файл {DEFAULT_CLASSES_FILE} не знайдено!")
+        return False
+        
+    if not os.path.exists(DEFAULT_SIGNATURES_FILE):
+        print("Signatures не знайдено. Створюємо нові...")
+        try:
+            classifier = SDCClassifier()
+            classifier.load_classes(DEFAULT_CLASSES_FILE)
+            result = classifier.initialize_signatures(
+                force_rebuild=True,
+                signatures_file=DEFAULT_SIGNATURES_FILE
+            )
+            print(f"Результат ініціалізації: {result}")
+            return True
+        except Exception as e:
+            print(f"ПОМИЛКА при ініціалізації: {str(e)}")
+            return False
+            
+    return True
+
+def main():
     # Константи файлів
     DEFAULT_CLASSES_FILE = "classes_short.json"
     DEFAULT_SIGNATURES_FILE = "signatures.npz"
     CACHE_FILE = "embeddings_cache.db"
     
-    print("Завантаження початкових класів...")
+    # Перевіряємо та ініціалізуємо середовище
+    if not initialize_environment():
+        print("Не вдалося ініціалізувати середовище")
+        return
     
+    # Ініціалізуємо класифікатор
+    classifier = SDCClassifier()
+    
+    print("Завантаження початкових класів...")
     # Перевірка наявності кешу
     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)