Gradio 月活用户达到 100 万啦！

阅读更多
Gradio logo
  1. 使用 Blocks 构建
  2. Blocks 和事件监听器
←

四种界面

控制布局

→

Blocks 和事件监听器

我们在快速入门中简要介绍了 Blocks 类，作为构建自定义演示的一种方式。让我们深入了解一下。

Blocks 结构

看看下面的演示。

import gradio as gr


def greet(name):
    return "Hello " + name + "!"


with gr.Blocks() as demo:
    name = gr.Textbox(label="Name")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Greet")
    greet_btn.click(fn=greet, inputs=name, outputs=output, api_name="greet")

demo.launch()

  • 首先，请注意 with gr.Blocks() as demo: 子句。Blocks 应用代码将包含在此子句中。
  • 接下来是组件 (Components)。这些组件与 Interface 中使用的组件相同。但是，组件不是传递给某个构造函数，而是在 with 子句中创建时自动添加到 Blocks。
  • 最后是 click() 事件监听器。事件监听器定义了应用内的数据流。在上面的示例中，监听器将两个文本框 (Textboxes) 连接在一起。文本框 name 作为输入，文本框 output 作为 greet 方法的输出。当按钮 greet_btn 被点击时，会触发此数据流。与 Interface 类似，事件监听器可以接受多个输入或输出。

你还可以使用装饰器附加事件监听器 - 跳过 fn 参数并直接分配 inputs 和 outputs

import gradio as gr

with gr.Blocks() as demo:
    name = gr.Textbox(label="Name")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Greet")

    @greet_btn.click(inputs=name, outputs=output)
    def greet(name):
        return "Hello " + name + "!"

demo.launch()

事件监听器和交互性

在上面的示例中，你会注意到你可以编辑文本框 name，但不能编辑文本框 output。这是因为任何作为事件监听器输入的组件都会变为交互式的。但是，由于文本框 output 仅作为输出，Gradio 确定它不应该被设置为交互式的。你可以覆盖默认行为，并使用布尔值 interactive 关键字参数直接配置组件的交互性，例如 gr.Textbox(interactive=True)。

output = gr.Textbox(label="Output", interactive=True)

注意：如果 Gradio 组件既不是输入也不是输出会发生什么？如果组件是使用默认值构建的，那么它被假定为显示内容并且被渲染为非交互式的。否则，它将被渲染为交互式的。同样，可以通过为 interactive 参数指定值来覆盖此行为。

事件监听器类型

看看下面的演示

import gradio as gr

def welcome(name):
    return f"Welcome to Gradio, {name}!"

with gr.Blocks() as demo:
    gr.Markdown(
    """
    # Hello World!
    Start typing below to see the output.
    """)
    inp = gr.Textbox(placeholder="What is your name?")
    out = gr.Textbox()
    inp.change(welcome, inp, out)

demo.launch()

welcome 函数不是由点击触发，而是由在文本框 inp 中输入内容触发。这是由于 change() 事件监听器。不同的组件支持不同的事件监听器。例如，Video 组件支持 play() 事件监听器，当用户按下播放时触发。请参阅文档，了解每个组件的事件监听器。

多数据流

Blocks 应用不限于像 Interface 那样的单个数据流。看看下面的演示

import gradio as gr

def increase(num):
    return num + 1

with gr.Blocks() as demo:
    a = gr.Number(label="a")
    b = gr.Number(label="b")
    atob = gr.Button("a > b")
    btoa = gr.Button("b > a")
    atob.click(increase, a, b)
    btoa.click(increase, b, a)

demo.launch()

请注意，num1 可以作为 num2 的输入，反之亦然！随着你的应用变得更加复杂，你将拥有连接各种组件的多个数据流。

这是一个“多步骤”演示的示例，其中一个模型（语音转文本模型）的输出被馈送到下一个模型（情感分类器）。

from transformers import pipeline

import gradio as gr

asr = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
classifier = pipeline("text-classification")

def speech_to_text(speech):
    text = asr(speech)["text"]  
    return text

def text_to_sentiment(text):
    return classifier(text)[0]["label"]  

demo = gr.Blocks()

with demo:
    audio_file = gr.Audio(type="filepath")
    text = gr.Textbox()
    label = gr.Label()

    b1 = gr.Button("Recognize Speech")
    b2 = gr.Button("Classify Sentiment")

    b1.click(speech_to_text, inputs=audio_file, outputs=text)
    b2.click(text_to_sentiment, inputs=text, outputs=label)

