File size: 2,115 Bytes
6751ef9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import gradio as gr
from gradio import ChatMessage
import time
sleep_time = 0.1
long_sleep_time = 1
def generate_response(history):
history.append(
ChatMessage(
role="user", content="What is the weather in San Francisco right now?"
)
)
yield history
time.sleep(sleep_time)
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(sleep_time)
history.append(
ChatMessage(
role="assistant",
content="",
metadata={"title": "Gathering Weather Websites", "id": 1},
)
)
yield history
time.sleep(long_sleep_time)
history[-1].content = "Will check: weather.com and sunny.org"
yield history
time.sleep(sleep_time)
history.append(
ChatMessage(
role="assistant",
content="Received weather from weather.com.",
metadata={"title": "API Success ✅", "parent_id": 1, "id": 2},
)
)
yield history
time.sleep(sleep_time)
history.append(
ChatMessage(
role="assistant",
content="API Error when connecting to sunny.org.",
metadata={"title": "API Error 💥 ", "parent_id": 1, "id": 3},
)
)
yield history
time.sleep(sleep_time)
history.append(
ChatMessage(
role="assistant",
content="I will try yet again",
metadata={"title": "I will try again", "id": 4, "parent_id": 3},
)
)
yield history
time.sleep(sleep_time)
history.append(
ChatMessage(
role="assistant",
content="Failed again",
metadata={"title": "Failed again", "id": 6, "parent_id": 4},
)
)
yield history
with gr.Blocks() as demo:
chatbot = gr.Chatbot(type="messages", height=500, show_copy_button=True)
demo.load(generate_response, chatbot, chatbot)
if __name__ == "__main__":
demo.launch()
|