From 611904cef9a440ec2e40e38fc42c414b72a92a15 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Fri, 11 Apr 2025 19:43:34 +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=E9=80=89=E6=8B=A9=E5=8A=9F=E8=83=BD=E5=88=B0?=
 =?UTF-8?q?=E8=AE=AD=E7=BB=83=E9=A1=B5=E9=9D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 train_page.py 中添加数据集选择下拉框
- 从全局变量中获取数据集列表并设置初始值
- 添加交互性和自定义值支持
---
 frontend/train_page.py | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/frontend/train_page.py b/frontend/train_page.py
index 44e2d82..62b5d8f 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -1,9 +1,29 @@
 import gradio as gr
+import sys
+from pathlib import Path
+sys.path.append(str(Path(__file__).resolve().parent.parent))
+from global_var import get_model, get_tokenizer, get_datasets
 
 def train_page():
     with gr.Blocks() as demo:
+
         gr.Markdown("## 微调")
-        with gr.Row():
-            with gr.Column():
-                pass
-    return demo
\ No newline at end of file
+        # 获取数据集列表并设置初始值
+        datasets_list = [str(ds["name"]) for ds in get_datasets().all()]
+        initial_dataset = datasets_list[0] if datasets_list else None
+        dataset_dropdown = gr.Dropdown(
+            choices=datasets_list,
+            value=initial_dataset,  # 设置初始选中项
+            label="选择数据集",
+            allow_custom_value=True,
+            interactive=True
+        )
+
+    return demo
+
+if __name__ == "__main__":
+    from global_var import init_global_var
+    init_global_var("workdir")
+    demo = train_page()
+    demo.queue()
+    demo.launch()
\ No newline at end of file