Updating key part

app.py

@@ -1,7 +1,7 @@
 import os
 import openai 
 from openai import OpenAI
-from dotenv import load_dotenv, find_dotenv
+from dotenv import load_dotenv
 
 import re
 import matplotlib.pyplot as plt
@@ -187,9 +187,16 @@ def get_completion(conversation, model="gpt-4-1106-preview"):
 
     [{sample_output}]
     """
+    load_dotenv()
+    openai_api_key = os.getenv('OPENAI_API_KEY')
+
+    if openai_api_key is None:
+        raise ValueError("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
     
+
+    openai.api_key = openai_api_key
+
     client = OpenAI()
-    openai.api_key = os.environ['OPENAI_KEY']
 
     messages = [{"role": "user", "content": prompt}]
     response = client.chat.completions.create(