import gradio as gr
from frontend.setting_page import setting_page
from frontend import chat_page,setting_page,train_page
from db import initialize_db as init_db,get_engine

if __name__ == "__main__":
    init_db(get_engine("workdir"))

    setting_demo = setting_page()
    chat_demo = chat_page()
    train_demo = train_page()

    with gr.Blocks() as app:
        gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架")
        with gr.Tabs():
            with gr.TabItem("微调"):
                train_demo.render()
            with gr.TabItem("聊天"):
                chat_demo.render()
            with gr.TabItem("设置"):
                setting_demo.render()
    
    app.launch()