Update app.py

app.py

@@ -27,9 +27,9 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 if not HF_TOKEN_WRITE:
     logging.error("HF_TOKEN_WRITE не установлен. Убедитесь, что переменная окружения HF_TOKEN задана.")
 else:
-    logging.info("HF_TOKEN_WRITE успешно установлен (первые 5 символов: {0}...)".format(HF_TOKEN_WRITE[:5]))
+    logging.info(f"HF_TOKEN_WRITE установлен (первые 5 символов: {HF_TOKEN_WRITE[:5]}...)")
 
-# Инициализация файла базы данных при старте
+# Инициализация файла базы данных только если он не существует
 def initialize_data_file():
     if not os.path.exists(DATA_FILE):
         logging.info(f"Файл {DATA_FILE} не найден, создается новый.")
@@ -39,26 +39,35 @@ def initialize_data_file():
 @cache.memoize(timeout=300)
 def load_data():
     try:
-        download_db_from_hf()
-        with open(DATA_FILE, 'r', encoding='utf-8') as file:
-            data = json.load(file)
-            if not isinstance(data, dict):
-                logging.warning("Данные не в формате dict, инициализация пустой базы")
-                return {'users': {}, 'files': {}}
-            data.setdefault('users', {})
-            data.setdefault('files', {})
-            for token, user_data in data['users'].items():
-                if 'folders' in user_data:
-                    user_data['files'] = user_data['folders'].get('root', {}).get('files', [])
-                    del user_data['folders']
-                if 'storage_used' not in user_data:
-                    user_data['storage_used'] = 0
-            logging.info("Данные успешно загружены")
-            return data
+        # Сначала пытаемся загрузить локальный файл, если он существует
+        if os.path.exists(DATA_FILE):
+            with open(DATA_FILE, 'r', encoding='utf-8') as file:
+                data = json.load(file)
+                logging.info("Данные загружены из локального файла")
+        else:
+            # Если локального файла нет, скачиваем с Hugging Face
+            download_db_from_hf()
+            with open(DATA_FILE, 'r', encoding='utf-8') as file:
+                data = json.load(file)
+                logging.info("Данные загружены с Hugging Face")
+
+        if not isinstance(data, dict):
+            logging.warning("Данные не в формате dict, инициализация пустой базы")
+            return {'users': {}, 'files': {}}
+        data.setdefault('users', {})
+        data.setdefault('files', {})
+        for token, user_data in data['users'].items():
+            if 'folders' in user_data:
+                user_data['files'] = user_data['folders'].get('root', {}).get('files', [])
+                del user_data['folders']
+            if 'storage_used' not in user_data:
+                user_data['storage_used'] = 0
+        return data
     except Exception as e:
         logging.error(f"Ошибка при загрузке данных: {e}")
         initialize_data_file()
-        return {'users': {}, 'files': {}}
+        with open(DATA_FILE, 'r', encoding='utf-8') as file:
+            return json.load(file)
 
 def save_data(data):
     try:
@@ -75,7 +84,6 @@ def upload_db_to_hf():
     try:
         if not os.path.exists(DATA_FILE):
             logging.warning(f"Файл {DATA_FILE} не существует, пропускаем загрузку")
-            initialize_data_file()
             return
         api = HfApi()
         api.upload_file(
@@ -86,9 +94,9 @@ def upload_db_to_hf():
             token=HF_TOKEN_WRITE,
             commit_message=f"Бэкап {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
         )
-        logging.info("База данных загружена на Hugging Face")
+        logging.info("База данных успешно загружена на Hugging Face")
     except Exception as e:
-        logging.error(f"Ошибка при загрузке базы данных: {e}")
+        logging.error(f"Ошибка при загрузке базы данных на Hugging Face: {e}")
 
 def download_db_from_hf():
     try:
@@ -100,7 +108,7 @@ def download_db_from_hf():
             local_dir=".",
             local_dir_use_symlinks=False
         )
-        logging.info("База данных скачана с Hugging Face")
+        logging.info("База данных успешно скачана с Hugging Face")
     except Exception as e:
         logging.error(f"Ошибка при скачивании базы данных: {e}")
         if not os.path.exists(DATA_FILE):
@@ -344,7 +352,7 @@ def register():
     
     html = '''
 <!DOCTYPE html>
-<html lang="en">
+<html lang="ru">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
@@ -385,7 +393,7 @@ def login():
     
     html = '''
 <!DOCTYPE html>
-<html lang="en">
+<html lang="ru">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
@@ -456,10 +464,10 @@ def dashboard():
                         file.save(temp_path)
                         file_size = os.path.getsize(temp_path)
                         
-                        # Логика загрузки файла на Hugging Face (взята из Code 1)
                         api = HfApi()
                         file_path = f"cloud_files/{token}_{filename}"
                         try:
+                            logging.info(f"Попытка загрузки файла: {filename} в {file_path}")
                             api.upload_file(
                                 path_or_fileobj=temp_path,
                                 path_in_repo=file_path,
@@ -479,7 +487,7 @@ def dashboard():
                         file_info = {
                             'filename': filename,
                             'path': file_path,
-                            'type': get_file_type(filename),
+                            ' éxtra': get_file_type(filename),
                             'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                             'size': file_size
                         }
@@ -495,7 +503,7 @@ def dashboard():
 
     html = '''
 <!DOCTYPE html>
-<html lang="en">
+<html lang="ru">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
@@ -613,6 +621,7 @@ 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)