46b4453ccd
- 移除各前端页面中重复的数据库引擎初始化代码 - 在 global_var.py 中统一初始化和存储数据库引擎 - 更新 setting_page.py 和 main.py 中的数据库连接逻辑 - 优化代码结构，提高可维护性和可扩展性
22 lines
693 B
Python
22 lines
693 B
Python
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 global_var import sql_engine
|
|
|
|
if __name__ == "__main__":
|
|
initialize_db(sql_engine)
|
|
|
|
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()
|
|
with gr.TabItem("设置"):
|
|
setting_page()
|
|
|
|
app.launch() |