Update app.py

app.py

@@ -1,13 +1,16 @@
-import requests
 import gradio as gr
-import time
+from constants import METRICS_TAB_TEXT
+import requests
 import re
 import uuid
-import os
+import random
+import time
+
+
+PRESET_ANSWERS = ["刚到美国的时候，觉得美国人像傻子一样，到处都是漏洞。任何地方的厕所都有免费纸，有些人定期去扯很多回家，纸都不用买。快餐店的饮料，有的可以无限续杯，有些几个人买一份饮料，接回来灌到各自的杯子里；等等。",
+                  "尽管美国有许多“漏洞”，但作为超级大国，显然能带给人以无尽的故事与思考。我来分享一下哪些是去了美国才知道的事，主题主要围绕着生活、衣食住行、文化冲击、教育医疗等展开叙说，本文有5千字左右，你也可以跳到感兴趣的部分阅读。"
+                  "美国的城市风貌与基础设施1、去到了美国才知道，纽约的城市样貌跟我想象的发达不一样，真实的纽约街景是很嘈杂和市井。例如，在曼哈顿区路旁，随处可见的小摊位，卖鲜花的、卖各种小食、卖自制首饰的，卖艺术品等等。我留意一下，发现每个路边摊都有合法的营业执照。"]
 
-handshake_url = os.environ['HANDSHAKE_URL']
-dialog_url = os.environ['DIALOG_URL']
-finish_query_url = os.environ['FINISH_QUERY_URL']
 
 def get_internet_ip():
     r = requests.get("http://txt.go.sohu.com/ip/soip")
@@ -17,30 +20,38 @@ def get_internet_ip():
     return None
 
 
-def get_remote_stream(query, chatbot):
+def get_chat_answer(query, chatbot):
+    chatbot.append([query, ""])
     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']
-            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(0.5)
-
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot([[None, get_internet_ip()]], show_label=False, height=650)
-    query = gr.Textbox(show_label=False, placeholder="请输入提问内容，按回车进行提交")
-    
-    query.submit(get_remote_stream,[query, chatbot], [chatbot, query])
-
-demo.queue(concurrency_count=10)
-demo.launch()
+    this_answer = random.choice(PRESET_ANSWERS)
+    for i in range(0, len(this_answer), 3):
+        time.sleep(0.5)
+        print(uuid.uuid1())
+        chatbot[-1][-1] = this_answer[:i] + "▌"
+        yield "", chatbot
+    chatbot[-1][-1] = this_answer
+    yield "", chatbot
+
+
+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"):
+            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.Row():
+                with gr.Column():
+                    ft_task_name = gr.Textbox(value="", label="任务名称")
+                    ft_task_type = gr.Dropdown(["cat", "dog", "bird"], label="任务类型", info="要创建任务的类型")
+                with gr.Column():
+                    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"):
+            gr.Markdown(METRICS_TAB_TEXT, elem_classes="markdown-text")
+
+demo.queue(max_size=20).launch(server_port=8080)