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"],
    # 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)