feat(frontend): 添加数据集页面并重构主页面布局

- 新增 dataset_page 模块,实现数据集页面的基本布局
- 重构 main.py 中的页面加载方式,使用列表收集所有页面
- 更新主页面布局,将聊天页面作为第一个选项卡
- 调整设置页面的加载方式,直接使用函数调用
This commit is contained in:
carry 2025-04-06 22:49:37 +08:00
parent 2ff077bb1c
commit 841e14a093
3 changed files with 23 additions and 9 deletions

View File

@ -1,3 +1,4 @@
from .chat_page import chat_page
from .setting_page import setting_page
from .train_page import train_page
from .dataset_page import dataset_page

9
frontend/dataset_page.py Normal file
View File

@ -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

20
main.py
View File

@ -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()