diff --git a/src/interface/desktop/main_window.py b/src/interface/desktop/main_window.py index 9f2b0544..fbe0fa00 100644 --- a/src/interface/desktop/main_window.py +++ b/src/interface/desktop/main_window.py @@ -134,7 +134,7 @@ class MainWindow(QtWidgets.QMainWindow): action_bar_layout = QtWidgets.QHBoxLayout(action_bar) self.configure_button = QtWidgets.QPushButton("Configure", clicked=self.configure_app) - self.search_button = QtWidgets.QPushButton("Search", clicked=lambda: webbrowser.open('http://localhost:8000/')) + self.search_button = QtWidgets.QPushButton("Search", clicked=lambda: webbrowser.open(f'http://{state.host}:{state.port}/')) self.search_button.setEnabled(not self.first_run) action_bar_layout.addWidget(self.configure_button) diff --git a/src/interface/desktop/system_tray.py b/src/interface/desktop/system_tray.py index 39d257e8..c24a9911 100644 --- a/src/interface/desktop/system_tray.py +++ b/src/interface/desktop/system_tray.py @@ -5,7 +5,7 @@ import webbrowser from PyQt6 import QtGui, QtWidgets # Internal Packages -from src.utils import constants +from src.utils import constants, state def create_system_tray(gui: QtWidgets.QApplication, main_window: QtWidgets.QMainWindow): @@ -24,7 +24,7 @@ def create_system_tray(gui: QtWidgets.QApplication, main_window: QtWidgets.QMain # Create the menu and menu actions menu = QtWidgets.QMenu() menu_actions = [ - ('Search', lambda: webbrowser.open('http://localhost:8000/')), + ('Search', lambda: webbrowser.open(f'http://{state.host}:{state.port}/')), ('Configure', main_window.show), ('Quit', gui.quit), ] diff --git a/src/main.py b/src/main.py index 45c00e77..ef8f90ae 100644 --- a/src/main.py +++ b/src/main.py @@ -79,6 +79,8 @@ def set_state(args): state.config_file = args.config_file state.config = args.config state.verbose = args.verbose + state.host = args.host + state.port = args.port def start_server(app, host=None, port=None, socket=None): diff --git a/src/utils/state.py b/src/utils/state.py index 90f9296a..4f194331 100644 --- a/src/utils/state.py +++ b/src/utils/state.py @@ -13,4 +13,6 @@ processor_config = ProcessorConfigModel() config_file: Path = "" verbose: int = 0 device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu") # Set device to GPU if available +host: str = None +port: int = None cli_args = None \ No newline at end of file