from huggingface_hub import HfApi

api = HfApi()

print("Creating dataset repo...")
api.create_repo('pharma-agent-db', repo_type='dataset', private=False)

print("Uploading drugbank_lite.db (this will take a few minutes)...")
api.upload_file(
    path_or_fileobj='drugbank_lite.db',
    path_in_repo='drugbank_lite.db',
    repo_id='Arshdeep2k5/pharma-agent-db',
    repo_type='dataset'
)

print("Done! Database uploaded to HuggingFace.")