File size: 5,585 Bytes
7a3e44b 323c7ea 7a3e44b 80287f4 7a3e44b 323c7ea 3bf7116 323c7ea 3bf7116 323c7ea 3bf7116 323c7ea 3bf7116 323c7ea be02ece 323c7ea 4410066 5a57f54 3bf7116 323c7ea be02ece 3bf7116 7a3e44b 2ab9625 7a3e44b 458e2d8 3bf7116 7a3e44b 3bf7116 7a3e44b 80287f4 7a3e44b be02ece 7a3e44b 323c7ea be02ece 323c7ea 7a3e44b 323c7ea 3bf7116 2ab9625 7a3e44b 2ab9625 |
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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 |
import gradio as gr
import os
import time
from agent import DevOn
image_temp = "https://miro.medium.com/v2/resize:fit:1200/0*n-2bW82Z6m6U2bij.jpeg"
# devon = DevOn(
# editor_image=image_temp, browser_image=image_temp, scratchpad_image=image_temp
# )
devon = None
# multion_api_key = ""
# openai_api_key = ""
# replit_email = ""
# replit_password = ""
def add_message(history, message):
# for x in message["files"]:
# history.append(((x,), None))
if message["text"] is not None:
history.append((message["text"], None))
return history, gr.MultimodalTextbox(value=None, interactive=False)
# def multion_api_key_update(x):
# # global multion_api_key
# multion_api_key = x
# def openai_api_key_update(x):
# # global openai_api_key
# openai_api_key = x
# def replit_email_update(x):
# # global replit_email
# replit_email = x
# def replit_password_update(x):
# # global replit_password
# replit_password = x
def bot(
history,
multion_api_key_in,
openai_api_key_in,
replit_email_in,
replit_password_in,
local,
):
if len(multion_api_key_in) == 0:
raise gr.Error("MultiOn API Key is Required.")
if len(openai_api_key_in) == 0:
raise gr.Error("OpenAI API Key is Required.")
start_time = time.time()
devon = DevOn(
editor_image=image_temp,
browser_image=image_temp,
scratchpad_image=image_temp,
multion_api_key=multion_api_key_in,
openai_api_key=openai_api_key_in,
replit_email=replit_email_in,
replit_password=replit_password_in,
local=local,
)
for r in devon.run(history[-1][0]):
curr_time = time.time()
print(curr_time - start_time)
if curr_time - start_time >= 300:
break
text, editor_image, browser_image, scratchpad_image = r
if type(text) == str:
history.append((None, text))
if editor_image is None:
editor_image = devon.editor_image
browser_image = devon.browser_image
scratchpad_image = devon.scratchpad_image
yield history, editor_image, browser_image, scratchpad_image
with gr.Blocks(css="footer {visibility: hidden}") as demo:
md = gr.Markdown(
"""Notes:
- Use "Execute Locally" for better results.
- For local execution, you need to download the [MultiOn Browser Extension](https://chromewebstore.google.com/detail/multion/ddmjhdbknfidiopmbaceghhhbgbpenmm) and have "API Enabled" in the settings.
- The Huggingface Spaces demo will timeout after 5 minutes by default. To test with longer tasks, [clone the repo](https://github.com/lordspline/DevOn) and run DevOn locally."""
)
with gr.Row():
with gr.Column():
multion_api_key_in = gr.Textbox(label="MultiOn API Key")
openai_api_key_in = gr.Textbox(label="OpenAI API Key")
with gr.Column():
replit_email_in = gr.Textbox(label="Replit Email")
replit_password_in = gr.Textbox(label="Replit Password")
with gr.Row():
with gr.Column():
chatbot = gr.Chatbot(
[], elem_id="chatbot", bubble_full_width=False, height=300
)
chat_input = gr.MultimodalTextbox(
value={
"text": "benchmark the perplexity api's resposne time with the api key abcdef"
},
interactive=True,
file_types=["text"],
placeholder="Enter message or upload file...",
show_label=False,
)
with gr.Row():
local = gr.Checkbox(True, label="Execute Locally")
terminate = gr.Button("Terminate")
with gr.Column():
if devon:
editor_view = gr.Image(
devon.editor_image,
label="Editor",
)
else:
editor_view = gr.Image()
with gr.Row():
with gr.Column():
if devon:
browser_view = gr.Image(
devon.browser_image,
label="Browser",
)
else:
browser_view = gr.Image()
with gr.Column():
if devon:
scratchpad_view = gr.Image(
devon.scratchpad_image,
label="Scratchpad",
)
else:
scratchpad_view = gr.Image()
chat_msg = chat_input.submit(
add_message, [chatbot, chat_input], [chatbot, chat_input]
)
bot_msg = chat_msg.then(
bot,
[
chatbot,
multion_api_key_in,
openai_api_key_in,
replit_email_in,
replit_password_in,
local,
],
[chatbot, editor_view, browser_view, scratchpad_view],
api_name="bot_response",
)
bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
# multion_api_key_in.change(multion_api_key_update, multion_api_key_in)
# openai_api_key_in.change(openai_api_key_update, openai_api_key_in)
# replit_email_in.change(replit_email_update, replit_email_in)
# replit_password_in.change(replit_password_update, replit_password_in)
terminate.click(fn=None, inputs=None, outputs=None, cancels=[bot_msg])
# chatbot.like(print_like_dislike, None, None)
if __name__ == "__main__":
demo.queue()
demo.launch(debug=True)
|