import gradio as gr from typing import List from sqlmodel import Session, select from db import get_engine from schema import APIProvider import os from global_var import sql_engine def setting_page(): def get_providers() -> List[List[str]]: try: # 添加异常处理 with Session(sql_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(sql_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)}") selected_row = None # 保存当前选中行的全局变量 def select_record(evt: gr.SelectData): global selected_row selected_row = evt.row_value print(f"Selected row: {selected_row}") 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", variant="primary") 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(): refresh_button = gr.Button("刷新数据", variant="secondary") edit_button = gr.Button("编辑选中行", variant="primary") delete_button = gr.Button("删除选中行", variant="stop") 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] ) provider_table.select(select_record, [], [], show_progress="hidden") return demo