db/__init__.py

@@ -1,3 +1,3 @@
-from .init_db import get_engine, initialize_sqlite_db
+from .init_db import get_engine, initialize_db
 
-__all__ = ['get_engine', 'initialize_sqlite_db']
+__all__ = ['get_engine', 'initialize_db']

db/init_db.py

@@ -37,7 +37,7 @@ def get_engine(workdir: str) -> Engine:
         _engine = create_engine(db_url)
     return _engine
 
-def initialize_sqlite_db(engine: Engine) -> None:
+def initialize_db(engine: Engine) -> None:
     """
     初始化数据库，创建所有表结构，并插入初始数据（如果不存在）。
 
@@ -76,4 +76,4 @@ if __name__ == "__main__":
     # 获取数据库引擎
     engine = get_engine(workdir)
     # 初始化数据库
-    initialize_sqlite_db(engine)
+    initialize_db(engine)

frontend/setting_page.py

@@ -41,7 +41,7 @@ def setting_page():
                 model_id_input = gr.Textbox(label="Model ID")
                 base_url_input = gr.Textbox(label="Base URL")
                 api_key_input = gr.Textbox(label="API Key")
-                add_button = gr.Button("添加新API", variant="primary")
+                add_button = gr.Button("添加新API")
 
             with gr.Column(scale=3):
                 provider_table = gr.DataFrame(
@@ -54,9 +54,9 @@ def setting_page():
                 )
 
                 with gr.Row():
+                    edit_button = gr.Button("编辑选中行")
+                    delete_button = gr.Button("删除选中行")
                     refresh_button = gr.Button("刷新数据", variant="secondary")
-                    edit_button = gr.Button("编辑选中行", variant="primary")
-                    delete_button = gr.Button("删除选中行", variant="stop")
 
         # 绑定刷新按钮事件
         refresh_button.click(

main.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from frontend.setting_page import setting_page
 from frontend import chat_page,setting_page,train_page,dataset_page
-from db import initialize_sqlite_db 
+from db import initialize_db 
 from global_var import sql_engine
 
 if __name__ == "__main__":
-    initialize_sqlite_db(sql_engine)
+    initialize_db(sql_engine)
 
     with gr.Blocks() as app:
         gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")