Update app.py

app.py

@@ -4,13 +4,11 @@ import gradio as gr
 import openai
 import google.generativeai as palm
 
-#from dotenv import load_dotenv
-#load_dotenv()
-
 
 llm_api_options = ["OpenAI API","Azure OpenAI API","Google PaLM API", "Llama 2"]
 TEST_MESSAGE = "Write an introductory paragraph to explain Generative AI to the reader of this content."
-openai_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-turbo", 
+openai_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-turbo",
+ 
                      "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613", "text-davinci-003", 
                      "text-davinci-002", "text-curie-001", "text-babbage-001", "text-ada-001"]
 
@@ -19,7 +17,8 @@ google_palm_models = ["models/text-bison-001", "models/chat-bison-001","models/e
 azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
 azure_deployment_name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
 google_palm_key = os.getenv("GOOGLE_PALM_AI_API_KEY")
-
+openai_api_key = os.getenv("OPENAI_API_KEY")
+azure_openai_api_key = os.getenv("AZURE_OPENAI_KEY")
 
 def openai_text_completion(prompt: str, model: str):
     try:
@@ -34,7 +33,7 @@ def openai_text_completion(prompt: str, model: str):
             {"role": "assistant", "content": f"{assistant_prompt}"}
         ]
         
-        openai.api_key = os.getenv("OPENAI_API_KEY")
+        openai.api_key = openai_api_key
         openai.api_version = '2020-11-07'
         
         completion = openai.ChatCompletion.create(
@@ -61,7 +60,7 @@ def azure_openai_text_completion(prompt: str, model: str):
             {"role": "assistant", "content": f"{assistant_prompt}"}
         ]
         
-        openai.api_key = os.getenv("AZURE_OPENAI_KEY")
+        openai.api_key = azure_openai_api_key
         openai.api_type = "azure"
         openai.api_version = "2023-05-15" 
         openai.api_base = f"https://{azure_endpoint}.openai.azure.com"