change openai

app.py

@@ -8,7 +8,7 @@ import time
 import openai
 import requests
 
-# openai.api_key = os.environ.get('APIKEY')
+openai.api_key = os.environ.get('APIKEY')
 rd.seed(time.time())
 
 def postprocess(self, y):
@@ -66,20 +66,20 @@ def showInput(input, chatbot):
 def predict(input, chatbot, messages, idx):
     chatbot.append((parse_text(input), ""))
     messages.append({"role": 'user', "content": input})
-    # completion = openai.ChatCompletion.create(
-    #     model="gpt-3.5-turbo",
-    #     messages=messages,
-    #     logit_bias={42468: 10, 28938: 10}
-    # )
-    # response=completion.choices[0].message.content.strip()
-    # chatbot[-1] = (parse_text(input), parse_text(response))
-    # messages.append({"role": "assistant", "content": response})
-    # data = {'predict': messages}
-    # requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
-    data = {'predict': messages, 'idx': idx}
-    response=str(requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8')).content, encoding="utf-8")
+    completion = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        logit_bias={42468: 20, 28938: 20}
+    )
+    response=completion.choices[0].message.content.strip()
     chatbot[-1] = (parse_text(input), parse_text(response))
     messages.append({"role": "assistant", "content": response})
+    data = {'predict': messages, 'idx': idx}
+    requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
+    # data = {'predict': messages, 'idx': idx}
+    # response=str(requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8')).content, encoding="utf-8")
+    # chatbot[-1] = (parse_text(input), parse_text(response))
+    # messages.append({"role": "assistant", "content": response})
     return chatbot, messages