Update app.py

app.py

@@ -7,6 +7,9 @@ import random
 import time
 
 
+handshake_url = "http://127.0.0.1:8000/chatbot/"
+dialog_url = "http://127.0.0.1:8000/chatbot/dialoginfo"
+finish_query_url = "http://127.0.0.1:8000/chatbot/finishquery"
 PRESET_ANSWERS = ["刚到美国的时候，觉得美国人像傻子一样，到处都是漏洞。任何地方的厕所都有免费纸，有些人定期去扯很多回家，纸都不用买。快餐店的饮料，有的可以无限续杯，有些几个人买一份饮料，接回来灌到各自的杯子里；等等。",
                   "尽管美国有许多“漏洞”，但作为超级大国，显然能带给人以无尽的故事与思考。我来分享一下哪些是去了美国才知道的事，主题主要围绕着生活、衣食住行、文化冲击、教育医疗等展开叙说，本文有5千字左右，你也可以跳到感兴趣的部分阅读。"
                   "美国的城市风貌与基础设施1、去到了美国才知道，纽约的城市样貌跟我想象的发达不一样，真实的纽约街景是很嘈杂和市井。例如，在曼哈顿区路旁，随处可见的小摊位，卖鲜花的、卖各种小食、卖自制首饰的，卖艺术品等等。我留意一下，发现每个路边摊都有合法的营业执照。"]
@@ -33,16 +36,36 @@ def get_chat_answer(query, chatbot):
     yield "", chatbot
 
 
+def get_chat_answer_request(query, chatbot):
+    q_id = str(uuid.uuid1())
+    r = requests.post(handshake_url, json={"q_id": q_id, "query": query, 'history': chatbot})
+    if r.json()['code'] == 202:
+        for _ in range(1000):
+            r = requests.get(dialog_url, {"dialogid": q_id})
+            history = r.json()['history']
+            print(history)
+            while len(history[-1]) == 0:
+                    history.pop()
+            r_finish = requests.get(finish_query_url, {"dialogid": q_id})
+            if r_finish.json() == True:
+                yield "", history
+                print('break')
+                break
+
+            yield "", history
+            time.sleep(1)
+
+
 css = """
 col-container {max-width: 510px; margin-left: auto; margin-right: auto; margin-up: auto; margin-bottom: auto;}
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        with gr.Tab("🧠 模型对话 Dialog", elem_id="od-benchmark-tab-table"):
+        with gr.Tab("🧠 模型对话 Dialog"):
             chatbot = gr.Chatbot([[None, get_internet_ip()]], show_label=False, elem_id="chatbot")
             query = gr.Textbox(show_label=False, placeholder="请输入提问内容，按回车进行提交")
             query.submit(get_chat_answer,[query, chatbot], [query, chatbot])
-        with gr.Tab("🎚️ 模型微调 Fine-tune", elem_id="od-benchmark-tab-table"):
+        with gr.Tab("🎚️ 模型微调 Fine-tune"):
             with gr.Row():
                 with gr.Column():
                     ft_task_name = gr.Textbox(value="", label="任务名称")
@@ -51,7 +74,8 @@ with gr.Blocks(css=css) as demo:
                     ft_epochs = gr.Textbox(value="", label="epochs")
                     ft_start_lr = gr.Textbox(value="", label="start LR")
                     ft_end_lr = gr.Textbox(value="", label="end LR")
-        with gr.Tab("📈 信息 Info", elem_id="od-benchmark-tab-table"):
+                    submit_btn = gr.Button(value="Submit")
+        with gr.Tab("📈 信息 Info"):
             gr.Markdown(METRICS_TAB_TEXT, elem_classes="markdown-text")
 
 demo.queue(max_size=20).launch()