Update app.py

app.py

@@ -14,7 +14,7 @@ import traceback
 
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey")
-DATA_FILE = 'cloud_data.json'
+DATA_FILE = 'cloud_data.json'  # Can be changed to an absolute path if needed, e.g., os.path.join(os.getcwd(), 'cloud_data.json')
 REPO_ID = "Eluza133/Z1e1u"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
@@ -30,6 +30,13 @@ if not HF_TOKEN_WRITE:
 else:
     logging.info("HF_TOKEN_WRITE успешно установлен (первые 5 символов: {0}...)".format(HF_TOKEN_WRITE[:5]))
 
+# Инициализация файла базы данных при старте, если он не существует
+def initialize_data_file():
+    if not os.path.exists(DATA_FILE):
+        logging.info(f"Файл {DATA_FILE} не найден, создается новый.")
+        with open(DATA_FILE, 'w', encoding='utf-8') as f:
+            json.dump({'users': {}, 'files': {}}, f)
+
 # Функции для работы с базой данных и Hugging Face
 @cache.memoize(timeout=300)
 def load_data():
@@ -51,23 +58,30 @@ def load_data():
             logging.info("Данные успешно загружены")
             return data
     except Exception as e:
-        logging.error(f"Ошибка при загрузке данных: {e}")
+        logging.error(f"Ошибка при загрузке данных: {e}\n{traceback.format_exc()}")
+        initialize_data_file()  # Создаем файл, если загрузка не удалась
         return {'users': {}, 'files': {}}
 
 def save_data(data):
     try:
         with open(DATA_FILE, 'w', encoding='utf-8') as file:
             json.dump(data, file, ensure_ascii=False, indent=4)
+        logging.info(f"Файл {DATA_FILE} успешно сохранен локально")
         upload_db_to_hf()
         cache.clear()
         logging.info("Данные сохранены и загружены на HF")
     except Exception as e:
-        logging.error(f"Ошибка при сохранении данных: {e}")
+        logging.error(f"Ошибка при сохранении данных: {e}\n{traceback.format_exc()}")
         raise
 
 def upload_db_to_hf():
     try:
+        if not os.path.exists(DATA_FILE):
+            logging.warning(f"Файл {DATA_FILE} не существует, пропускаем загрузку на Hugging Face")
+            initialize_data_file()  # Создаем пустой файл, чтобы избежать повторных ошибок
+            return
         api = HfApi()
+        logging.info(f"Загрузка файла {DATA_FILE} в {REPO_ID}")
         api.upload_file(
             path_or_fileobj=DATA_FILE,
             path_in_repo=DATA_FILE,
@@ -94,8 +108,7 @@ def download_db_from_hf():
     except Exception as e:
         logging.error(f"Ошибка при скачивании базы данных: {e}\n{traceback.format_exc()}")
         if not os.path.exists(DATA_FILE):
-            with open(DATA_FILE, 'w', encoding='utf-8') as f:
-                json.dump({'users': {}, 'files': {}}, f)
+            initialize_data_file()
 
 def periodic_backup():
     while True:
@@ -453,13 +466,11 @@ def dashboard():
                         file_size = os.path.getsize(temp_path)
                         
                         api = HfApi()
-                        # Используем плоскую структуру для простоты
-                        file_path = f"cloud_files/{token}_{filename}"  # Уникальное имя файла
+                        file_path = f"cloud_files/{token}_{filename}"
                         try:
                             logging.info(f"Попытка загрузки файла: {file_path} в репозиторий {REPO_ID}")
                             if not HF_TOKEN_WRITE:
                                 raise ValueError("HF_TOKEN_WRITE не установлен")
-                            # Проверка доступности репозитория и токена
                             api.repo_info(repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN_WRITE)
                             logging.info(f"Репозиторий доступен, загрузка файла: {filename}")
                             api.upload_file(
@@ -616,5 +627,6 @@ def logout():
     return redirect(url_for('login'))
 
 if __name__ == '__main__':
+    initialize_data_file()  # Инициализация файла при запуске приложения
     threading.Thread(target=periodic_backup, daemon=True).start()
     app.run(debug=True, host='0.0.0.0', port=7860)