From 841e14a0931df9b6205d74ad1de02b3df2858a98 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Sun, 6 Apr 2025 22:49:37 +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=A1=B5=E9=9D=A2=E5=B9=B6=E9=87=8D=E6=9E=84?=
 =?UTF-8?q?=E4=B8=BB=E9=A1=B5=E9=9D=A2=E5=B8=83=E5=B1=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 新增 dataset_page 模块，实现数据集页面的基本布局
- 重构 main.py 中的页面加载方式，使用列表收集所有页面
- 更新主页面布局，将聊天页面作为第一个选项卡
- 调整设置页面的加载方式，直接使用函数调用
---
 frontend/__init__.py     |  3 ++-
 frontend/dataset_page.py |  9 +++++++++
 main.py                  | 20 ++++++++++++--------
 3 files changed, 23 insertions(+), 9 deletions(-)
 create mode 100644 frontend/dataset_page.py

diff --git a/frontend/__init__.py b/frontend/__init__.py
index 7313f97..fe412b5 100644
--- a/frontend/__init__.py
+++ b/frontend/__init__.py
@@ -1,3 +1,4 @@
 from .chat_page import chat_page
 from .setting_page import setting_page
-from .train_page import train_page
\ No newline at end of file
+from .train_page import train_page
+from .dataset_page import dataset_page
\ No newline at end of file
diff --git a/frontend/dataset_page.py b/frontend/dataset_page.py
new file mode 100644
index 0000000..8d721be
--- /dev/null
+++ b/frontend/dataset_page.py
@@ -0,0 +1,9 @@
+import gradio as gr
+
+def dataset_page():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 数据集")
+        with gr.Row():
+            with gr.Column():
+                pass
+    return demo
\ No newline at end of file
diff --git a/main.py b/main.py
index d0eabe0..8f8dc88 100644
--- a/main.py
+++ b/main.py
@@ -1,23 +1,27 @@
 import gradio as gr
 from frontend.setting_page import setting_page
-from frontend import chat_page,setting_page,train_page
+from frontend import chat_page,setting_page,train_page,dataset_page
 from db import initialize_db as init_db,get_engine
 
 if __name__ == "__main__":
     init_db(get_engine("workdir"))
 
-    setting_demo = setting_page()
-    chat_demo = chat_page()
-    train_demo = train_page()
+    pages = []
+    pages.append(setting_page())
+    pages.append(chat_page())
+    pages.append(train_page())
+    pages.append(dataset_page())
 
     with gr.Blocks() as app:
         gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
         with gr.Tabs():
-            with gr.TabItem("微调"):
-                train_demo.render()
             with gr.TabItem("聊天"):
-                chat_demo.render()
+                chat_page()
+            with gr.TabItem("微调"):
+                train_page()
+            with gr.TabItem("数据集"):
+                dataset_page()
             with gr.TabItem("设置"):
-                setting_demo.render()
+                setting_page()
     
     app.launch()
\ No newline at end of file