demo.launch()

函数输入列表 vs 字典

到目前为止，你看到的事件监听器都只有一个输入组件。如果你想让多个输入组件将数据传递给函数，你可以选择两种方式让函数接受输入组件值

  1. 作为参数列表，或者
  2. 作为单个值字典，键是组件

让我们看每种方式的示例

import gradio as gr

with gr.Blocks() as demo:
    a = gr.Number(label="a")
    b = gr.Number(label="b")
    with gr.Row():
        add_btn = gr.Button("Add")
        sub_btn = gr.Button("Subtract")
    c = gr.Number(label="sum")

    def add(num1, num2):
        return num1 + num2
    add_btn.click(add, inputs=[a, b], outputs=c)

    def sub(data):
        return data[a] - data[b]
    sub_btn.click(sub, inputs={a, b}, outputs=c)

demo.launch()

add() 和 sub() 都将 a 和 b 作为输入。但是，这些监听器之间的语法不同。

  1. 对于 add_btn 监听器，我们将输入作为列表传递。函数 add() 将这些输入中的每一个作为参数。a 的值映射到参数 num1，b 的值映射到参数 num2。
  2. 对于 sub_btn 监听器，我们将输入作为集合传递（请注意花括号！）。函数 sub() 接受单个字典参数 data，其中键是输入组件，值是这些组件的值。

选择哪种语法取决于你的偏好！对于具有许多输入组件的函数，选项 2 可能更容易管理。

函数返回列表 vs 字典

类似地，你可以将多个输出组件的值作为以下形式返回：

  1. 值列表，或
  2. 以组件为键的字典

让我们首先看一个示例 (1)，其中我们通过返回两个值来设置两个输出组件的值

with gr.Blocks() as demo:
    food_box = gr.Number(value=10, label="Food Count")
    status_box = gr.Textbox()

    def eat(food):
        if food > 0:
            return food - 1, "full"
        else:
            return 0, "hungry"

    gr.Button("Eat").click(
        fn=eat,
        inputs=food_box,
        outputs=[food_box, status_box]
    )

上面，每个 return 语句都返回两个值，分别对应于 food_box 和 status_box。

除了返回与每个输出组件按顺序对应的值列表之外，你还可以返回一个字典，其中键对应于输出组件，值是新值。这也允许你跳过更新某些输出组件。

with gr.Blocks() as demo:
    food_box = gr.Number(value=10, label="Food Count")
    status_box = gr.Textbox()

    def eat(food):
        if food > 0:
            return {food_box: food - 1, status_box: "full"}
        else:
            return {status_box: "hungry"}

    gr.Button("Eat").click(
        fn=eat,
        inputs=food_box,
        outputs=[food_box, status_box]
    )

请注意，当没有食物时，我们只更新了 status_box 元素。我们跳过了更新 food_box 组件。

当事件监听器在返回时影响许多组件，或有条件地影响某些输出而不是其他输出时，字典返回会很有用。

请记住，对于字典返回，我们仍然需要在事件监听器中指定可能的输出。

更新组件配置

事件监听器函数的返回值通常是相应输出组件的更新值。有时我们还希望更新组件的配置，例如可见性。在这种情况下，我们会返回一个新的组件，并设置我们想要更改的属性。

import gradio as gr

def change_textbox(choice):
    if choice == "short":
        return gr.Textbox(lines=2, visible=True)
    elif choice == "long":
        return gr.Textbox(lines=8, visible=True, value="Lorem ipsum dolor sit amet")
    else:
        return gr.Textbox(visible=False)

with gr.Blocks() as demo:
    radio = gr.Radio(
        ["short", "long", "none"], label="What kind of essay would you like to write?"
    )
    text = gr.Textbox(lines=2, interactive=True, show_copy_button=True)
    radio.change(fn=change_textbox, inputs=radio, outputs=text)

demo.launch()

了解一下我们如何通过新的 gr.Textbox() 方法来配置 Textbox 组件本身。value= 参数仍然可以用于更新值以及组件配置。任何我们未设置的参数都将保留其先前的值。

不更改组件的值

在某些情况下，您可能希望保持组件的值不变。Gradio 包含一个特殊函数 gr.skip()，可以从您的函数中返回。返回此函数将使输出组件（或多个组件）的值保持原样。让我们用一个例子来说明

import random
import gradio as gr

