from huggingface_hub import HfApi, create_repo, HfFileSystem
from dotenv import load_dotenv
import os

load_dotenv('.env')

HF_TOKEN = os.getenv('TOKEN')


REPO_NAME = "Barzabel777/test_model2"

# Создаем репозиторий через API
create_repo(
    repo_id=REPO_NAME,
    token=HF_TOKEN,
    exist_ok=True,  # Если True, не будет ошибки при существующем репозитории
    repo_type="model",  # Может быть: model, dataset, space
)

# Инициализируем API
api = HfApi(token=HF_TOKEN)

# Загружаем README.md
api.upload_file(
    path_or_fileobj="data/README.md",  # Локальный путь к файлу
    path_in_repo="README.md",     # Путь в репозитории
    repo_id=REPO_NAME,
    repo_type="model",
)

# Загружаем config.json
api.upload_file(
    path_or_fileobj="data/test_data.json",
    path_in_repo="config.json",
    repo_id=REPO_NAME,
    repo_type="model",
)

print(f"Репозиторий создан: https://huggingface.co/{REPO_NAME}")


fs = HfFileSystem()

repo_path = "Barzabel777/test_model2"  


files = fs.ls(repo_path, detail=False)

print("Список файлов:")
for file in files:
    print(file)