diff --git a/libs/computer/computer/ui/gradio/app.py b/libs/computer/computer/ui/gradio/app.py index b1d131d9..7825de3f 100644 --- a/libs/computer/computer/ui/gradio/app.py +++ b/libs/computer/computer/ui/gradio/app.py @@ -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 )