with gr.Blocks() as demo:
    with gr.Row():
        clear_button = gr.Button("Clear")
        skip_button = gr.Button("Skip")
        random_button = gr.Button("Random")
    numbers = [gr.Number(), gr.Number()]

    clear_button.click(lambda : (None, None), outputs=numbers)
    skip_button.click(lambda : [gr.skip(), gr.skip()], outputs=numbers)
    random_button.click(lambda : (random.randint(0, 100), random.randint(0, 100)), outputs=numbers)

demo.launch()

请注意返回 None (通常会将组件的值重置为空状态) 与返回 gr.skip() (使组件值保持不变) 之间的区别。

提示： 如果您有多个输出组件，并且希望保持所有组件的值不变，您可以只返回一个 gr.skip()，而不是为每个元素返回一个 skip 元组。

连续运行事件

您还可以通过使用事件监听器的 then 方法来连续运行事件。这将在前一个事件运行完成后运行一个事件。这对于运行分多步更新组件的事件非常有用。

例如，在下面的聊天机器人示例中，我们首先立即使用用户消息更新聊天机器人，然后在模拟延迟后使用计算机响应更新聊天机器人。

import gradio as gr
import random
import time

with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.Button("Clear")

    def user(user_message, history):
        return "", history + [[user_message, None]]

    def bot(history):
        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
        time.sleep(2)
        history[-1][1] = bot_message
        return history

    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
        bot, chatbot, chatbot
    )
    clear.click(lambda: None, None, chatbot, queue=False)

demo.launch()

事件监听器的 `.then()` 方法执行后续事件，无论之前的事件是否引发任何错误。如果您只想在之前的事件成功执行后才运行后续事件，请使用 `.success()` 方法，该方法接受与 `.then()` 相同的参数。

将多个触发器绑定到一个函数

通常，您可能希望将多个触发器绑定到同一个函数。例如，您可能希望允许用户单击提交按钮或按 Enter 键来提交表单。您可以使用 gr.on 方法并将触发器列表传递给 trigger 来实现这一点。

import gradio as gr

with gr.Blocks() as demo:
    name = gr.Textbox(label="Name")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Greet")
    trigger = gr.Textbox(label="Trigger Box")

    def greet(name, evt_data: gr.EventData):
        return "Hello " + name + "!", evt_data.target.__class__.__name__

    def clear_name(evt_data: gr.EventData):
        return ""

    gr.on(
        triggers=[name.submit, greet_btn.click],
        fn=greet,
        inputs=name,
        outputs=[output, trigger],
    ).then(clear_name, outputs=[name])

demo.launch()

您也可以使用装饰器语法

import gradio as gr

with gr.Blocks() as demo:
    name = gr.Textbox(label="Name")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Greet")

    @gr.on(triggers=[name.submit, greet_btn.click], inputs=name, outputs=output)
    def greet(name):
        return "Hello " + name + "!"

demo.launch()

您可以使用 gr.on 通过绑定到实现了 change 事件的组件的 change 事件来创建“实时”事件。如果您不指定任何触发器，该函数将自动绑定到所有包含 change 事件的输入组件的所有 change 事件 (例如，gr.Textbox 有 change 事件，而 gr.Button 没有)。

import gradio as gr

with gr.Blocks() as demo:
    with gr.Row():
        num1 = gr.Slider(1, 10)
        num2 = gr.Slider(1, 10)
        num3 = gr.Slider(1, 10)
    output = gr.Number(label="Sum")

    @gr.on(inputs=[num1, num2, num3], outputs=output)
    def sum(a, b, c):
        return a + b + c

demo.launch()

您可以像任何常规事件监听器一样，在 gr.on 之后使用 `.then`。这个方便的方法应该可以避免您编写大量重复代码！

将组件值直接绑定到其他组件的函数

如果您想将组件的值设置为始终是其他组件值的函数，您可以使用以下简写

with gr.Blocks() as demo:
  num1 = gr.Number()
  num2 = gr.Number()
  product = gr.Number(lambda a, b: a * b, inputs=[num1, num2])

这在功能上与以下代码相同

with gr.Blocks() as demo:
  num1 = gr.Number()
  num2 = gr.Number()
  product = gr.Number()

  gr.on(
    [num1.change, num2.change, demo.load], 
    lambda a, b: a * b, 
    inputs=[num1, num2], 
    outputs=product
  )
←

四种界面

控制布局

→