Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template_string, request, redirect, url_for, session, flash, jsonify
+from flask import Flask, render_template_string, request, redirect, url_for, session, flash
 from flask_caching import Cache
 import json
 import os
@@ -10,11 +10,10 @@ from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
 import random
 import string
-import traceback
 
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey")
-DATA_FILE = 'cloud_data.json'  # Can be changed to an absolute path if needed, e.g., os.path.join(os.getcwd(), 'cloud_data.json')
+DATA_FILE = '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,14 +29,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():
     try:
@@ -58,30 +56,28 @@ def load_data():
             logging.info("Данные успешно загружены")
             return data
     except Exception as e:
-        logging.error(f"Ошибка при загрузке данных: {e}\n{traceback.format_exc()}")
-        initialize_data_file()  # Создаем файл, если загрузка не удалась
+        logging.error(f"Ошибка при загрузке данных: {e}")
+        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} успешно сохранен локально")
+        logging.info(f"Данные успешно сохранены в {DATA_FILE}")
         upload_db_to_hf()
         cache.clear()
-        logging.info("Данные сохранены и загружены на HF")
     except Exception as e:
-        logging.error(f"Ошибка при сохранении данных: {e}\n{traceback.format_exc()}")
+        logging.error(f"Ошибка при сохранении данных: {e}")
         raise
 
 def upload_db_to_hf():
     try:
         if not os.path.exists(DATA_FILE):
-            logging.warning(f"Файл {DATA_FILE} не существует, пропускаем загрузку на Hugging Face")
-            initialize_data_file()  # Создаем пустой файл, чтобы избежать повторных ошибок
+            logging.warning(f"Файл {DATA_FILE} не существует, пропускаем загрузку")
+            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,
@@ -92,7 +88,7 @@ def upload_db_to_hf():
         )
         logging.info("База данных загружена на Hugging Face")
     except Exception as e:
-        logging.error(f"Ошибка при загрузке базы данных: {e}\n{traceback.format_exc()}")
+        logging.error(f"Ошибка при загрузке базы данных: {e}")
 
 def download_db_from_hf():
     try:
@@ -106,20 +102,18 @@ def download_db_from_hf():
         )
         logging.info("База данных скачана с Hugging Face")
     except Exception as e:
-        logging.error(f"Ошибка при скачивании базы данных: {e}\n{traceback.format_exc()}")
+        logging.error(f"Ошибка при скачивании базы данных: {e}")
         if not os.path.exists(DATA_FILE):
             initialize_data_file()
 
 def periodic_backup():
     while True:
         upload_db_to_hf()
-        time.sleep(1800)
+        time.sleep(1800)  # 30 минут
 
-# Генерация 13-значного токена
 def generate_token():
     return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
 
-# Определение типа файла для предпросмотра
 def get_file_type(filename):
     video_extensions = ('.mp4', '.mov', '.avi')
     image_extensions = ('.jpg', '.jpeg', '.png', '.gif')
@@ -129,7 +123,6 @@ def get_file_type(filename):
         return 'image'
     return 'other'
 
-# Базовый стиль CSS (без изменений)
 BASE_STYLE = '''
 :root {
     --primary: #ff4d6d;
@@ -331,7 +324,6 @@ body.dark .progress-text, body.dark .storage-text {
 }
 '''
 
-# Регистрация через /admhosto
 @app.route('/admhosto', methods=['GET', 'POST'])
 def register():
     if request.method == 'POST':
@@ -380,7 +372,6 @@ def register():
 '''
     return render_template_string(html)
 
-# Главная страница с вводом токена
 @app.route('/', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
@@ -429,7 +420,6 @@ def login():
 '''
     return render_template_string(html)
 
-# Личный dashboard
 @app.route('/dashboard', methods=['GET', 'POST'])
 def dashboard():
     if 'token' not in session:
@@ -457,22 +447,19 @@ def dashboard():
             if storage_used + total_size > max_storage_bytes:
                 flash('Недостаточно места для загрузки файлов!')
             else:
+                uploads_dir = 'uploads'
+                os.makedirs(uploads_dir, exist_ok=True)
                 for file in files:
                     if file and file.filename:
                         filename = secure_filename(file.filename)
-                        temp_path = os.path.join('uploads', filename)
-                        os.makedirs('uploads', exist_ok=True)
+                        temp_path = os.path.join(uploads_dir, filename)
                         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"Попытка загрузки файла: {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(
                                 path_or_fileobj=temp_path,
                                 path_in_repo=file_path,
@@ -483,7 +470,7 @@ def dashboard():
                             )
                             logging.info(f"Файл {filename} успешно загружен в Hugging Face")
                         except Exception as e:
-                            logging.error(f"Ошибка при загрузке файла на Hugging Face: {e}\n{traceback.format_exc()}")
+                            logging.error(f"Ошибка при загрузке файла на Hugging Face: {e}")
                             flash(f"Ошибка при загрузке файла {filename}: {str(e)}")
                             if os.path.exists(temp_path):
                                 os.remove(temp_path)
@@ -620,13 +607,12 @@ def dashboard():
         MAX_STORAGE_GB=MAX_STORAGE_GB
     )
 
-# Выход
 @app.route('/logout')
 def logout():
     session.pop('token', None)
     return redirect(url_for('login'))
 
 if __name__ == '__main__':
-    initialize_data_file()  # Инициализация файла при запуске приложения
+    initialize_data_file()
     threading.Thread(target=periodic_backup, daemon=True).start()
     app.run(debug=True, host='0.0.0.0', port=7860)