From 314434951d2e7652724fcbbbe72ad587dae35921 Mon Sep 17 00:00:00 2001 From: carry Date: Sat, 19 Apr 2025 14:47:01 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E5=AE=9E=E7=8E=B0=E4=BA=86?= =?UTF-8?q?=E6=96=87=E6=A1=A3=E3=80=81=E6=8F=90=E7=A4=BA=E5=92=8C=20API=20?= =?UTF-8?q?=E6=8F=90=E4=BE=9B=E5=95=86=E7=9A=84=E8=8E=B7=E5=8F=96=E9=80=BB?= =?UTF-8?q?=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- frontend/dataset_generate_page.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index 88f78a8..b281042 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -98,6 +98,11 @@ def dataset_generate_page(): return selected_prompt, dataframe_value def on_generate_click(doc_state, prompt_state, api_state, variables_dataframe, rounds, progress=gr.Progress()): + doc = [i for i in get_docs() if i.name == doc_state][0] + prompt = [i for i in get_prompt_store().all() if i["id"] == int(prompt_state.split(" ")[0])][0] + with Session(get_sql_engine()) as session: + api_provider = session.exec(select(APIProvider).where(APIProvider.id == int(api_state.split(" ")[0]))).first() + variables_dict = {} # 正确遍历DataFrame的行数据 for _, row in variables_dataframe.iterrows(): @@ -112,9 +117,6 @@ def dataset_generate_page(): # 模拟每个步骤的工作负载 time.sleep(0.5) - # 更新进度条 - # 第一个参数是当前的进度比例 (0.0 到 1.0) - # desc 参数可以动态更新进度条旁边的描述文字 current_progress = (i + 1) / total_steps progress(current_progress, desc=f"处理步骤 {i + 1}/{total_steps}")