feat(frontend): 更新数据集生成页面并添加独立运行功能

- 重构导入路径,使用绝对路径替换相对路径
- 新增文档和模板选择的事件处理函数
- 添加独立运行数据集生成页面的功能
- 优化代码结构,提高可读性和可维护性
This commit is contained in:
carry 2025-04-15 17:13:52 +08:00
parent 8023233bb2
commit b715b36a5f

View File

@ -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
# 新增事件绑定
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()