| import gradio as gr | |
| import os | |
| import threading | |
| import time | |
| from agent.cua_agent import ComputerUsingAgent | |
| # Initialize Agent | |
| agent = ComputerUsingAgent() | |
| def execute_task(task_description): | |
| """Execute a task using the agent""" | |
| return agent.execute_task(task_description) | |
| def get_status(): | |
| """Get agent status""" | |
| return { | |
| "status": agent.task_status, | |
| "current_task": agent.current_task | |
| } | |
| # Gradio UI | |
| with gr.Blocks(title="Tiny X11 Agent") as demo: | |
| gr.Markdown("# 🐜 Tiny X11 Desktop Agent") | |
| with gr.Row(): | |
| # Desktop View | |
| gr.HTML(""" | |
| <iframe src="/vnc.html?autoconnect=true&resize=scale" | |
| width="100%" height="600" | |
| style="border: 1px solid #ccc;"></iframe> | |
| """) | |
| # Controls | |
| with gr.Column(): | |
| task_input = gr.Textbox(label="Task", placeholder="Open Firefox...") | |
| btn_run = gr.Button("Run") | |
| output = gr.JSON(label="Output") | |
| btn_run.click(execute_task, inputs=task_input, outputs=output) | |
| if __name__ == "__main__": | |
| demo.launch(server_name="0.0.0.0", server_port=7860) | |