frontend/__init__.py

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

frontend/dataset_page.py

@@ -1,9 +0,0 @@
-import gradio as gr
-
-def dataset_page():
-    with gr.Blocks() as demo:
-        gr.Markdown("## 数据集")
-        with gr.Row():
-            with gr.Column():
-                pass
-    return demo

main.py

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