feat(frontend): 添加数据集页面并重构主页面布局
- 新增 dataset_page 模块,实现数据集页面的基本布局 - 重构 main.py 中的页面加载方式,使用列表收集所有页面 - 更新主页面布局,将聊天页面作为第一个选项卡 - 调整设置页面的加载方式,直接使用函数调用
This commit is contained in:
parent
2ff077bb1c
commit
841e14a093
@ -1,3 +1,4 @@
|
||||
from .chat_page import chat_page
|
||||
from .setting_page import setting_page
|
||||
from .train_page import train_page
|
||||
from .train_page import train_page
|
||||
from .dataset_page import dataset_page
|
9
frontend/dataset_page.py
Normal file
9
frontend/dataset_page.py
Normal 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
20
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()
|
Loading…
x
Reference in New Issue
Block a user