Update ai_generate.py

ai_generate.py

@@ -3,16 +3,12 @@ from openai import OpenAI
 import os
 from transformers import pipeline
 
-client = OpenAI(
-    api_key=os.environ.get("openai_key"),
-)
-
 pipes = {
     'GPT-Neo': pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B"),
     'Llama 3': pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
 }
 
-def generate(text, model):
+def generate(text, model, api):
     if model is "GPT-Neo":
         response = pipes[model](text)
         return response[0]
@@ -20,6 +16,9 @@ def generate(text, model):
         response = pipes[model](text)
         return response[0] 
     elif model is "OpenAI GPT 3.5":
+        client = OpenAI(
+            api_key=api,
+        )
         message=[{"role": "user", "content": text}]
         response = client.chat.completions.create(
             model="gpt-3.5-turbo",
@@ -30,6 +29,9 @@ def generate(text, model):
         )
         return response[0].message.content
     elif model is "OpenAI GPT 4":
+        client = OpenAI(
+            api_key=api,
+        )
         message=[{"role": "user", "content": text}]
         response = client.chat.completions.create(
             model="gpt-4-turbo",
@@ -40,6 +42,9 @@ def generate(text, model):
         )
         return response[0].message.content
     elif model is "OpenAI GPT 4o":
+        client = OpenAI(
+            api_key=api,
+        )
         message=[{"role": "user", "content": text}]
         response = client.chat.completions.create(
             model="gpt-4o",