From 0a6ae7a4ee7e818a7268400dd9b82044db2237f3 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Wed, 9 Apr 2025 08:11:40 +0800
Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E9=87=8D=E6=9E=84=E5=89=8D?=
 =?UTF-8?q?=E7=AB=AF=E9=A1=B5=E9=9D=A2=E5=B9=B6=E6=B7=BB=E5=8A=A0=E6=96=B0?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 重命名 dataset_page 为 prompt_manage_page，支持提示词模板管理
- 新增 dataset_generate_page 和 dataset_manage_page 页面
- 更新 main.py 中的页面引用和标签名称
- 修改前端初始化文件，使用 * 导入所有页面模块
---
 frontend/__init__.py                               | 10 ++++++----
 frontend/dataset_generate_page.py                  |  9 +++++++++
 frontend/dataset_manage_page.py                    |  9 +++++++++
 .../{dataset_page.py => prompt_manage_page.py}     |  4 ++--
 main.py                                            | 14 +++++++++-----
 5 files changed, 35 insertions(+), 11 deletions(-)
 create mode 100644 frontend/dataset_generate_page.py
 create mode 100644 frontend/dataset_manage_page.py
 rename frontend/{dataset_page.py => prompt_manage_page.py} (67%)

diff --git a/frontend/__init__.py b/frontend/__init__.py
index fe412b5..7480f89 100644
--- a/frontend/__init__.py
+++ b/frontend/__init__.py
@@ -1,4 +1,6 @@
-from .chat_page import chat_page
-from .setting_page import setting_page
-from .train_page import train_page
-from .dataset_page import dataset_page
\ No newline at end of file
+from .chat_page import *
+from .setting_page import *
+from .train_page import *
+from .dataset_manage_page import *
+from .dataset_generate_page import *
+from .prompt_manage_page import *
\ No newline at end of file
diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py
new file mode 100644
index 0000000..0666b02
--- /dev/null
+++ b/frontend/dataset_generate_page.py
@@ -0,0 +1,9 @@
+import gradio as gr
+
+def dataset_generate_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/frontend/dataset_manage_page.py b/frontend/dataset_manage_page.py
new file mode 100644
index 0000000..d11f31c
--- /dev/null
+++ b/frontend/dataset_manage_page.py
@@ -0,0 +1,9 @@
+import gradio as gr
+
+def dataset_manage_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/frontend/dataset_page.py b/frontend/prompt_manage_page.py
similarity index 67%
rename from frontend/dataset_page.py
rename to frontend/prompt_manage_page.py
index 8d721be..7f05327 100644
--- a/frontend/dataset_page.py
+++ b/frontend/prompt_manage_page.py
@@ -1,8 +1,8 @@
 import gradio as gr
 
-def dataset_page():
+def prompt_manage_page():
     with gr.Blocks() as demo:
-        gr.Markdown("## 数据集")
+        gr.Markdown("## 提示词管理")
         with gr.Row():
             with gr.Column():
                 pass
diff --git a/main.py b/main.py
index b220894..9a7b9fc 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,6 @@
 import gradio as gr
 from frontend.setting_page import setting_page
-from frontend import chat_page,setting_page,train_page,dataset_page
+from frontend import *
 from db import initialize_sqlite_db 
 from global_var import sql_engine
 
@@ -10,12 +10,16 @@ if __name__ == "__main__":
     with gr.Blocks() as app:
         gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
         with gr.Tabs():
-            with gr.TabItem("聊天"):
+            with gr.TabItem("模型推理"):
                 chat_page()
-            with gr.TabItem("微调"):
+            with gr.TabItem("模型微调"):
                 train_page()
-            with gr.TabItem("数据集"):
-                dataset_page()
+            with gr.TabItem("数据集生成"):
+                dataset_generate_page()
+            with gr.TabItem("数据集管理"):
+                dataset_manage_page()
+            with gr.TabItem("提示词模板管理"):
+                prompt_manage_page()
             with gr.TabItem("设置"):
                 setting_page()