import gradio as gr
from typing import List
from sqlmodel import Session, select
from db import get_engine
from schema import APIProvider
import os

engine = get_engine(os.path.join(os.path.dirname(__file__), "..", "workdir"))

def setting_page():
    def get_providers() -> List[List[str]]:
        try:  # 添加异常处理
            with Session(engine) as session:
                providers = session.exec(select(APIProvider)).all()
                return [
                    [p.id, p.model_id, p.base_url, p.api_key or ""]
                    for p in providers
                ]
        except Exception as e:
            raise gr.Error(f"获取数据失败: {str(e)}")

    def add_provider(model_id, base_url, api_key):
        try:
            with Session(engine) as session:
                new_provider = APIProvider(
                    model_id=model_id if model_id else None,
                    base_url=base_url if base_url else None,
                    api_key=api_key if api_key else None
                )
                session.add(new_provider)
                session.commit()
                session.refresh(new_provider)
                return get_providers()
        except Exception as e:
            raise gr.Error(f"添加失败: {str(e)}")

    with gr.Blocks() as demo:
        gr.Markdown("## API Provider 管理")

        with gr.Row():
            with gr.Column(scale=1):
                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")

            with gr.Column(scale=3):
                provider_table = gr.DataFrame(
                    headers=["id", "model id", "base URL", "API Key"],
                    datatype=["number", "str", "str", "str"],
                    interactive=True,
                    value=get_providers(),
                    wrap=True,
                    col_count=(4, "auto")
                )

                with gr.Row():
                    edit_button = gr.Button("编辑选中行")
                    delete_button = gr.Button("删除选中行")
                    refresh_button = gr.Button("刷新数据", variant="secondary")

        # 绑定刷新按钮事件
        refresh_button.click(
            fn=get_providers,
            outputs=[provider_table],
            queue=False  # 立即刷新不需要排队
        )

        add_button.click(
            fn=add_provider,
            inputs=[model_id_input, base_url_input, api_key_input],
            outputs=[provider_table]
        )

        return demo