Feature/Chore: improve server flags (#80)

* feat: add silent flag and wrap all messages with that

* feat: pass silent flag to all ways of invoking the server

* fix: add missing parameter to enable execution

* feat: send global config along with request

* fix: add settings to the post api request

* fix: removes double json parsing

* chore: cleanup print statements

* fix: formatting

Author: madcampos

src/server/package/src/model_explorer/apis.py

@@ -68,6 +68,7 @@ def visualize(
     reuse_server_host: str = DEFAULT_HOST,
     reuse_server_port: Union[int, None] = None,
     enable_execution: bool = True,
+    silent: bool = False,
 ) -> None:
   """Starts the ME local server and visualizes the models by the given paths.
 
@@ -83,6 +84,7 @@ def visualize(
     reuse_server_host: the host of the server to reuse. Default to localhost.
     reuse_server_port: the port of the server to reuse. If unspecified, it will
         try to find a running server from port 8080 to 8099.
+    silent: If the server should be silent and only output error messages.
   """
   # Construct config.
   cur_config = config()
@@ -108,6 +110,7 @@ def visualize(
       colab_height=colab_height,
       extensions=extensions,
       enable_execution=enable_execution,
+      silent=silent,
   )
 
 
@@ -124,6 +127,7 @@ def visualize_pytorch(
     reuse_server_host: str = DEFAULT_HOST,
     reuse_server_port: Union[int, None] = None,
     enable_execution: bool = True,
+    silent: bool = False,
 ) -> None:
   """Visualizes a pytorch model.
 
@@ -141,6 +145,7 @@ def visualize_pytorch(
     reuse_server_host: the host of the server to reuse. Default to localhost.
     reuse_server_port: the port of the server to reuse. If unspecified, it will
         try to find a running server from port 8080 to 8099.
+    silent: If the server should be silent and only output error messages.
   """
   # Construct config.
   cur_config = config()
@@ -163,6 +168,7 @@ def visualize_pytorch(
       colab_height=colab_height,
       extensions=extensions,
       enable_execution=enable_execution,
+      silent=silent,
   )
 
 
@@ -175,6 +181,7 @@ def visualize_from_config(
     no_open_in_browser: bool = False,
     colab_height=DEFAULT_COLAB_HEIGHT,
     enable_execution: bool = True,
+    silent: bool = False,
 ) -> None:
   """Visualizes with a config.
 
@@ -187,6 +194,7 @@ def visualize_from_config(
       won't be present if it is None.
     no_open_in_browser: Don't open the web app in browser after server starts.
     colab_height: The height of the embedded iFrame when running in colab.
+    silent: If the server should be silent and only output error messages.
   """
   # Start server.
   server.start(
@@ -198,6 +206,7 @@ def visualize_from_config(
       no_open_in_browser=no_open_in_browser,
       colab_height=colab_height,
       enable_execution=enable_execution,
+      silent=silent,
   )
 
 

src/server/package/src/model_explorer/cmdline.py

@@ -70,6 +70,11 @@
         'to 8099'
     ),
 )
+parser.add_argument(
+    '--silent',
+    action='store_false',
+    help='The server should be silent and only display error messages',
+)
 args = parser.parse_args()
 
 
@@ -108,4 +113,6 @@ def main():
       cors_host=args.cors_host,
       no_open_in_browser=args.no_open_in_browser,
       skip_health_check=args.skip_health_check,
+      enable_execution=True,
+      silent=args.silent,
   )

src/server/package/src/model_explorer/server.py

@@ -179,7 +179,6 @@ def _is_internal_colab() -> bool:
 
 
 def start(
-    enable_execution: bool,
     host=DEFAULT_HOST,
     port=DEFAULT_PORT,
     config: Union[ModelExplorerConfig, None] = None,
@@ -188,6 +187,8 @@ def start(
     colab_height: int = DEFAULT_COLAB_HEIGHT,
     cors_host: Union[str, None] = None,
     skip_health_check: bool = False,
+    enable_execution: bool = True,
+    silent: bool = False,
 ):
   """Starts the local server that serves the web app.
 
@@ -253,23 +254,28 @@ def start(
   cli.show_server_banner = lambda *x: None
 
   # Print a info message when used in colab.
-  if colab:
+  if colab and not silent:
     print('ℹ️ Please re-run the cell in each new session')
     print()
 
   # Load extensions.
-  print('Loading extensions...')
+  if not silent:
+    print('Loading extensions...')
+
   extension_manager = ExtensionManager(extensions)
   extension_manager.load_extensions()
   extension_metadata_list = extension_manager.get_extensions_metadata(
       enable_execution
   )
   num_extensions = len(extension_metadata_list)
-  print(
-      f'Loaded {num_extensions} extension{"" if num_extensions == 1 else "s"}:'
-  )
-  for extension in extension_metadata_list:
-    print(f' - {extension["name"]}')
+
+  if not silent:
+    print(
+        'Loaded'
+        f' {num_extensions} extension{"" if num_extensions == 1 else "s"}:'
+    )
+    for extension in extension_metadata_list:
+      print(f' - {extension["name"]}')
 
   @app.route('/api/v1/check_new_version')
   def check_new_version():
@@ -297,6 +303,15 @@ def upload_file():
   @app.route('/api/v1/send_command')
   def send_command():
     cmd_json = json.loads(request.args.get('json', '{}'))
+
+    # Initialize settings if it is not provided
+    if 'settings' not in cmd_json:
+      cmd_json['settings'] = {}
+
+    # Overwrite settings with globals
+    cmd_json['settings']['enable_execution'] = enable_execution
+    cmd_json['settings']['silent'] = silent
+
     try:
       resp = extension_manager.run_cmd(cmd_json)
       return _make_json_response(resp)
@@ -309,7 +324,17 @@ def send_command():
   @app.route('/apipost/v1/send_command', methods=['POST'])
   def send_command_post():
     try:
-      resp = extension_manager.run_cmd(request.json)
+      cmd_json = request.json
+
+      # Initialize settings if it is not provided
+      if 'settings' not in cmd_json:
+        cmd_json['settings'] = {}
+
+      # Overwrite settings with globals
+      cmd_json['settings']['enable_execution'] = enable_execution
+      cmd_json['settings']['silent'] = silent
+
+      resp = extension_manager.run_cmd(cmd_json)
       return _make_json_response(resp)
     except Exception as err:
       traceback.print_exc()
@@ -459,21 +484,25 @@ def start_server():
 
     if len(url_params) > 0:
       server_address = f'{server_address}/?{"&".join(url_params)}'
-    print(
-        f'\nStarting Model Explorer server at:\n{server_address}\n\nPress'
-        ' Ctrl+C to stop.'
-    )
+
+    if not silent:
+      print(
+          f'\nStarting Model Explorer server at:\n{server_address}\n\nPress'
+          ' Ctrl+C to stop.'
+      )
+
     if not no_open_in_browser:
       webbrowser.open_new_tab(f'{server_address}')
 
     # Check installed version vs published version.
-    threading.Thread(target=lambda: _check_new_version()).start()
+    threading.Thread(target=lambda: _check_new_version(not silent)).start()
 
     try:
       while app_thread.is_alive():
         sleep(1)
     except KeyboardInterrupt:
-      print('Stopping server...')
+      if not silent:
+        print('Stopping server...')
       pass
 
   def embed_in_colab():