refactor(frontend): 调整数据集生成页面布局并优化代码结构
- 使用 gr.Column(scale=1) 和 gr.Column(scale=2) 调整列宽比例 - 移除多余的空行和缩进,提高代码可读性 - 优化变量声明和组件创建的顺序,使页面结构更清晰
This commit is contained in:
@@ -12,7 +12,7 @@ def dataset_generate_page():
|
||||
with gr.Blocks() as demo:
|
||||
gr.Markdown("## 数据集生成")
|
||||
with gr.Row():
|
||||
with gr.Column():
|
||||
with gr.Column(scale=1):
|
||||
docs_list = [str(doc.name) for doc in get_docs()]
|
||||
initial_doc = docs_list[0] if docs_list else None
|
||||
doc_dropdown = gr.Dropdown(
|
||||
@@ -22,8 +22,6 @@ def dataset_generate_page():
|
||||
interactive=True
|
||||
)
|
||||
doc_choice = gr.State(value=initial_doc)
|
||||
|
||||
with gr.Column():
|
||||
prompts = get_prompt_store().all()
|
||||
prompt_list = [f"{p['id']} {p['name']}" for p in prompts]
|
||||
initial_prompt = prompt_list[0] if prompt_list else None
|
||||
@@ -45,8 +43,6 @@ def dataset_generate_page():
|
||||
interactive=True
|
||||
)
|
||||
prompt_choice = gr.State(value=initial_prompt)
|
||||
|
||||
with gr.Column():
|
||||
# 从数据库获取API Provider列表
|
||||
with Session(get_sql_engine()) as session:
|
||||
providers = session.exec(select(APIProvider)).all()
|
||||
@@ -61,17 +57,17 @@ def dataset_generate_page():
|
||||
)
|
||||
api_choice = gr.State(value=initial_api)
|
||||
|
||||
generate_button = gr.Button("生成数据集",variant="primary")
|
||||
output_text = gr.Textbox(label="生成结果", interactive=False)
|
||||
generate_button = gr.Button("生成数据集",variant="primary")
|
||||
|
||||
output_text = gr.Textbox(label="生成结果", interactive=False)
|
||||
|
||||
variables_dataframe = gr.Dataframe(
|
||||
headers=["变量名", "变量值"],
|
||||
datatype=["str", "str"],
|
||||
interactive=True,
|
||||
label="变量列表",
|
||||
value=initial_dataframe_value # 设置初始化数据
|
||||
)
|
||||
with gr.Column(scale=2):
|
||||
variables_dataframe = gr.Dataframe(
|
||||
headers=["变量名", "变量值"],
|
||||
datatype=["str", "str"],
|
||||
interactive=True,
|
||||
label="变量列表",
|
||||
value=initial_dataframe_value # 设置初始化数据
|
||||
)
|
||||
|
||||
|
||||
def on_doc_change(selected_doc):
|
||||
@@ -92,7 +88,6 @@ 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()):
|
||||
variables_dict = {}
|
||||
# 正确遍历DataFrame的行数据
|
||||
@@ -116,7 +111,6 @@ def dataset_generate_page():
|
||||
|
||||
return "all done"
|
||||
|
||||
|
||||
doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_choice)
|
||||
prompt_dropdown.change(on_prompt_change, inputs=prompt_dropdown, outputs=[prompt_choice, variables_dataframe])
|
||||
api_dropdown.change(on_api_change, inputs=api_dropdown, outputs=api_choice)
|
||||
|
Reference in New Issue
Block a user