Compare commits
2 Commits
b52ca9b1af
...
dc28c25c65
Author | SHA1 | Date | |
---|---|---|---|
![]() |
dc28c25c65 | ||
![]() |
70b64dc3d3 |
@ -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']
|
@ -37,7 +37,7 @@ def get_engine(workdir: str) -> Engine:
|
|||||||
_engine = create_engine(db_url)
|
_engine = create_engine(db_url)
|
||||||
return _engine
|
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)
|
engine = get_engine(workdir)
|
||||||
# 初始化数据库
|
# 初始化数据库
|
||||||
initialize_db(engine)
|
initialize_sqlite_db(engine)
|
@ -41,7 +41,7 @@ def setting_page():
|
|||||||
model_id_input = gr.Textbox(label="Model ID")
|
model_id_input = gr.Textbox(label="Model ID")
|
||||||
base_url_input = gr.Textbox(label="Base URL")
|
base_url_input = gr.Textbox(label="Base URL")
|
||||||
api_key_input = gr.Textbox(label="API Key")
|
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):
|
with gr.Column(scale=3):
|
||||||
provider_table = gr.DataFrame(
|
provider_table = gr.DataFrame(
|
||||||
@ -54,9 +54,9 @@ def setting_page():
|
|||||||
)
|
)
|
||||||
|
|
||||||
with gr.Row():
|
with gr.Row():
|
||||||
edit_button = gr.Button("编辑选中行")
|
|
||||||
delete_button = gr.Button("删除选中行")
|
|
||||||
refresh_button = gr.Button("刷新数据", variant="secondary")
|
refresh_button = gr.Button("刷新数据", variant="secondary")
|
||||||
|
edit_button = gr.Button("编辑选中行", variant="primary")
|
||||||
|
delete_button = gr.Button("删除选中行", variant="stop")
|
||||||
|
|
||||||
# 绑定刷新按钮事件
|
# 绑定刷新按钮事件
|
||||||
refresh_button.click(
|
refresh_button.click(
|
||||||
|
4
main.py
4
main.py
@ -1,11 +1,11 @@
|
|||||||
import gradio as gr
|
import gradio as gr
|
||||||
from frontend.setting_page import setting_page
|
from frontend.setting_page import setting_page
|
||||||
from frontend import chat_page,setting_page,train_page,dataset_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
|
from global_var import sql_engine
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
initialize_db(sql_engine)
|
initialize_sqlite_db(sql_engine)
|
||||||
|
|
||||||
with gr.Blocks() as app:
|
with gr.Blocks() as app:
|
||||||
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
|
||||||
|
Loading…
x
Reference in New Issue
Block a user