bunny-lab/Borealis
2
0
Fork 0
Code Issues 10 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Implemented Websocket Support for Agents

Browse Source
This commit is contained in:
Nicole Rappe
2025-04-09 01:20:46 -06:00
parent 5c35447a15
commit 4e48d119f7
5 changed files with 131 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Data/Agent/borealis-agent.py
View File

@ -11,7 +11,8 @@ from PyQt5 import QtCore, QtGui, QtWidgets
from PIL import ImageGrab
# ---------------- Configuration ----------------
SERVER_URL = "http://localhost:5000" # WebSocket-enabled server URL
#SERVER_URL = "http://localhost:5000" # WebSocket-enabled Internal URL
SERVER_URL = "https://borealis.bunny-lab.io" # WebSocket-enabled Public URL"
HOSTNAME = socket.gethostname().lower()
RANDOM_SUFFIX = uuid.uuid4().hex[:8]
@ -32,7 +33,7 @@ sio = socketio.Client()
# ---------------- WebSocket Handlers ----------------
@sio.event
def connect():
print(f"[WS CONNECTED] Agent ID: {AGENT_ID} connected to Borealis.")
print(f"[WebSocket] Agent ID: {AGENT_ID} connected to Borealis.")
sio.emit('connect_agent', {"agent_id": AGENT_ID, "hostname": HOSTNAME})
sio.emit('request_config', {"agent_id": AGENT_ID})
@ -123,7 +124,6 @@ class ScreenshotRegion(QtWidgets.QWidget):
# ---------------- Screenshot Capture ----------------
def capture_loop():
print("[INFO] Screenshot capture loop started")
config_ready.wait()
while region_widget is None:
@ -172,7 +172,7 @@ if __name__ == "__main__":
app_instance = QtWidgets.QApplication(sys.argv)
region_launcher = RegionLauncher()
sio.connect(SERVER_URL)
sio.connect(SERVER_URL, transports=['websocket'])
threading.Thread(target=capture_loop, daemon=True).start()