mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-07-27 09:38:28 -06:00
Added Customizable Persistent Agent SERVER_URL Config File
This commit is contained in:
@ -9,12 +9,44 @@ import socketio
|
|||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
import socket
|
import socket
|
||||||
import os
|
import os
|
||||||
|
import json
|
||||||
|
|
||||||
from PyQt5 import QtCore, QtGui, QtWidgets
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
||||||
from PIL import ImageGrab
|
from PIL import ImageGrab
|
||||||
|
|
||||||
# ---------------- Configuration ----------------
|
# ---------------- Configuration ----------------
|
||||||
SERVER_URL = os.environ.get("SERVER_URL", "http://localhost:5000")
|
CONFIG_PATH = os.path.join(os.path.dirname(__file__), "agent_settings.json")
|
||||||
|
DEFAULT_SERVER_URL = "http://localhost:5000"
|
||||||
|
|
||||||
|
def load_config():
|
||||||
|
"""
|
||||||
|
Load agent_settings.json or prompt the user for SERVER_URL.
|
||||||
|
Returns a config dictionary with at least {"SERVER_URL": ...}
|
||||||
|
"""
|
||||||
|
config = {}
|
||||||
|
|
||||||
|
if os.path.exists(CONFIG_PATH):
|
||||||
|
try:
|
||||||
|
with open(CONFIG_PATH, "r") as f:
|
||||||
|
config = json.load(f)
|
||||||
|
if isinstance(config, dict) and "SERVER_URL" in config:
|
||||||
|
return config
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[WARN] Failed to parse agent_settings.json: {e}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
user_input = input(f"Enter Borealis Server URL [{DEFAULT_SERVER_URL}]: ").strip()
|
||||||
|
config["SERVER_URL"] = user_input if user_input else DEFAULT_SERVER_URL
|
||||||
|
with open(CONFIG_PATH, "w") as f:
|
||||||
|
json.dump(config, f, indent=2)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[ERROR] Prompt failed: {e}")
|
||||||
|
config["SERVER_URL"] = DEFAULT_SERVER_URL
|
||||||
|
|
||||||
|
return config
|
||||||
|
|
||||||
|
config = load_config()
|
||||||
|
SERVER_URL = config["SERVER_URL"]
|
||||||
|
|
||||||
HOSTNAME = socket.gethostname().lower()
|
HOSTNAME = socket.gethostname().lower()
|
||||||
RANDOM_SUFFIX = uuid.uuid4().hex[:8]
|
RANDOM_SUFFIX = uuid.uuid4().hex[:8]
|
||||||
|
Reference in New Issue
Block a user