Update app.py

app.py

@@ -6,6 +6,7 @@ import threading
 import time
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub.utils import RepositoryNotFoundError
 
 app = Flask(__name__)
 DATA_FILE = 'products.json'
@@ -20,19 +21,26 @@ logging.basicConfig(level=logging.DEBUG)
 
 # --- Функции для работы с JSON ---
 def load_data():
-    """Загружает данные из JSON-файла."""
-    if os.path.exists(DATA_FILE):
-        try:
-            with open(DATA_FILE, 'r', encoding='utf-8') as file:
-                return json.load(file)
-        except json.JSONDecodeError:
-            logging.error("Ошибка: Невозможно декодировать JSON файл.")
-            return []
-    else:
-        with open(DATA_FILE, 'w', encoding='utf-8') as file:
-            json.dump([], file, ensure_ascii=False, indent=4)
+    """Загружает данные из JSON-файла или скачивает их из репозитория Hugging Face."""
+    try:
+        # Пытаемся скачать базу данных из Hugging Face
+        download_db_from_hf()
+        with open(DATA_FILE, 'r', encoding='utf-8') as file:
+            return json.load(file)
+    except FileNotFoundError:
+        logging.warning("Локальный файл базы данных не найден после скачивания.")
+        return []
+    except json.JSONDecodeError:
+        logging.error("Ошибка: Невозможно декодировать JSON файл.")
+        return []
+    except RepositoryNotFoundError:
+        logging.error("Репозиторий не найден. Создание локальной базы данных.")
+        return [] # Возвращаем пустой список, чтобы начать с пустой базы
+    except Exception as e:
+        logging.error(f"Произошла ошибка при загрузке данных: {e}")
         return []
 
+
 def save_data(data):
     """Сохраняет данные в JSON-файл."""
     try:
@@ -71,8 +79,12 @@ def download_db_from_hf():
             local_dir_use_symlinks=False
         )
         logging.info("JSON база успешно скачана из Hugging Face.")
+    except RepositoryNotFoundError as e:
+        logging.error(f"Репозиторий не найден: {e}")
+        raise # Прокидываем исключение выше, чтобы load_data обработал его
     except Exception as e:
         logging.error(f"Ошибка при скачивании JSON базы: {e}")
+        raise # Прокидываем исключение выше, чтобы load_data обработал его
 
 def periodic_backup():
     """Периодически вызывает функцию upload_db_to_hf() каждые 100 секунд."""
@@ -258,4 +270,12 @@ if __name__ == '__main__':
     backup_thread = threading.Thread(target=periodic_backup, daemon=True)
     backup_thread.start()
 
+    # Попытка загрузить базу данных из репозитория перед запуском приложения
+    try:
+        load_data()
+    except Exception as e:
+        logging.error(f"Не удалось загрузить базу данных при запуске: {e}")
+        # Здесь можно добавить логику для создания пустой базы данных, если это необходимо
+        # Например: save_data([])
+
     app.run(debug=True, host='0.0.0.0', port=7860)