Compare commits

...

2 Commits

Author SHA1 Message Date
carry
dc28c25c65 feat(frontend): 更新设置页面按钮样式
- 为"添加新API"按钮添加 primary 样式
- 为"编辑选中行"按钮添加 primary 样式
- 为"删除选中行"按钮添加 stop 样式
- 保持"刷新数据"按钮的 secondary 样式
2025-04-08 14:23:31 +08:00
carry
70b64dc3d3 refactor(db): 重命名数据库初始化函数以明确其适用范围
- 将 initialize_db 函数重命名为 initialize_sqlite_db,以明确该函数专用于 SQLite 数据库
- 更新相关模块和文件中的引用,以确保代码一致性
- 此修改旨在提高代码的可读性和维护性,特别是未来可能接入多种数据库时
2025-04-08 14:16:12 +08:00
4 changed files with 9 additions and 9 deletions

View File

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

View File

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

View File

@ -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")
add_button = gr.Button("添加新API", variant="primary")
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(

View File

@ -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_db
from db import initialize_sqlite_db
from global_var import sql_engine
if __name__ == "__main__":
initialize_db(sql_engine)
initialize_sqlite_db(sql_engine)
with gr.Blocks() as app:
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")