|
|
import gradio as gr |
|
|
import modelscope_studio.components.antd as antd |
|
|
import modelscope_studio.components.antdx as antdx |
|
|
import modelscope_studio.components.base as ms |
|
|
|
|
|
from config import DEFAULT_SETTINGS, MODEL_OPTIONS, MAX_THINKING_BUDGET, MIN_THINKING_BUDGET, get_text |
|
|
|
|
|
|
|
|
def SettingsHeader(): |
|
|
state = gr.State({"open": True}) |
|
|
with antdx.Sender.Header(title=get_text("Settings", "设置"), |
|
|
open=True) as settings_header: |
|
|
with antd.Form(value=DEFAULT_SETTINGS) as settings_form: |
|
|
with antd.Form.Item(form_name="model", |
|
|
label=get_text("Chat Model", "对话模型")): |
|
|
with antd.Select(options=MODEL_OPTIONS): |
|
|
with ms.Slot("labelRender", |
|
|
params_mapping="""(option) => ({ |
|
|
label: option.label, |
|
|
link: { href: window.MODEL_OPTIONS_MAP[option.value].link }, |
|
|
})"""): |
|
|
antd.Typography.Text(as_item="label") |
|
|
antd.Typography.Link(get_text("Model Link", "模型链接"), |
|
|
href_target="_blank", |
|
|
as_item="link") |
|
|
|
|
|
with antd.Form.Item(form_name="thinking_budget", |
|
|
label=get_text("Thinking Budget", "思考预算"), |
|
|
elem_classes="setting-form-thinking-budget"): |
|
|
antd.Slider(elem_style=dict(flex=1, marginRight=14), |
|
|
min=MIN_THINKING_BUDGET, |
|
|
max=MAX_THINKING_BUDGET, |
|
|
tooltip=dict(formatter="(v) => `${v}k`")) |
|
|
antd.InputNumber(max=MAX_THINKING_BUDGET, |
|
|
min=MIN_THINKING_BUDGET, |
|
|
elem_style=dict(width=100), |
|
|
addon_after="k") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def close_header(state_value): |
|
|
state_value["open"] = False |
|
|
return gr.update(value=state_value) |
|
|
|
|
|
state.change(fn=lambda state_value: gr.update(open=state_value["open"]), |
|
|
inputs=[state], |
|
|
outputs=[settings_header]) |
|
|
|
|
|
settings_header.open_change(fn=close_header, |
|
|
inputs=[state], |
|
|
outputs=[state]) |
|
|
|
|
|
return state, settings_form |
|
|
|
