Merge pull request #3 from JarrettSJohnson/api_key_envvar

PLUGIN_DIR = os.path.dirname(os.path.abspath(__file__))

API_KEY_FILE = os.path.join(PLUGIN_DIR, "apikey.txt")

OPENAI_KEY_ENV = "OPENAI_API_KEY"

def set_api_key(api_key):

    api_key = api_key.strip()

    print("API key set and saved to file successfully.")

def load_api_key():

    api_key = os.getenv(OPENAI_KEY_ENV)

    if not api_key:

        try:

            with open(API_KEY_FILE, "r") as api_key_file:

                api_key = api_key_file.read().strip()

                openai.api_key = api_key

                print("API key loaded from file.")

        except FileNotFoundError:

        print("API key file not found. Please set your API key using 'set_api_key your_api_key_here' command.")

            print("API key file not found. Please set your API key using 'set_api_key your_api_key_here' command" +

                  f" or by environment variable '{OPENAI_KEY_ENV}'.")

load_api_key()