Style stop button

app.py

@@ -657,7 +657,7 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js) as demo:
                 label="Header"
             )
 
-    stop_btn = gr.Button("Stop the agent!")
+    stop_btn = gr.Button("Stop the agent!", variant="stop")
 
     chatbot_display = gr.Chatbot(
         elem_id="chatbot",
@@ -695,6 +695,9 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js) as demo:
     def set_interactive(request: gr.Request):
         return update_html(True, request)
 
+    def reactivate_stop_btn():
+        return gr.Button("Stop the agent!", variant="stop")
+
     is_interactive = gr.Checkbox(value=True, visible=False)
 
     # Chain the events
@@ -708,14 +711,17 @@ with gr.Blocks(theme=theme, css=custom_css, js=custom_js) as demo:
         outputs=[chatbot_display]
     ).then(
         fn=set_interactive,
-        inputs=[],
         outputs=[sandbox_html]
+    ).then(
+        fn=reactivate_stop_btn,
+        outputs=[stop_btn]
     )
 
     def interrupt_agent(session_state):
         session_state["agent"].interrupt()
+        return gr.Button("Stopping agent... (could take time)", variant="secondary")
 
-    stop_btn.click(fn=interrupt_agent, inputs=[session_state], cancels=[run_event])
+    stop_btn.click(fn=interrupt_agent, inputs=[session_state], outputs=[stop_btn])
 
     def set_logs_source(session_state):
         session_state["replay_log"] = "udupp2fyavq_1743170323"

e2bqwen.py

@@ -47,7 +47,7 @@ On top of performing computations in the Python code snippets that you create, y
     Returns an output of type: {{tool.output_type}}
 {%- endfor %}
 
-The desktop has a resolution of <<resolution_x>>x<<resolution_y>>, take it into account to decide clicking coordinates.
+The desktop has a resolution of <<resolution_x>>x<<resolution_y>> pixels, take it into account to decide clicking coordinates.
 If you clicked somewhere in the previous action, a red crosshair will appear at the exact location of the previous click.
 The image might have change since then but the cross stays at the previous click. If your click seems to have changed nothing, check that this location is exactly where you intended to click. Otherwise correct the click coordinates.
 </tools>
@@ -146,6 +146,7 @@ Use click to move through menus on the desktop and scroll for web and specific a
 Always analyze the latest screenshot carefully before performing actions. Make sure to:
 To navigate the desktop you should open menus and click. Menus usually expand with more options, the tiny triangle next to some text in a menu means that menu expands. For example in Office in the Applications menu expands showing presentation or writing applications. 
 Always analyze the latest screenshot carefully before performing actions.
+The desktop has a resolution of <<resolution_x>>x<<resolution_y>> pixels.
 </general_guidelines>
 """
 
@@ -377,7 +378,7 @@ class E2BVisionAgent(CodeAgent):
         current_step = memory_step.step_number
 
         time.sleep(2.0)  # Let things happen on the desktop
-        screenshot_bytes = self.desktop.screenshot()
+        screenshot_bytes = self.desktop.screenshot(format="bytes")
         image = Image.open(BytesIO(screenshot_bytes))
 
         if getattr(self, "click_coordinates", None):