diff --git a/frontend/__init__.py b/frontend/__init__.py index 7313f97..fe412b5 100644 --- a/frontend/__init__.py +++ b/frontend/__init__.py @@ -1,3 +1,4 @@ from .chat_page import chat_page from .setting_page import setting_page -from .train_page import train_page \ No newline at end of file +from .train_page import train_page +from .dataset_page import dataset_page \ No newline at end of file diff --git a/frontend/dataset_page.py b/frontend/dataset_page.py new file mode 100644 index 0000000..8d721be --- /dev/null +++ b/frontend/dataset_page.py @@ -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 \ No newline at end of file diff --git a/main.py b/main.py index d0eabe0..8f8dc88 100644 --- a/main.py +++ b/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() \ No newline at end of file