From 86bcf90c66f99cd47afa62cb691f736e8ac6854a Mon Sep 17 00:00:00 2001 From: carry Date: Fri, 18 Apr 2025 15:47:37 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=B7=BB=E5=8A=A0=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E9=9B=86=E7=94=9F=E6=88=90=E8=BD=AE=E6=AC=A1=E6=8E=A7?= =?UTF-8?q?=E5=88=B6=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在数据集生成页面添加"生成轮次"输入框，支持设置生成轮数 - 更新生成逻辑，根据设置的轮次进行多次生成 --- frontend/dataset_generate_page.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index 770ba2e..88f78a8 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -57,9 +57,18 @@ def dataset_generate_page(): ) api_choice = gr.State(value=initial_api) - output_text = gr.Textbox(label="生成结果", interactive=False) + rounds_input = gr.Number( + value=1, + label="生成轮次", + minimum=1, + maximum=100, + step=1, + interactive=True + ) generate_button = gr.Button("生成数据集",variant="primary") + output_text = gr.Textbox(label="生成结果", interactive=False) + with gr.Column(scale=2): variables_dataframe = gr.Dataframe( headers=["变量名", "变量值"], @@ -88,7 +97,7 @@ def dataset_generate_page(): dataframe_value = [] if input_variables is None else input_variables return selected_prompt, dataframe_value - def on_generate_click(doc_state, prompt_state, api_state, variables_dataframe, progress=gr.Progress()): + def on_generate_click(doc_state, prompt_state, api_state, variables_dataframe, rounds, progress=gr.Progress()): variables_dict = {} # 正确遍历DataFrame的行数据 for _, row in variables_dataframe.iterrows(): @@ -98,7 +107,7 @@ def dataset_generate_page(): variables_dict[var_name] = var_value import time - total_steps = 10 + total_steps = rounds for i in range(total_steps): # 模拟每个步骤的工作负载 time.sleep(0.5) @@ -117,7 +126,7 @@ def dataset_generate_page(): generate_button.click( on_generate_click, - inputs=[doc_choice, prompt_choice, api_choice, variables_dataframe], + inputs=[doc_choice, prompt_choice, api_choice, variables_dataframe, rounds_input], outputs=output_text )