Spaces:
Sleeping
Sleeping
File size: 2,035 Bytes
b812ab1 7acf0e2 b812ab1 7acf0e2 b812ab1 7acf0e2 b812ab1 7acf0e2 | 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 | import gradio as gr
from sidekick import Sidekick
async def setup():
sidekick = Sidekick()
await sidekick.setup()
return sidekick
async def process_message(sidekick, message, success_criteria, history):
if sidekick is None:
sidekick = Sidekick()
await sidekick.setup()
results = await sidekick.run_superstep(message, success_criteria, history)
return results, sidekick
async def reset():
new_sidekick = Sidekick()
await new_sidekick.setup()
return "", "", None, new_sidekick
def free_resources(sidekick):
print("Cleaning up")
try:
if sidekick:
sidekick.cleanup()
except Exception as e:
print(f"Exception during cleanup: {e}")
with gr.Blocks(title="Sidekick") as ui:
gr.Markdown("## Shima's Sidekick Personal Co-Worker")
sidekick = gr.State(None, delete_callback=free_resources)
with gr.Row():
chatbot = gr.Chatbot(label="Sidekick", height=300)
with gr.Group():
with gr.Row():
message = gr.Textbox(show_label=False, placeholder="Your request to the Sidekick")
with gr.Row():
success_criteria = gr.Textbox(
show_label=False, placeholder="What are your success criteria?"
)
with gr.Row():
reset_button = gr.Button("Reset", variant="stop")
go_button = gr.Button("Go!", variant="primary")
ui.load(setup, [], [sidekick])
message.submit(
process_message, [sidekick, message, success_criteria, chatbot], [chatbot, sidekick]
)
success_criteria.submit(
process_message, [sidekick, message, success_criteria, chatbot], [chatbot, sidekick]
)
go_button.click(
process_message, [sidekick, message, success_criteria, chatbot], [chatbot, sidekick]
)
reset_button.click(reset, [], [message, success_criteria, chatbot, sidekick])
ui.launch(
theme=gr.themes.Default(primary_hue="emerald"),
inbrowser=True,
share=True,
server_name="0.0.0.0",
server_port=7860
) |