# from huggingface_hub import HfApi
# import os
# import sys

# def verify_authentication():
#     api = HfApi()
#     try:
#         user_info = api.whoami()
#         print(f"Authenticated as: {user_info['name']}")
#         return True
#     except Exception as e:
#         print("Authentication failed. Please ensure you're logged in to Hugging Face.")
#         print(f"Error: {e}")
#         return False

# def upload_to_hf_space(api, local_path, repo_id, path_in_repo):
#     try:
#         url = api.upload_file(
#             path_or_fileobj=local_path,
#             path_in_repo=path_in_repo,
#             repo_id=repo_id,
#             token=os.environ.get("HF_TOKEN"),
#             repo_type="space",
#         )
#         print(f"File uploaded successfully: {local_path} -> {url}")
#     except Exception as e:
#         print(f"An error occurred while uploading {local_path}: {e}")

# def upload_folder(api, repo_id, local_dir, folder_name):
#     folder_path = os.path.join(local_dir, folder_name)
#     for root, dirs, files in os.walk(folder_path):
#         for file in files:
#             local_path = os.path.join(root, file)
#             relative_path = os.path.relpath(local_path, local_dir)
#             upload_to_hf_space(api, local_path, repo_id, relative_path)

# def upload_items(repo_id, local_dir, items):
#     if not verify_authentication():
#         sys.exit(1)

#     api = HfApi()

#     for item in items:
#         full_path = os.path.join(local_dir, item)
#         if os.path.isfile(full_path):
#             upload_to_hf_space(api, full_path, repo_id, item)
#         elif os.path.isdir(full_path):
#             upload_folder(api, repo_id, local_dir, item)
#         else:
#             print(f"The item {item} does not exist in {local_dir}.")

# if __name__ == "__main__":
#     repo_id = "liuganghuggingface/polymer-design"
#     local_dir = "./"  # Use the current directory
#     items_to_upload = ["app.py", "requirements.txt", "model_labeled", "model_all"]  # List of files and folders to upload

#     # Ensure the HF_TOKEN is set
#     if "HF_TOKEN" not in os.environ:
#         print("Please set the HF_TOKEN environment variable with your Hugging Face token.")
#         sys.exit(1)

#     upload_items(repo_id, local_dir, items_to_upload)



from huggingface_hub import HfApi
import os
import sys

def upload_single_file(api, repo_id, local_path, remote_path):
    print(f"Attempting to upload: {local_path} to {remote_path}")
    try:
        if not os.path.exists(local_path):
            print(f"Error: Local file does not exist: {local_path}")
            return

        url = api.upload_file(
            path_or_fileobj=local_path,
            path_in_repo=remote_path,
            repo_id=repo_id,
            token=os.environ.get("HF_TOKEN"),
            repo_type="space",
        )
        print(f"File uploaded successfully: {local_path} -> {url}")
        
        # Verify the upload
        files = api.list_repo_files(repo_id=repo_id, repo_type="space")
        if remote_path in files:
            print(f"Verified: {remote_path} is present in the repository.")
        else:
            print(f"Warning: {remote_path} was uploaded but not found in the repository listing.")
    except Exception as e:
        print(f"An error occurred while uploading {local_path}: {e}")

def main():
    repo_id = "liuganghuggingface/polymer-design"
    local_dir = "./"  # Use the current directory
    
    files_to_upload = [
        ("model_labeled/data.meta.json", "model_labeled/data.meta.json"),
        ("model_all/config.yaml", "model_all/config.yaml"),
        ("model_all/model.pt", "model_all/model.pt")
    ]

    api = HfApi()

    for local_path, remote_path in files_to_upload:
        full_local_path = os.path.join(local_dir, local_path)
        upload_single_file(api, repo_id, full_local_path, remote_path)

if __name__ == "__main__":
    if "HF_TOKEN" not in os.environ:
        print("Please set the HF_TOKEN environment variable with your Hugging Face token.")
        sys.exit(1)

    main()