Eluza133 commited on
Commit
1dab35c
·
verified ·
1 Parent(s): 840dca4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
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: