diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py index ad344fe..afaedb2 100644 --- a/frontend/dataset_generate_page.py +++ b/frontend/dataset_generate_page.py @@ -23,11 +23,11 @@ def dataset_generate_page(): with gr.Column(): # 获取模板列表并设置初始值 prompts = get_prompt_store().all() - prompt_choices = [f"{p['id']} {p['name']}" for p in prompts] - initial_prompt = prompt_choices[0] if prompt_choices else None + prompt_list = [f"{p['id']} {p['name']}" for p in prompts] + initial_prompt = prompt_list[0] if prompt_list else None prompt_dropdown = gr.Dropdown( - choices=prompt_choices, + choices=prompt_list, value=initial_prompt, # 设置初始选中项 label="选择模板", allow_custom_value=True,