Chatbot with Tools

Adapted from Gradio Playground.

import gradio as gr
from gradio import ChatMessage
import time

def generate_response(history):
    history.append(
        ChatMessage(
            role="user", content="What is the weather in San Francisco right now?"
        )
    )
    yield history
    time.sleep(0.25)
    history.append(
        ChatMessage(
            role="assistant",
            content="In order to find the current weather in San Francisco, I will need to use my weather tool.",
        )
    )
    yield history
    time.sleep(0.25)

    history.append(
        ChatMessage(
            role="assistant",
            content="API Error when connecting to weather service.",
            metadata={"title": "💥 Error using tool 'Weather'"},
        )
    )
    yield history
    time.sleep(0.25)

    history.append(
        ChatMessage(
            role="assistant",
            content="I will try again",
        )
    )
    yield history
    time.sleep(0.25)

    history.append(
        ChatMessage(
            role="assistant",
            content="Weather 72 degrees Fahrenheit with 20% chance of rain.",
            metadata={"title": "🛠️ Used tool 'Weather'"},
        )
    )
    yield history
    time.sleep(0.25)

    history.append(
        ChatMessage(
            role="assistant",
            content="Now that the API succeeded I can complete my task.",
        )
    )
    yield history
    time.sleep(0.25)

    history.append(
        ChatMessage(
            role="assistant",
            content="It's a sunny day in San Francisco with a current temperature of 72 degrees Fahrenheit and a 20% chance of rain. Enjoy the weather!",
        )
    )
    yield history

def like(evt: gr.LikeData):
    print("User liked the response")
    print(evt.index, evt.liked, evt.value)

with gr.Blocks() as demo:
    chatbot = gr.Chatbot(type="messages", height=500, show_copy_button=True)
    button = gr.Button("Get San Francisco Weather")
    button.click(generate_response, chatbot, chatbot)
    chatbot.like(like)

if __name__ == "__main__":
    demo.launch()

import gradio as gr from gradio import ChatMessage import time def generate_response(history): history.append( ChatMessage( role="user", content="What is the weather in San Francisco right now?" ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="In order to find the current weather in San Francisco, I will need to use my weather tool.", ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="API Error when connecting to weather service.", metadata={"title": "💥 Error using tool 'Weather'"}, ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="I will try again", ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="Weather 72 degrees Fahrenheit with 20% chance of rain.", metadata={"title": "🛠️ Used tool 'Weather'"}, ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="Now that the API succeeded I can complete my task.", ) ) yield history time.sleep(0.25) history.append( ChatMessage( role="assistant", content="It's a sunny day in San Francisco with a current temperature of 72 degrees Fahrenheit and a 20% chance of rain. Enjoy the weather!", ) ) yield history def like(evt: gr.LikeData): print("User liked the response") print(evt.index, evt.liked, evt.value) with gr.Blocks() as demo: chatbot = gr.Chatbot(type="messages", height=500, show_copy_button=True) button = gr.Button("Get San Francisco Weather") button.click(generate_response, chatbot, chatbot) chatbot.like(like) if __name__ == "__main__": demo.launch()