add finish

app.py

@@ -63,7 +63,7 @@ def showInput(input, chatbot):
     return chatbot
 
 
-def predict(input, chatbot, messages, idx):
+def predict(input, chatbot, messages, idx, answer):
     chatbot.append((parse_text(input), ""))
     messages.append({"role": 'user', "content": input})
     completion = openai.ChatCompletion.create(
@@ -72,15 +72,27 @@ def predict(input, chatbot, messages, idx):
         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, '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})
-    # messages1 = messages + [{"role": 'user', "content": '我猜到汤底了吗？请回答是或否。'}]
+    messages1 = messages + [{"role": 'user', "content": '我猜到汤底了吗？请回答是或否。'}]
+    completion1 = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages1,
+        logit_bias={42468: 10, 28938: 10}
+    )
+    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}
+    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
 
 
@@ -125,20 +137,20 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=4):
             with gr.Column(scale=12):
-                user_input = gr.Textbox(show_label=False, placeholder="Input...", lines=7, max_lines=7).style(
+                user_input = gr.Textbox(show_label=False, placeholder="Input...", lines=3, max_lines=3).style(
                     container=False)
             with gr.Column(min_width=32, scale=1):
                 submitBtn = gr.Button("Submit", variant="primary")
             emptyBtn = gr.Button("New game")
         with gr.Column(scale=1):
-            answer_output = gr.Textbox(show_label=False, lines=10, max_lines=10).style(
+            answer_output = gr.Textbox(show_label=False, lines=6, max_lines=6).style(
                 container=False)
             answerBtn = gr.Button("Show Answer")
 
     messages = gr.State(data['messages'])
 
     # submitBtn.click(showInput, [user_input, chatbot], [chatbot])
-    submitBtn.click(predict, [user_input, chatbot, messages, idx], [chatbot, messages],
+    submitBtn.click(predict, [user_input, chatbot, messages, idx, answer], [chatbot, messages],
                     show_progress=True)
     submitBtn.click(reset_user_input, [], [user_input])