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