Update ai_generate.py

ai_generate.py

@@ -1,13 +1,15 @@
 import torch
-import openai
+from openai import OpenAI
 import os
 
 model = "gpt-3.5-turbo"
-openai.api_key = os.environ['openai_key']
+client = OpenAI(
+    api_key=os.environ.get("openai_key"),
+)
 
 def generate(text):
     message=[{"role": "user", "content": text}]
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model=model,
         messages = message,
         temperature=0.2,