add api

app.py

@@ -66,33 +66,34 @@ def showInput(input, chatbot):
 def predict(input, chatbot, messages, idx, answer):
     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()
-    messages.append({"role": "assistant", "content": response})
-    # 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})
-    messages1 = messages + [{"role": 'user', "content": '我猜到汤底了吗？请回答是或否。'}]
-    completion1 = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages1,
-        # logit_bias={42468: 5, 28938: 5}
-    )
-    response1=completion1.choices[0].message.content.strip()
-    if '不' in response1 or '否' in response1 or '没' in response1:
-        response = response
-        finish = False
+    data = {'predict': messages, 'idx': idx, 'isfinished': False, 'answer': answer}
+    response=str(requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8')).content, encoding="utf-8")
+    if not response == 'timeout':
+        chatbot[-1] = (parse_text(input), parse_text(response))
+        messages.append({"role": "assistant", "content": response})
     else:
-        response += f'恭喜您已经猜到汤底，汤底是：{answer}\n点击"New Game"按钮开始下一局游戏。'
-        finish = True
-    data = {'predict': messages, 'idx': idx, 'isfinished': finish}
-    requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
-    chatbot[-1] = (parse_text(input), parse_text(response))
+        completion = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=messages,
+            logit_bias={42468: 10, 28938: 10}
+        )
+        response=completion.choices[0].message.content.strip()
+        messages.append({"role": "assistant", "content": response})
+        messages1 = messages + [{"role": 'user', "content": '我猜到汤底了吗？请回答是或否。'}]
+        completion1 = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=messages1,
+        )
+        response1=completion1.choices[0].message.content.strip()
+        if '不' in response1 or '否' in response1 or '没' in response1:
+            response = response
+            finish = False
+        else:
+            response += f'恭喜您已经猜到汤底，汤底是：{answer}\n点击"New Game"按钮开始下一局游戏。'
+            finish = True
+        data = {'predict': messages, 'idx': idx, 'isfinished': finish, 'answer': answer}
+        requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
+        chatbot[-1] = (parse_text(input), parse_text(response))
     return chatbot, messages