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}")