Spaces:
Runtime error
Runtime error
File size: 2,243 Bytes
5cbccd9 69d6754 5cbccd9 | 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 81 82 | import gradio as gr
from gradio import ChatMessage
import time
sleep_time = 0.1
long_sleep_time = 1
def generate_response(message, history):
start_time = time.time()
responses = [
ChatMessage(
content="In order to find the current weather in San Francisco, I will need to use my weather tool.",
)
]
yield responses
time.sleep(sleep_time)
main_thought = ChatMessage(
content="",
metadata={"title": "Using Weather Tool", "id": 1, "status": "pending"},
)
responses.append(main_thought)
yield responses
time.sleep(long_sleep_time)
responses[-1].content = "Will check: weather.com and sunny.org"
yield responses
time.sleep(sleep_time)
responses.append(
ChatMessage(
content="Received weather from weather.com.",
metadata={"title": "Checking weather.com", "parent_id": 1, "id": 2, "duration": 0.05},
)
)
yield responses
sunny_start_time = time.time()
time.sleep(sleep_time)
sunny_thought = ChatMessage(
content="API Error when connecting to sunny.org 💥",
metadata={"title": "Checking sunny.org", "parent_id": 1, "id": 3, "status": "pending"},
)
responses.append(sunny_thought)
yield responses
time.sleep(sleep_time)
responses.append(
ChatMessage(
content="Failed again",
metadata={"title": "I will try again", "id": 4, "parent_id": 3, "duration": 0.1},
)
)
sunny_thought.metadata["status"] = "done"
sunny_thought.metadata["duration"] = time.time() - sunny_start_time
main_thought.metadata["status"] = "done"
main_thought.metadata["duration"] = time.time() - start_time
yield responses
time.sleep(long_sleep_time)
responses.append(
ChatMessage(
content="Based on the data only from weather.com, the current weather in San Francisco is 60 degrees and sunny.",
)
)
yield responses
demo = gr.ChatInterface(
generate_response,
title="Nested Thoughts Chat Interface",
examples=["What is the weather in San Francisco right now?"],
api_name="chat"
)
if __name__ == "__main__":
demo.launch()
|