Compare commits

..

2 Commits

Author SHA1 Message Date
carry
841e14a093 feat(frontend): 添加数据集页面并重构主页面布局
- 新增 dataset_page 模块,实现数据集页面的基本布局
- 重构 main.py 中的页面加载方式,使用列表收集所有页面
- 更新主页面布局,将聊天页面作为第一个选项卡
- 调整设置页面的加载方式,直接使用函数调用
2025-04-06 22:49:37 +08:00
carry
2ff077bb1c refactor(frontend): 重构前端页面导入方式
- 在 main.py 中使用更简洁的导入方式
- 新增 __init__.py 文件以简化前端页面的导入
2025-04-06 22:46:31 +08:00
3 changed files with 25 additions and 9 deletions

4
frontend/__init__.py Normal file
View File

@ -0,0 +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

21
main.py
View File

@ -1,24 +1,27 @@
import gradio as gr import gradio as gr
from frontend.setting_page import setting_page from frontend.setting_page import setting_page
from frontend.chat_page import chat_page from frontend import chat_page,setting_page,train_page,dataset_page
from frontend.train_page import train_page
from db import initialize_db as init_db,get_engine from db import initialize_db as init_db,get_engine
if __name__ == "__main__": if __name__ == "__main__":
init_db(get_engine("workdir")) init_db(get_engine("workdir"))
setting_demo = setting_page() pages = []
chat_demo = chat_page() pages.append(setting_page())
train_demo = train_page() pages.append(chat_page())
pages.append(train_page())
pages.append(dataset_page())
with gr.Blocks() as app: with gr.Blocks() as app:
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架") gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
with gr.Tabs(): with gr.Tabs():
with gr.TabItem("微调"):
train_demo.render()
with gr.TabItem("聊天"): with gr.TabItem("聊天"):
chat_demo.render() chat_page()
with gr.TabItem("微调"):
train_page()
with gr.TabItem("数据集"):
dataset_page()
with gr.TabItem("设置"): with gr.TabItem("设置"):
setting_demo.render() setting_page()
app.launch() app.launch()