From e8a893156e097d8c964d0a8ff195b00fdb536fad Mon Sep 17 00:00:00 2001
From: Christopher Carroll Smith
 <75859865+chriscarrollsmith@users.noreply.github.com>
Date: Wed, 12 Apr 2023 13:45:54 -0400
Subject: [PATCH] Implemented .env file API key storage

---
 .env.sample  | 1 +
 .gitignore   | 2 ++
 README.md    | 5 ++++-
 wolverine.py | 8 ++++----
 4 files changed, 11 insertions(+), 5 deletions(-)
 create mode 100644 .env.sample

diff --git a/.env.sample b/.env.sample
new file mode 100644
index 0000000..3ad03db
--- /dev/null
+++ b/.env.sample
@@ -0,0 +1 @@
+OPENAI_API_KEY=your_api_key
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
index 673fe09..32fabad 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
 venv
 openai_key.txt
+.venv
+.env
\ No newline at end of file
diff --git a/README.md b/README.md
index 55813f7..c4f589f 100644
--- a/README.md
+++ b/README.md
@@ -13,8 +13,11 @@ For a quick demonstration see my [demo video on twitter](https://twitter.com/bio
     python3 -m venv venv
     source venv/bin/activate
     pip install -r requirements.txt
+    cp .env.sample .env
 
-Add your openAI api key to `openai_key.txt` - _warning!_ by default this uses GPT-4 and may make many repeated calls to the api.
+Add your openAI api key to `.env` 
+
+_warning!_ By default wolverine uses GPT-4 and may make many repeated calls to the api.
 
 ## Example Usage
 
diff --git a/wolverine.py b/wolverine.py
index 42f7ff8..f1d910d 100644
--- a/wolverine.py
+++ b/wolverine.py
@@ -5,13 +5,14 @@ import os
 import shutil
 import subprocess
 import sys
-
 import openai
 from termcolor import cprint
+from dotenv import load_dotenv
+
 
 # Set up the OpenAI API
-with open("openai_key.txt") as f:
-    openai.api_key = f.read().strip()
+load_dotenv()
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
 def run_script(script_name, script_args):
@@ -51,7 +52,6 @@ def send_error_to_gpt(file_path, args, error_message, model):
     )
 
     # print(prompt)
-
     response = openai.ChatCompletion.create(
         model=model,
         messages=[