diff --git a/Data/Agent/borealis-agent.py b/Data/Agent/borealis-agent.py index f3a68ad..a97e700 100644 --- a/Data/Agent/borealis-agent.py +++ b/Data/Agent/borealis-agent.py @@ -133,6 +133,7 @@ def stop_all_roles(): running_roles.clear() running_threads.clear() + def start_role_thread(role_cfg): role = role_cfg.get("role") node_id = role_cfg.get("node_id") @@ -194,5 +195,13 @@ def run_screenshot_loop(node_id, cfg): # ---------------- Main ---------------- if __name__ == "__main__": app_instance = QtWidgets.QApplication(sys.argv) - sio.connect(SERVER_URL, transports=["websocket"]) + retry_interval = 5 # seconds between connection attempts + while True: + try: + print(f"[WebSocket] Connecting to {SERVER_URL}...") + sio.connect(SERVER_URL, transports=["websocket"]()) + break + except Exception as e: + print(f"[WebSocket] Borealis Server is Not Running - Retrying in {retry_interval} seconds...") + time.sleep(retry_interval) sys.exit(app_instance.exec_()) diff --git a/Launch-Borealis.ps1 b/Launch-Borealis.ps1 index 42ad17e..2e34521 100644 --- a/Launch-Borealis.ps1 +++ b/Launch-Borealis.ps1 @@ -133,7 +133,7 @@ switch ($choice) { #> Run-Step "Install Python Dependencies into Virtual Python Environment" { if (Test-Path "$dataSource\Server\server-requirements.txt") { - & $venvPython -m pip install -q -r "$dataSource\Server\server-requirements.txt" | Out-Null + & $venvPython -m pip install --disable-pip-version-check -q -r "$dataSource\Server\server-requirements.txt" | Out-Null } } @@ -217,21 +217,16 @@ switch ($choice) { #> Run-Step "Install Python Dependencies for Agent" { if (Test-Path $agentRequirements) { - & $venvPython -m pip install -q -r $agentRequirements | Out-Null + & $venvPython -m pip install --disable-pip-version-check -q -r $agentRequirements | Out-Null } } - <# - Step: Launch Agent - #> - Run-Step "Launch Borealis Agent" { Write-Host "`nLaunching Borealis Agent..." -ForegroundColor Blue Write-Host "====================================================================================" # call python with the absolute interpreter path and the absolute script path $agentScript = Join-Path $scriptDir $agentDestinationFile - & $venvPython $agentScript + & $venvPython -W ignore::SyntaxWarning $agentScript } - } default { Write-Host "Invalid selection. Exiting..." -ForegroundColor Yellow diff --git a/Launch-Borealis.sh b/Launch-Borealis.sh index d73162d..ee363ca 100644 --- a/Launch-Borealis.sh +++ b/Launch-Borealis.sh @@ -97,7 +97,7 @@ launch_server() { source "${venvFolder}/bin/activate" run_step "Install Python Dependencies" bash -c " - pip install -q -r '${dataSource}/server-requirements.txt' + pip install --disable-pip-version-check -q -r '${dataSource}/server-requirements.txt' " run_step "Install React Dependencies" bash -c " @@ -143,7 +143,7 @@ launch_agent() { source "${venvFolder}/bin/activate" run_step "Install Python Dependencies for Agent" bash -c " - pip install -q -r '${agentRequirements}' + pip install --disable-pip-version-check -q -r '${agentRequirements}' " echo -e "\n${GREEN}Launching Borealis Agent...${RESET}"