From b715b36a5f9936282eec0b38be3ee0a7878dbdc0 Mon Sep 17 00:00:00 2001 From: carry Date: Tue, 15 Apr 2025 17:13:52 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=9B=B4=E6=96=B0=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E9=9B=86=E7=94=9F=E6=88=90=E9=A1=B5=E9=9D=A2=E5=B9=B6?= =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=8B=AC=E7=AB=8B=E8=BF=90=E8=A1=8C=E5=8A=9F?= =?UTF-8?q?=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 重构导入路径,使用绝对路径替换相对路径 - 新增文档和模板选择的事件处理函数 - 添加独立运行数据集生成页面的功能 - 优化代码结构,提高可读性和可维护性 --- frontend/dataset_generate_page.py | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index afaedb2..f106a64 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -1,6 +1,9 @@ import gradio as gr -from tools import scan_docs_directory -from global_var import get_docs, scan_docs_directory, get_prompt_store +import sys +from pathlib import Path + +sys.path.append(str(Path(__file__).resolve().parent.parent)) +from global_var import get_docs, get_prompt_store def dataset_generate_page(): with gr.Blocks() as demo: @@ -39,4 +42,22 @@ def dataset_generate_page(): doc_dropdown.change(lambda x: x, inputs=doc_dropdown, outputs=doc_state) prompt_dropdown.change(lambda x: x, inputs=prompt_dropdown, outputs=prompt_state) - return demo \ No newline at end of file + # 新增事件绑定 + def on_doc_change(selected_doc): + print(f"文档选择已更改为: {selected_doc}") + return selected_doc + + def on_prompt_change(selected_prompt): + print(f"模板选择已更改为: {selected_prompt}") + return selected_prompt + + doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state) + prompt_dropdown.change(on_prompt_change, inputs=prompt_dropdown, outputs=prompt_state) + + return demo + +if __name__ == "__main__": + from global_var import init_global_var + init_global_var("workdir") + demo = dataset_generate_page() + demo.launch() \ No newline at end of file