#!/usr/bin/env python3
'''
Upload script for djmgnn-base
'''

from huggingface_hub import HfApi, create_repo

def upload_model():
    # Initialize API
    api = HfApi()
    
    # Create repository (if it doesn't exist)
    repo_id = "saketh11/MoML-CA"
    
    try:
        create_repo(repo_id, exist_ok=True)
        print(f"Repository {repo_id} created/verified")
    except Exception as e:
        print(f"Error creating repository: {e}")
        return
    
    # Upload files to base_model subdirectory
    api.upload_folder(
        folder_path="hf_models/djmgnn_base",
        repo_id=repo_id,
        path_in_repo="base_model",
        commit_message="Upload djmgnn-base model"
    )
    
    print(f"Model uploaded successfully to https://huggingface.co/{repo_id}")

if __name__ == "__main__":
    upload_model()