Update app.py
Browse files
app.py
CHANGED
|
@@ -23,6 +23,12 @@ MAX_STORAGE_GB = 500
|
|
| 23 |
cache = Cache(app, config={'CACHE_TYPE': 'simple'})
|
| 24 |
logging.basicConfig(level=logging.INFO)
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
# Функции для работы с базой данных и Hugging Face
|
| 27 |
@cache.memoize(timeout=300)
|
| 28 |
def load_data():
|
|
@@ -36,7 +42,6 @@ def load_data():
|
|
| 36 |
data.setdefault('users', {})
|
| 37 |
data.setdefault('files', {})
|
| 38 |
for token, user_data in data['users'].items():
|
| 39 |
-
# Удаляем структуру папок, храним файлы напрямую
|
| 40 |
if 'folders' in user_data:
|
| 41 |
user_data['files'] = user_data['folders'].get('root', {}).get('files', [])
|
| 42 |
del user_data['folders']
|
|
@@ -450,14 +455,18 @@ def dashboard():
|
|
| 450 |
file_path = f"cloud_files/{token}/{filename}"
|
| 451 |
try:
|
| 452 |
logging.info(f"Попытка загрузки файла: {file_path} в репозиторий {REPO_ID}")
|
|
|
|
|
|
|
| 453 |
logging.info(f"Используемый токен: {HF_TOKEN_WRITE[:5]}... (скрыт для безопасности)")
|
|
|
|
|
|
|
| 454 |
api.upload_file(
|
| 455 |
path_or_fileobj=temp_path,
|
| 456 |
path_in_repo=file_path,
|
| 457 |
repo_id=REPO_ID,
|
| 458 |
repo_type="dataset",
|
| 459 |
token=HF_TOKEN_WRITE,
|
| 460 |
-
commit_message=f"Загружен файл для {token}"
|
| 461 |
)
|
| 462 |
logging.info(f"Файл {filename} успешно загружен в Hugging Face")
|
| 463 |
except Exception as e:
|
|
|
|
| 23 |
cache = Cache(app, config={'CACHE_TYPE': 'simple'})
|
| 24 |
logging.basicConfig(level=logging.INFO)
|
| 25 |
|
| 26 |
+
# Проверка токена
|
| 27 |
+
if not HF_TOKEN_WRITE:
|
| 28 |
+
logging.error("HF_TOKEN_WRITE не установлен. Убедитесь, что переменная окружения HF_TOKEN задана.")
|
| 29 |
+
else:
|
| 30 |
+
logging.info("HF_TOKEN_WRITE успешно установлен (первые 5 символов: {0}...)".format(HF_TOKEN_WRITE[:5]))
|
| 31 |
+
|
| 32 |
# Функции для работы с базой данных и Hugging Face
|
| 33 |
@cache.memoize(timeout=300)
|
| 34 |
def load_data():
|
|
|
|
| 42 |
data.setdefault('users', {})
|
| 43 |
data.setdefault('files', {})
|
| 44 |
for token, user_data in data['users'].items():
|
|
|
|
| 45 |
if 'folders' in user_data:
|
| 46 |
user_data['files'] = user_data['folders'].get('root', {}).get('files', [])
|
| 47 |
del user_data['folders']
|
|
|
|
| 455 |
file_path = f"cloud_files/{token}/{filename}"
|
| 456 |
try:
|
| 457 |
logging.info(f"Попытка загрузки файла: {file_path} в репозиторий {REPO_ID}")
|
| 458 |
+
if not HF_TOKEN_WRITE:
|
| 459 |
+
raise ValueError("HF_TOKEN_WRITE не установлен")
|
| 460 |
logging.info(f"Используемый токен: {HF_TOKEN_WRITE[:5]}... (скрыт для безопасности)")
|
| 461 |
+
# Проверка доступности репозитория
|
| 462 |
+
api.repo_info(repo_id=REPO_ID, token=HF_TOKEN_WRITE)
|
| 463 |
api.upload_file(
|
| 464 |
path_or_fileobj=temp_path,
|
| 465 |
path_in_repo=file_path,
|
| 466 |
repo_id=REPO_ID,
|
| 467 |
repo_type="dataset",
|
| 468 |
token=HF_TOKEN_WRITE,
|
| 469 |
+
commit_message=f"Загружен файл {filename} для {token}"
|
| 470 |
)
|
| 471 |
logging.info(f"Файл {filename} успешно загружен в Hugging Face")
|
| 472 |
except Exception as e:
|