From 051d1a7535fd02e7c7b812ae913034a2ed86edf4 Mon Sep 17 00:00:00 2001
From: carry <2641257231@qq.com>
Date: Thu, 10 Apr 2025 17:37:45 +0800
Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=B7=BB=E5=8A=A0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E7=AE=A1=E7=90=86=E9=A1=B5=E9=9D=A2=E5=B9=B6=E5=88=9D?=
 =?UTF-8?q?=E5=A7=8B=E5=8C=96=E6=A8=A1=E5=9E=8B=E7=9B=B8=E5=85=B3=E5=85=A8?=
 =?UTF-8?q?=E5=B1=80=E5=8F=98=E9=87=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 frontend/__init__.py 中添加 model_manage_page 模块引用
- 新增 model_manage_page.py 文件，实现模型管理页面的基本框架
- 在 global_var.py 中添加 model 和 tokenizer 全局变量
- 在 main.py 中集成模型管理页面到主应用的 Tabs 组件中
---
 frontend/__init__.py          | 1 +
 frontend/model_manage_page.py | 9 +++++++++
 global_var.py                 | 5 ++++-
 main.py                       | 2 ++
 4 files changed, 16 insertions(+), 1 deletion(-)
 create mode 100644 frontend/model_manage_page.py

diff --git a/frontend/__init__.py b/frontend/__init__.py
index 7480f89..a7b2d6a 100644
--- a/frontend/__init__.py
+++ b/frontend/__init__.py
@@ -1,6 +1,7 @@
 from .chat_page import *
 from .setting_page import *
 from .train_page import *
+from .model_manage_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/model_manage_page.py b/frontend/model_manage_page.py
new file mode 100644
index 0000000..5126aad
--- /dev/null
+++ b/frontend/model_manage_page.py
@@ -0,0 +1,9 @@
+import gradio as gr
+
+def model_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/global_var.py b/global_var.py
index b35ab07..328cb13 100644
--- a/global_var.py
+++ b/global_var.py
@@ -4,4 +4,7 @@ from tools import scan_docs_directory
 prompt_store = get_prompt_tinydb("workdir")
 sql_engine = get_sqlite_engine("workdir")
 docs = scan_docs_directory("workdir")
-datasets = get_all_dataset("workdir")
\ No newline at end of file
+datasets = get_all_dataset("workdir")
+
+model = None
+tokenizer = None
\ No newline at end of file
diff --git a/main.py b/main.py
index c3437e5..3797c1e 100644
--- a/main.py
+++ b/main.py
@@ -10,6 +10,8 @@ if __name__ == "__main__":
     with gr.Blocks() as app:
         gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
         with gr.Tabs():
+            with gr.TabItem("模型管理"):
+                model_manage_page()
             with gr.TabItem("模型推理"):
                 chat_page()
             with gr.TabItem("模型微调"):