frontend/__init__.py

@@ -1,6 +1,4 @@
-from .chat_page import *
-from .setting_page import *
-from .train_page import *
-from .dataset_manage_page import *
-from .dataset_generate_page import *
-from .prompt_manage_page import *
+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_generate_page.py

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

frontend/dataset_manage_page.py

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

frontend/dataset_page.py

@@ -1,8 +1,8 @@
 import gradio as gr
 
-def prompt_manage_page():
+def dataset_page():
     with gr.Blocks() as demo:
-        gr.Markdown("## 提示词管理")
+        gr.Markdown("## 数据集")
         with gr.Row():
             with gr.Column():
                 pass

frontend/setting_page.py

@@ -34,24 +34,10 @@ def setting_page():
         except Exception as e:
             raise gr.Error(f"添加失败: {str(e)}")
 
-    def edit_provider():
+    def edit_provider(model_id, base_url, api_key):
         global selected_row
-        if not selected_row:
-            raise gr.Error("请先选择要编辑的行")
-        try:
-            with Session(sql_engine) as session:
-                provider = session.get(APIProvider, selected_row[0])
-                if not provider:
-                    raise gr.Error("找不到选中的记录")
-                provider.model_id = selected_row[1] if selected_row[1] else None
-                provider.base_url = selected_row[2] if selected_row[2] else None
-                provider.api_key = selected_row[3] if selected_row[3] else None
-                session.add(provider)
-                session.commit()
-                session.refresh(provider)
-                return get_providers()
-        except Exception as e:
-            raise gr.Error(f"编辑失败: {str(e)}")
+        print(f"edit_provider Selected row: {selected_row}")
+        return get_providers()
 
     def delete_provider():
         global selected_row

main.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from frontend.setting_page import setting_page
-from frontend import *
+from frontend import chat_page,setting_page,train_page,dataset_page
 from db import initialize_sqlite_db 
 from global_var import sql_engine
 
@@ -10,16 +10,12 @@ if __name__ == "__main__":
     with gr.Blocks() as app:
         gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
         with gr.Tabs():
-            with gr.TabItem("模型推理"):
+            with gr.TabItem("聊天"):
                 chat_page()
-            with gr.TabItem("模型微调"):
+            with gr.TabItem("微调"):
                 train_page()
-            with gr.TabItem("数据集生成"):
-                dataset_generate_page()
-            with gr.TabItem("数据集管理"):
-                dataset_manage_page()
-            with gr.TabItem("提示词模板管理"):
-                prompt_manage_page()
+            with gr.TabItem("数据集"):
+                dataset_page()
             with gr.TabItem("设置"):
                 setting_page()