add term

app.py

@@ -60,7 +60,7 @@ def CheckTrue(chatbot, key):
     data = {'AgentCheck': True, 'key': key}
     response=requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
     if not response.status_code == 200:
-        chatbot.append((parse_text("出现错误，请联系负责人"), None))
+        chatbot.append((parse_text(str(response.content, encoding="utf-8")), None))
         return chatbot
     chatbot.append((parse_text("是"), parse_text(str(response.content, encoding="utf-8"))))
     return chatbot
@@ -69,7 +69,7 @@ def CheckFalse(chatbot, key):
     data = {'AgentCheck': False, 'key': key}
     response=requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
     if not response.status_code == 200:
-        chatbot.append((parse_text("出现错误，请联系负责人"), None))
+        chatbot.append((parse_text(str(response.content, encoding="utf-8")), None))
         return chatbot
     chatbot.append((parse_text("否"), parse_text(str(response.content, encoding="utf-8"))))
     return chatbot
@@ -77,16 +77,25 @@ def CheckFalse(chatbot, key):
 def CheckTerm(chatbot, key):
     data = {'AgentFinish': True, 'key': key}
     response=requests.post(os.environ.get("URL"), data=json.dumps(data, ensure_ascii=False).encode('utf-8'))
+    msg = str(response.content, encoding="utf-8")
     if not response.status_code == 200:
-        chatbot.append((parse_text("出现错误，请联系负责人"), None))
-        return chatbot
-    chatbot = [(None, parse_text(str(response.content, encoding="utf-8")))]
-    return chatbot
+        chatbot.append((None, parse_text(msg)))
+        return chatbot, gr.update(value=msg), gr.update(value=msg)
+    response = json.loads(msg)
+    chatbot = [(None, parse_text(response['response']))]
+    return chatbot, gr.update(value=response['story']), gr.update(value=response['answer'])
 
 with gr.Blocks() as demo:
     gr.HTML("""<h1 align="center">Generative Agents测试平台</h1>""")
     user_key = gr.Textbox(label='Key', placeholder="Input your key, press enter to apply", lines=1, max_lines=1).style(
         container=False)
+    with gr.Row():
+        with gr.Column(scale=1):
+            story = gr.Textbox(label='Story', lines=3, max_lines=3).style(
+                container=False)
+        with gr.Column(scale=1):
+            answer = gr.Textbox(label='Answer', lines=3, max_lines=3).style(
+                container=False)
     chatbot = gr.Chatbot([])
 
     with gr.Row():
@@ -95,11 +104,11 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             falseBtn = gr.Button('否')
         with gr.Column(scale=1):
-            termBtn = gr.Button('得到正确结果')
+            termBtn = gr.Button('达到终止条件')
         
-    user_key.submit(CheckTerm, [chatbot, user_key], [chatbot])
+    user_key.submit(CheckTerm, [chatbot, user_key], [chatbot, story, answer])
     trueBtn.click(CheckTrue, [chatbot, user_key], [chatbot])
     falseBtn.click(CheckFalse, [chatbot, user_key], [chatbot])
-    termBtn.click(CheckTerm, [chatbot, user_key], [chatbot])
+    termBtn.click(CheckTerm, [chatbot, user_key], [chatbot, story, answer])
     
 demo.queue().launch()