cometml_key.py

import os

try:
    # noinspection PyUnresolvedReferences
    from kaggle_secrets import UserSecretsClient
except ImportError:
    using_kaggle = False
else:  # if we are using kaggle, we need to set the api key
    using_kaggle = True


def get_comet_api_key() -> str:
    """
    Returns the Comet API key from the file "final_project/evaluate/comet_api_key.txt"
    if this file does not exist, asks the user to enter the key manually and saves it to the file
    """
    if using_kaggle:
        return UserSecretsClient().get_secret("comet_ml_api_key")
    if os.getcwd() == "/content":
        # if running on colab
        api_key_file = "final_project/evaluation/comet_ml_api_key.txt"
    else:
        # if running locally
        api_key_file = "comet_ml_api_key.txt"
    if os.path.exists(api_key_file):
        with open(api_key_file, "r") as f:
            return f.read().strip()
    api_key = input("Please enter your api_key for comet ml: ")
    with open(api_key_file, "w") as f:
        f.write(api_key)
    return api_key