Update src/webui.py

src/webui.py

@@ -13,17 +13,6 @@ from pipeline import Pipeline
 
 
 examples = [
-    {
-        "task": "Base",
-        "mode": "quick",
-        "use_file": False,
-        "text": "合力治堵!济南交通部门在拥堵路段定点研究交通治理方案",
-        "instruction": "请帮我抽取这个新闻事件",
-        "constraint": "",
-        "file_path": None,
-        "update_case": False,
-        "truth": "",
-    },
     {
         "task": "NER",
         "mode": "quick",
@@ -100,28 +89,6 @@ examples = [
         "file_path": None,
         "update_case": False,
         "truth": "",
-    },
-    {
-        "task": "Base",
-        "mode": "quick",
-        "use_file": False,
-        "text": "张三，男，60岁，主诉背部酸痛已持续约两周，伴有轻微的头晕。患者有高血压病史，已服用降压药物多年，且控制良好；此外，患者曾在五年前接受过一次胆囊切除手术。体检时，心率为75次/分钟，血压为130/85 mmHg。背部触诊时无明显压痛，但活动时出现轻微不适。胸部X光显示无异常，腰部CT检查提示轻度腰椎退行性变。经医生诊断，患者被认为是由于长时间的不良姿势引起的腰椎退行性病变，建议进行物理治疗，并配合止痛药物。",
-        "instruction": "请从这个病例描述中，提取出重要的医疗信息",
-        "constraint": "",
-        "file_path": None,
-        "update_case": False,
-        "truth": "",
-    },
-    {
-        "task": "Base",
-        "mode": "quick",
-        "use_file": False,
-        "text": "中国政府近日宣布了一项新的环保政策，旨在减少工业污染，并改善空气质量。此次政策将在全国范围内实施，涉及多个行业，尤其是钢铁和煤炭行业。环保部门负责人表示，这项政策的实施标志着中国环保工作的新阶段，预计将在未来五年内显著改善空气质量。",
-        "instruction": "请从这段新闻描述中提取出重要的事件信息，包括事件名称、时间、参与人员、事件目的、实施过程及预期结果。",
-        "constraint": "",
-        "file_path": None,
-        "update_case": False,
-        "truth": "",
     }
 ]
 
@@ -182,7 +149,7 @@ def create_interface():
                     choices=["quick", "standard", "customized"],
                     value="quick",
                 )
-                schema_agent_gr = gr.Dropdown(choices=["Not Required", "get_default_schema", "get_retrieved_schema", "get_deduced_schema"], value="Not Required", label="🤖 Select your Schema-Agent", visible=False)
+                schema_agent_gr = gr.Dropdown(choices=["Not Required", "get_default_schema", "get_deduced_schema"], value="Not Required", label="🤖 Select your Schema-Agent", visible=False)
                 extraction_Agent_gr = gr.Dropdown(choices=["Not Required", "extract_information_direct", "extract_information_with_case"], value="Not Required", label="🤖 Select your Extraction-Agent", visible=False)
                 reflection_agent_gr = gr.Dropdown(choices=["Not Required", "reflect_with_case"], value="Not Required", label="🤖 Select your Reflection-Agent", visible=False)
 
@@ -190,7 +157,7 @@ def create_interface():
         file_path_gr = gr.File(label="📖 Upload a File", visible=True)
         text_gr = gr.Textbox(label="📖 Text", lines=5, placeholder="Enter your Text please.", visible=False)
         instruction_gr = gr.Textbox(label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.", visible=True)
-        constraint_gr = gr.Textbox(label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.", visible=False)
+        constraint_gr = gr.Textbox(label="🕹️ Constraint", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.", visible=False)
 
         update_case_gr = gr.Checkbox(label="💰 Update Case", value=False)
         # update_schema_gr = gr.Checkbox(label="📟 Update Schema", value=False)
@@ -223,13 +190,14 @@ def create_interface():
             if task == "Base" or task == "":
                 return gr.update(visible=True, label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names."), gr.update(visible=False)
             elif task == "NER":
-                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
+                # TODO
+                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Constraint", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
             elif task == "RE":
-                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
+                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Constraint", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
             elif task == "EE":
-                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
+                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Constraint", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
             elif task == "Triple":
-                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Instruction", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
+                return gr.update(visible=False), gr.update(visible=True, label="🕹️ Constraint", lines=3, placeholder="You can enter any type of information you want to extract here, for example: Please help me extract all the person names.")
 
         def update_input_fields(use_file):
             if use_file:
@@ -249,21 +217,12 @@ def create_interface():
         #     else:
         #         return gr.update(visible=False)
 
+        idx = 0
         def start_with_example():
-            example_index = random.randint(-3, len(examples) - 1)
-            example_index = max(example_index, 0)
-            example = examples[example_index]
-
-            if example_index == 0:
-                with open("data/input_files/ChineseNewsExample.json", "r", encoding="utf-8") as file:
-                    lines = file.readlines()
-                    random_line = random.choice(lines).strip()
-                    try:
-                        json_data = json.loads(random_line)
-                        title = json_data.get("title", "No title found")
-                    except json.JSONDecodeError:
-                        title = "Error decoding JSON"
-                    example["text"] = title
+            example = examples[idx]
+            idx += 1
+            if idx >= len(examples):
+                idx = 0
 
             return (
                 gr.update(value=example["task"]),