update

.gitignore

@@ -2,6 +2,7 @@ local
 local-new
 data
 examples
+figs
 README_CN.md
 **/__pycache__
 *.pyc

src/pipeline.py

@@ -75,7 +75,7 @@ class Pipeline:
 
         if isgui and mode == "customized":
             process_method = three_agents
-            # print("Customized Agents: ", three_agents)
+            print("Customized 3-Agents: ", three_agents)
 
         sorted_process_method = self.__init_method(data, process_method)
         print("Process Method: ", sorted_process_method)

src/webui.py

@@ -61,7 +61,7 @@ examples = [
 
 
 def create_interface():
-    with gr.Blocks(title="OneKE Demo") as demo:
+    with gr.Blocks(title="OneKE Demo", theme=gr.themes.Soft(text_size="lg")) as demo:
         gr.HTML("""
             <div style="text-align:center;">
                 <p align="center">
@@ -83,15 +83,33 @@ def create_interface():
 
         with gr.Row():
             with gr.Column():
-                model_gr = gr.Dropdown(choices=["deepseek-chat", "deepseek-reasoner", "gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"], label="🪄 Select your Model")
-                api_key_gr = gr.Textbox(label="🔑 Enter your API-Key")
-                base_url_gr = gr.Textbox(label="🔗 Enter your Base-URL", value="DEFAULT")
+                model_gr = gr.Dropdown(
+                    label="🪄 Select your Model",
+                    choices=["deepseek-chat", "deepseek-reasoner", "gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"],
+                    value="deepseek-chat",
+                )
+                api_key_gr = gr.Textbox(
+                    label="🔑 Enter your API-Key",
+                    value="sk-...",
+                )
+                base_url_gr = gr.Textbox(
+                    label="🔗 Enter your Base-URL",
+                    value="DEFAULT",
+                )
             with gr.Column():
-                task_gr = gr.Dropdown(choices=["Base", "NER", "RE", "EE"], label="🎯 Select your Task")
-                mode_gr = gr.Dropdown(choices=["quick", "standard", "customized"], label="🧭 Select your Mode")
-                schema_agent_gr = gr.Dropdown(choices=["NOT REQUIRED", "get_default_schema", "get_retrieved_schema", "get_deduced_schema"], label="🤖 Select your Schema-Agent", visible=False)
-                extraction_Agent_gr = gr.Dropdown(choices=["NOT REQUIRED", "extract_information_direct", "extract_information_with_case"], label="🤖 Select your Extraction-Agent", visible=False)
-                reflection_agent_gr = gr.Dropdown(choices=["NOT REQUIRED", "reflect_with_case"], label="🤖 Select your Reflection-Agent", visible=False)
+                task_gr = gr.Dropdown(
+                    label="🎯 Select your Task",
+                    choices=["Base", "NER", "RE", "EE"],
+                    value="Base",
+                )
+                mode_gr = gr.Dropdown(
+                    label="🧭 Select your Mode",
+                    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)
+                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)
 
         with gr.Row():
             use_file_gr = gr.Checkbox(label="📂 Use File", value=True)
@@ -105,9 +123,11 @@ def create_interface():
         def customized_mode(mode):
             if mode == "customized":
                 return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+            else:
+                return gr.update(visible=False, value="NOT REQUIRED"), gr.update(visible=False, value="NOT REQUIRED"), gr.update(visible=False, value="NOT REQUIRED")
 
         def update_fields(task):
-            if task == "Base":
+            if task == "Base" or task == "":
                 return gr.update(visible=True, label="🕹️ Instruction", placeholder="Enter your Instruction"), gr.update(visible=False)
             elif task == "NER":
                 return gr.update(visible=False), gr.update(visible=True, label="🕹️ Constraint", placeholder="Enter your NER Constraint")
@@ -141,7 +161,7 @@ def create_interface():
 
         def submit(model, api_key, base_url, task, mode, instruction, constraint, text, use_file, file_path, update_case, schema_agent, extraction_Agent, reflection_agent):
             try:
-                if base_url == "DEFAULT":
+                if base_url == "DEFAULT" or base_url == "":
                     if model in ["gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"]:
                         pipeline = Pipeline(ChatGPT(model_name_or_path=model, api_key=api_key))
                     elif model in ["deepseek-chat", "deepseek-reasoner"]:
@@ -201,9 +221,9 @@ def create_interface():
 
         def clear_all():
             return (
-                gr.update(value="", visible=False),  # sechema_agent
-                gr.update(value="", visible=False),  # extraction_Agent
-                gr.update(value="", visible=False),  # reflection_agent
+                gr.update(value="NOT REQUIRED", visible=False),  # sechema_agent
+                gr.update(value="NOT REQUIRED", visible=False),  # extraction_Agent
+                gr.update(value="NOT REQUIRED", visible=False),  # reflection_agent
                 gr.update(value=""),  # task
                 gr.update(value=""),  # mode
                 gr.update(value="", visible=False),  # instruction
@@ -298,6 +318,6 @@ def create_interface():
 
     return demo
 
-
-interface = create_interface()
-interface.launch()
+if __name__ == "__main__":
+    interface = create_interface()
+    interface.launch()