Starting better version

(And not whatever hacky version I first though of)

yolo.py

@@ -12,8 +12,6 @@ import subprocess
 from termcolor import colored
 from colorama import init
 
-import dotenv
-
 # Check if the user globally disabled the safety switch
 def get_yolo_safety_switch_config():
   
@@ -34,7 +32,7 @@ def get_full_prompt(user_prompt, shell):
   prompt_path = os.path.dirname(yolo_path)
 
   ## Load the prompt and prep it
-  prompt_file = os.path.join(prompt_path, "yolo_prompt.txt")
+  prompt_file = os.path.join(prompt_path, "prompt.txt")
   pre_prompt = open(prompt_file,"r").read()
   pre_prompt = pre_prompt.replace("{shell}", shell)
   pre_prompt = pre_prompt.replace("{os}", get_os_friendly_name())
@@ -80,9 +78,10 @@ if __name__ == "__main__":
   ask_flag = False           # safety switch -a command line argument
   yolo = ""                  # user's answer to safety switch (-a) question y/n
 
-  # Presumes the folder yolo.py is copied to has a .env
-  dotenv.load_dotenv(".env")
+  # Two options for the user to specify they openai api key
+  home_path = os.path.expanduser("~")
   openai.api_key = os.getenv("OPENAI_API_KEY")
+  openai.api_key_path = os.path.join(home_path,".openai.apikey")
 
   # Parse arguments and make sure we have at least a single word
   if len(sys.argv) < 2:
@@ -154,4 +153,4 @@ if yolo == "Y" or yolo == "":
     subprocess.run([shell, "/c", res_command], shell=False)  
   else: 
     # Unix: /bin/bash /bin/zsh: uses -c both Ubuntu and macOS should work, others might not
-    subprocess.run([shell, "-c", res_command], shell=False)
+    subprocess.run([shell, "-c", res_command], shell=False)