Add to computer UI

This commit is contained in:
Dillon DuPont
2025-06-13 11:39:49 -04:00
parent 84de38bac9
commit 6a5bac8b19

View File

@@ -529,10 +529,10 @@ async def execute(name, action, arguments):
return results
async def handle_init_computer(os_choice: str, app_list=None, provider="lume"):
"""Initialize the computer instance and tools for macOS or Ubuntu
"""Initialize the computer instance and tools for macOS or Ubuntu or Windows
Args:
os_choice: The OS to use ("macOS" or "Ubuntu")
os_choice: The OS to use ("macOS" or "Ubuntu" or "Windows")
app_list: Optional list of apps to focus on using the app-use experiment
provider: The provider to use ("lume" or "self")
"""
@@ -548,6 +548,9 @@ async def handle_init_computer(os_choice: str, app_list=None, provider="lume"):
if os_choice == "Ubuntu":
os_type_str = "linux"
image_str = "ubuntu-noble-vanilla:latest"
elif os_choice == "Windows":
os_type_str = "windows"
image_str = "windows-11-vanilla:latest"
else:
os_type_str = "macos"
image_str = "macos-sequoia-cua:latest"
@@ -1065,7 +1068,7 @@ def create_gradio_ui():
with gr.Row():
os_choice = gr.Radio(
label="OS",
choices=["macOS", "Ubuntu"],
choices=["macOS", "Ubuntu", "Windows"],
value="macOS",
interactive=False # disable until the ubuntu image is ready
)