84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
from nicegui import ui, binding, app, run
|
|
import subprocess
|
|
from collections import defaultdict
|
|
import docker
|
|
import time
|
|
from loguru import logger
|
|
|
|
docker_client = docker.from_env()
|
|
|
|
# Dictionary of applications: key is the button text, value is the command to execute
|
|
applications = {
|
|
"Terminal": ["qterminal"],
|
|
"rqt": ["rqt"],
|
|
"rviz2": ["rviz2"],
|
|
"Gazebo": ["/gz_entrypoint.sh"],
|
|
"Gazebo Standalone": "gz sim -v4".split(),
|
|
# Add more applications here if needed
|
|
}
|
|
|
|
# Function to launch an application
|
|
def launch_app(command):
|
|
try:
|
|
subprocess.Popen(command)
|
|
except FileNotFoundError:
|
|
print(f"{command[0]} not found. Make sure it's installed and accessible in the PATH.")
|
|
|
|
# Create the NiceGUI interface
|
|
ui.label("Spiri Robotics SDK").style('font-size: 40px; margin-bottom: 10px;').classes('w-full text-center')
|
|
|
|
robots = []
|
|
from spiri_sdk_guitools.sim_drone import robot_types
|
|
import asyncio
|
|
|
|
@ui.page('/')
|
|
async def main():
|
|
with ui.tabs().classes('w-full') as tabs:
|
|
tab_tools = ui.tab('Tools')
|
|
tab_robots = ui.tab('Robots')
|
|
# two = ui.tab('Two')
|
|
with ui.tab_panels(tabs, value=tab_tools).classes("w-full"):
|
|
with ui.tab_panel(tab_tools):
|
|
# Create and place buttons dynamically based on the dictionary
|
|
with ui.grid(columns=3):
|
|
for app_name, command in applications.items():
|
|
ui.button(app_name, on_click=lambda cmd=command: launch_app(cmd)).style('width: 150px; height: 50px; margin: 5px;')
|
|
with ui.tab_panel(tab_robots):
|
|
new_robot_widget = ui.element().classes("w-1/2")
|
|
with ui.element().classes("w-1/2"):
|
|
ui.label("Debug").classes("text-xl")
|
|
def cleanup_all_containers():
|
|
#Find all containers that start with spiri-sdk
|
|
containers = docker_client.containers.list(all=True)
|
|
removal_count = 0
|
|
for container in containers:
|
|
if container.name.startswith("robot-sim-"):
|
|
container.remove(force=True)
|
|
removal_count += 1
|
|
ui.label(f"Removed {removal_count} containers")
|
|
ui.button("Cleanup all containers", on_click=cleanup_all_containers)
|
|
|
|
ui.separator()
|
|
ui.label("Current robots").classes("text-3xl")
|
|
robots_widget = ui.element().classes("w-full")
|
|
|
|
#Add a new robot
|
|
with new_robot_widget:
|
|
for name, robot in robot_types.items():
|
|
ui.label(f"Add new {name}").classes("text-3xl")
|
|
await robot.launch_widget(robots_widget)
|
|
|
|
import aiomonitor
|
|
async def amonitor():
|
|
#
|
|
logger.info("Starting aiomonitor")
|
|
loop = asyncio.get_running_loop()
|
|
run_forever = loop.create_future()
|
|
with aiomonitor.start_monitor(loop):
|
|
await run_forever
|
|
|
|
app.on_startup(amonitor)
|
|
|
|
# Start the NiceGUI application
|
|
ui.run(title="Spiri SDK Launcher", port=8923, dark=None)
|