qairt-visualizer 0.2.0__py3-none-win_amd64.whl → 0.4.0__py3-none-win_amd64.whl

qairt_visualizer/__init__.py

@@ -7,6 +7,8 @@
 # ==============================================================================
 """qairt_visualizer module exports"""
 
+import importlib
+
 from qairt_visualizer.apis import view
 from qairt_visualizer.core.ui.helpers import post_install
 from qairt_visualizer.core.visualizer_logging.helpers import set_log_level
@@ -15,3 +17,4 @@ from qairt_visualizer.models.display_options import DisplayOptions
 post_install.run()
 
 __all__ = ["view", "set_log_level", "DisplayOptions"]
+__version__ = importlib.metadata.version("qairt_visualizer")

qairt_visualizer/apis.py

@@ -20,25 +20,35 @@ from qairt_visualizer.core.visualizer_logging.logging_config import LoggingConfi
 from qairt_visualizer.core.visualizer_service import VisualizerService
 from qairt_visualizer.helpers.asyncio_helpers import has_event_loop_running
 from qairt_visualizer.helpers.jupyter_helpers import is_running_in_jupyter
+from qairt_visualizer.helpers.qairt_helpers import process_model_and_reports
 from qairt_visualizer.models.app_environment_enum import AppEnvironment
 from qairt_visualizer.models.display_options import DisplayOptions
+from qairt_visualizer.models.validators.model_validator import validate_model
+from qairt_visualizer.models.validators.validate_args import validate_args
 
 
+@validate_args(
+    {
+        "model": validate_model,
+    }
+)
 def view(
-    path_to_model: Optional[str] = None,
+    model: Optional[Union[str, object]] = None,
     reports: Optional[Union[str, List[str]]] = None,
-    options: DisplayOptions = DisplayOptions(),
+    options: Optional[DisplayOptions] = None,
 ):
     """
     Opens a visualization window(s) to display a model, reports, or both.
 
-    :param path_to_model: Optional path to a given model
+    :param model: Optional path to a given model or QAIRT Model object
     :param reports: A single path or list of paths representing different reports for the
         visualization window to display.
     :param options: Customizes the visualization window behavior.
     """
     LoggingConfig.setup_logging()
 
+    options = options or DisplayOptions()
+
     if platform.system().lower() == "windows":
         asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())  # type: ignore
 
@@ -51,12 +61,18 @@ def view(
     launcher = get_launcher(mode)
     communicator = get_communicator(mode)
 
+    path_to_processed_model, path_to_processed_reports = process_model_and_reports(model, reports)
+
     if has_event_loop_running():
         loop = asyncio.get_running_loop()
-        t = loop.create_task(_view(launcher, communicator, path_to_model, reports, options))
+        t = loop.create_task(
+            _view(launcher, communicator, path_to_processed_model, path_to_processed_reports, options)
+        )
         asyncio.gather(t)
     else:
-        asyncio.run(_view(launcher, communicator, path_to_model, reports, options))
+        asyncio.run(
+            _view(launcher, communicator, path_to_processed_model, path_to_processed_reports, options)
+        )
 
 
 async def _view(

qairt_visualizer/core/launchers/web_launcher_context.py

@@ -64,11 +64,14 @@ class WebLauncherContext(BaseUILauncherContext):
                         await response.json()
                         return
                 except (aiohttp.ClientError, aiohttp.ClientResponseError, asyncio.TimeoutError) as e:
-                    self.logger.warning(
-                        "No response received from Visualizer UI. Attempting retry %d/%d",
-                        attempt + 1,
-                        ping_retries,
-                    )
+                    if attempt == 0:
+                        self.logger.warning("Waiting on response from Visualizer UI...")
+                    else:
+                        self.logger.debug(
+                            "No response received from Visualizer UI. Attempting retry %d/%d",
+                            attempt + 1,
+                            ping_retries,
+                        )
                     self.logger.debug(e)
 
                 await asyncio.sleep(0.05)
@@ -85,7 +88,7 @@ class WebLauncherContext(BaseUILauncherContext):
         display(
             HTML(
                 f"""
-                 <iframe src="http://{self._host}:{port}/#/{self.id}"
+                 <iframe src="http://{self._host}:{port}/#/workspaces/{self.id}?mode=jupyter"
                  width="100%" height="500px"></iframe>
                  """
             )

qairt_visualizer/core/ui/dist/browser/assets/i18n/common/en.json

@@ -17,6 +17,7 @@
     "DEVICES": "Devices",
     "DISCARD": "Discard"
   },
+  "common.save.label": "Save",
   "CONFIRM": "Confirm",
   "CONFIRM_EXIT": {
     "CONFIRM": "Exit QAIRT Visualizer"
@@ -40,6 +41,11 @@
   "FILTER": "Filter...",
   "GENERAL": "General",
   "GENERATE_REFERENCE": "Gen Ref",
+  "HELP": "Help",
+  "HOME": "Home",
+  "HOST": "Host",
+  "INPUTS": "Inputs",
+  "ITEMS_PER_PAGE": "Items per page:",
   "LANDING_PAGE": {
     "DROP_YOUR_FILES": "Drop your files in here, or select files from your computer.",
     "LEARN_MORE": "Learn more.",
@@ -49,11 +55,6 @@
     "SUPPORTED_FORMATS": "Supported formats:\nmodels: onnx, tensorflow, tflite, dlc, pytorch\nreports: optrace, qhas, qairt profile, layerwise accuracy",
     "WELCOME": "Welcome to Qualcomm AI Runtime Visualizer"
   },
-  "HELP": "Help",
-  "HOME": "Home",
-  "HOST": "Host",
-  "INPUTS": "Inputs",
-  "ITEMS_PER_PAGE": "Items per page:",
   "MENU": {
     "ABOUT": "About",
     "DOCUMENTATION": "Documentation",
@@ -89,9 +90,8 @@
   "OUTPUT_CARD_NAME_TOOLTIP": "Enter a descriptive name to identify the output of the \n{{action}}",
   "OUTPUTS": "Outputs",
   "PREVIOUS": "Previous",
-  "QID": "QID",
   "QAIRT_VISUALIZER": "QAIRT Visualizer",
-  "SAVE": "Save",
+  "QID": "QID",
   "SELECT": "Select",
   "SELECT_ALL": "Select/Unselect All",
   "SETTINGS": {
@@ -118,6 +118,7 @@
     "RECENTLY": "A few seconds ago"
   },
   "TOGGLE_THEME": "Toggle Theme",
+  "UNABLE_TO_OPEN": "There was an issue opening {{ filePath }}. Please try again or open it manually via your file explorer.",
   "UPDATE": "Update",
   "UPLOAD": "Upload",
   "USERNAME": "Username",
@@ -136,25 +137,18 @@
   },
   "VERSION": "Version",
   "VIEW_DOCS": "View Documentation",
-  "WELCOME_DIALOG": {
-    "AI_MODEL_ANALYTICS": "AI Model Analyics",
-    "AI_MODEL_ANALYTICS_DESCRIPTION": "Analyze comprehensive model reports for Performance and Accuracy metrics",
-    "AI_MODEL_VISUALIZATION": "AI Model Visualization",
-    "AI_MODEL_VISUALIZATION_DESCRIPTION": "Select a model or report to visualize",
-    "CLOSE": "Close",
-    "DO_NOT_SHOW_AGAIN": "Don't show again",
-    "DONE": "Done",
-    "GETTING_STARTED": "Getting Started",
-    "NEXT": "Next",
-    "WELCOME": "Welcome",
-    "WELCOME_DESCRIPTION": "Manage and run your AI models with a suite of powerful tools."
-  },
-  "UNABLE_TO_OPEN": "There was an issue opening {{ filePath }}. Please try again or open it manually via your file explorer.",
+  "workspace.dialog.replacePanel.acceptButton": "Replace File",
+  "workspace.dialog.replacePanel.description": "The {{panelType}} panel already has a visualization. Do you wish to replace it with {{filename}}?",
+  "workspace.dialog.replacePanel.title": "Replace visualized panel?",
   "WORKSPACES": {
-    "ADD_FILES_TO_WORKSPACE": "Add files to this workspace",
     "ADD_FILES_TO_NEW_WORKSPACE": "Add files to a new workspace",
+    "ADD_FILES_TO_WORKSPACE": "Add files to this workspace",
     "DELETE_SELECTED": "Delete Selected",
-    "RENAME": "Rename Workspace"
+    "SOCKET": {
+      "ERROR": {
+        "CONNECTION": "Unable to connect to the server. Please refresh and try again."
+      }
+    },
+    "UNTITLED": "Untitled"
   }
-
 }

qairt_visualizer/core/ui/dist/browser/assets/i18n/files/en.json

@@ -36,9 +36,9 @@
   "FILE_TYPE_NOT_SUPPORTED": "File type(s) not supported: {{fileExts}}",
   "FILE_TYPES": {
     "CSV": "CSV",
-    "DLC_MODEL": "DLC",
     "DLC_FP16_MODEL": "DLC FP16 Converted",
     "DLC_FP32_MODEL": "DLC FP32 Converted",
+    "DLC_MODEL": "DLC",
     "DLC_QUANT_MODEL": "DLC Quantized",
     "JPEG": "JPEG",
     "JSON": "JSON",
@@ -48,6 +48,7 @@
     "PNG": "PNG",
     "PROFILING_REPORT": "Performance Report",
     "PYTORCH_MODEL": "Pytorch",
+    "PYTORCH2_MODEL": "Pytorch2",
     "QTI_AISW_BIN": "QTI Bin",
     "QTI_AISW_CPP": "QTI CPP",
     "QTI_AISW_JSON": "QTI JSON",
@@ -66,6 +67,6 @@
   "FILES_NOT_FOUND": "No Files Found",
   "IMPORT": "Import",
   "IMPORT_FILES": "Import File(s)",
-  "UNSUPPORTED_FILE": "Unable to visualize file {{filename}} - file type {{viewType}} is not supported",
-  "PROFILER_FILE_FAILED": "Unable to visualize the performance report, please check its validity"
+  "INVALID_FILE": "Unable to visualize file: {{filename}}, please check its validity",
+  "UNSUPPORTED_FILE_TYPE": "Unable to visualize file: {{filename}} (file type is not supported)"
 }

qairt_visualizer/core/ui/dist/browser/assets/i18n/graph/en.json

@@ -13,9 +13,8 @@
     "ACCURACY_METRICS_TOOLTIP": "Analyze the model's accuracy through graphical and list views, comparing outputs using verifiers such as CosineSimilarity, RtolAtol, etc. Use the legend to toggle data series on and off",
     "ANALYSIS": "Analysis",
     "ANALYSIS_TAB_TOOLTIP": "Review comprehensive performance and accuracy reports, providing insights into the model's evaluation metrics and efficiency",
-    "COLLAPSE": "Collapse",
-    "EXPAND": "Expand",
     "FILE_NOT_FOUND": "Oops! Model download encountered an error",
+    "HIDE": "Hide",
     "LIST": "List",
     "MODEL_INPUTS_TOOLTIP": "Names and dimensions of the network input layers",
     "MODEL_OUTPUTS_TOOLTIP": "Names and dimensions of the network output layers",
@@ -30,24 +29,22 @@
     "NODE_PROPERTIES": "Node Properties",
     "NODE_PROPERTIES_TOOLTIP": "Provides information about the properties of individual nodes or layers in the neural network model",
     "PERFORMANCE_LIST_TOOLTIP": "View execution timing for each op",
-    "PERFORMANCE_METRICS": "Performance Metrics",
     "PERFORMANCE_METRICS_TOOLTIP": "Analyze the model's performance with graphical data, summary reports, and a detailed list. Use the legend to toggle data series on and off",
     "PERFORMANCE_SUMMARY_TOOLTIP": "View performance details, including execution and init times",
     "PROPERTIES": "Properties",
     "PROPERTIES_TAB_TOOLTIP": "View the model and node-specific properties, including layer types, configurations, and parameters",
-    "RESIZE": "Reset Zoom",
-    "RESTORE": "Restore",
+    "RESET_ZOOM": "Reset zoom",
     "RESULTS_FILTER_TOOLTIP": "Select a report from the dropdown to view detailed performance and accuracy analysis",
     "SAVE_AS_IMAGE": "Save as Image",
     "SEARCH": "Search nodes",
     "SEARCH_PLACEHOLDER": "Search nodes or edges",
-    "SEARCH_TOOLTIP": "Use the search bar to find and select nodes or edges from the list",
+    "SHOW": "Show",
     "SUMMARY": "Summary",
     "TITLE": "Graph",
     "USER_NOTE": "Click on a Node",
     "VISUALIZE": "Visualize AI Model",
-    "ZOOM_IN": "Zoom In",
-    "ZOOM_OUT": "Zoom Out"
+    "ZOOM_IN": "Zoom in",
+    "ZOOM_OUT": "Zoom out"
   },
   "GRAPHS": "Graphs",
   "OPTRACE_PANEL": {
@@ -58,11 +55,19 @@
     "PANEL_HEADING": "Optrace",
     "SHOW_PANEL": "Show Panel"
   },
+  "PANEL_CONTROLS": {
+    "CLOSE": "Close panel",
+    "COLLAPSE": "Collapse panel",
+    "EXIT_EXPAND": "Exit expand",
+    "EXPAND": "Expand panel",
+    "MORE": "More"
+  },
   "PERFORMANCE_METRICS": {
     "AVG": "Avg",
     "MAX": "Max",
     "MIN": "Min",
-    "NODE_NOT_FOUND": "{{nodeName}} could not be found in Performance Metrics"
+    "NODE_NOT_FOUND": "{{nodeName}} could not be found in Performance Metrics",
+    "RESET_CHART": "Reset chart"
   },
   "REPORT": {
     "VIEW_REPORT": "View Report"
@@ -71,7 +76,6 @@
   "SHOW_BOTH": "Show in Both",
   "SHOW_GRAPH": "Show in Graph",
   "SHOW_OPTRACE": "Show in Op-trace",
-  "UNMATCHED_FILE_TYPE": "Invalid file type for {{fileName}}, expected {{expectedFileType}}",
   "SUBGRAPH_DIALOG": {
     "SUBGRAPH_SIZE": "Subgraph size (nodes)",
     "SUBGRAPH_SIZE_TOOLTIP": "Number of nodes to display at a time",
@@ -81,15 +85,9 @@
     "VIEW_FULL_GRAPH": "View full graph",
     "VIEW_SUBGRAPH": "View subgraph"
   },
+  "UNMATCHED_FILE_TYPE": "Invalid file type for {{fileName}}, expected {{expectedFileType}}",
   "VISUALIZE": {
     "TITLE": "Visualize",
     "VISUALIZE_MODEL": "Visualize an AI Model"
-  },
-  "PANELS": {
-    "GRAPH": "Graph",
-    "OPTRACE": "Optrace",
-    "PERFORMANCE": "Performance",
-    "ACCURACY": "Accuracy",
-    "QHAS": "QNN HTP Analysis"
   }
-}
+}

qairt_visualizer/core/ui/dist/browser/assets/i18n/panels/en.json

@@ -0,0 +1,11 @@
+{
+  "panels.accuracy.info": "Accuracy",
+  "panels.graph.info": "Graphically view model architecture, including connectivity, node, and tensor parameters",
+  "panels.graph.label": "Graph",
+  "panels.optrace.info": "Opstats data for a model, which helps get insight on the performance bottlenecks in the AI model execution",
+  "panels.optrace.label": "Optrace",
+  "panels.performance.info": "Profiling details for a model run on a given Qualcomm AI Stack runtime",
+  "panels.performance.label": "Performance",
+  "panels.qhas.info": "QNN HTP Analysis Summary (QHAS) reports with summarized data and charts from running models with HTP runtime",
+  "panels.qhas.label": "QNN HTP Analysis"
+}

qairt_visualizer/core/ui/dist/browser/assets/i18n/panels/qhas/en.json

@@ -0,0 +1,5 @@
+{
+  "QHAS_ERROR": {
+    "UNABLE_TO_LOAD": "Unable to load QHAS report"
+  }
+}

qairt_visualizer/core/ui/dist/browser/assets/i18n/workspaces/en.json

@@ -0,0 +1,7 @@
+{
+  "workspaces.error.couldNotOpen": "Unable to open workspace due to an unknown error.",
+  "workspaces.error.couldNotUndock": "Unable to undock workspace due to an unknown error.",
+  "workspaces.menu.rename.label": "Rename Workspace",
+  "workspaces.menu.undock.label": "Move To New Window",
+  "workspaces.name.default": "Untitled"
+}

qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/CONTRIBUTING.md

@@ -4,7 +4,7 @@
 
 Netron can run as both an [Electron](https://electronjs.org) app or a Python web server.
 
-To start the Electron app, install [Node.js](https://nodejs.org) and run: 
+To start the Electron app, install [Node.js](https://nodejs.org) and run:
 
 ```bash
 npm install

qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/README.md

@@ -3,11 +3,11 @@
 <img width="400px" height="100px" src="https://github.com/lutzroeder/netron/raw/main/.github/logo-dark.svg#gh-dark-mode-only">
 </div>
 
-Netron is a viewer for neural network, deep learning and machine learning models. 
+Netron is a viewer for neural network, deep learning and machine learning models.
 
-Netron supports ONNX, TensorFlow Lite, Core ML, Keras, Caffe, Darknet, MXNet, PaddlePaddle, ncnn, MNN and TensorFlow.js.
+Netron supports ONNX, TensorFlow Lite, Core ML, Keras, Caffe, Darknet, PyTorch, TensorFlow.js, Safetensors and NumPy.
 
-Netron has experimental support for PyTorch, TorchScript, TensorFlow, OpenVINO, RKNN, MediaPipe, ML.NET and scikit-learn.
+Netron has experimental support for TorchScript, TensorFlow, MXNet, OpenVINO, RKNN, ML.NET, ncnn, MNN, PaddlePaddle, GGUF and scikit-learn.
 
 <p align='center'><a href='https://www.lutzroeder.com/ai'><img src='.github/screenshot.png' width='800'></a></p>
 

qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/jenkins/Jenkinsfile

@@ -16,16 +16,30 @@ node("QAIRT_TOOLS_CI_AGENTS") {
             git branch: "mainline",
                 url: 'git@github.qualcomm.com:MLG/aisw-ci-tools.git'
         }
+
+        if ( "${env.BRANCH_NAME}".contains("rel/") ) {
+            sh """
+               echo "Release build for ${env.BRANCH_NAME} branch, checking out branch for build..."
+               git checkout "${env.BRANCH_NAME}"
+            """
+        }
+        else if ( "${env.CHANGE_TARGET}" != "null" && "${env.CHANGE_TARGET}".contains("rel/") ) {
+            sh """
+               echo "Release branch PR for ${env.CHANGE_TARGET}, checking out release branch for build..."
+               git checkout "${env.CHANGE_TARGET}"
+            """
+        }
+
         def changeBranch = "${env.CHANGE_BRANCH}"
         if ("${changeBranch}" != "null" && "${changeBranch}" != "") {
-	    sh """
-           if git branch --list | grep -q "${changeBranch}" || git ls-remote --heads origin "${changeBranch}" | grep -q "${changeBranch}"; then
-	         echo "Found ${changeBranch} branch for repo, checking out for build..."
-		     git checkout "${changeBranch}"
-	       else
-	          echo "No PR branch "${changeBranch}" found for repo, proceeding with mainline"
-	       fi
-	    """
+            sh """
+               if git branch --list | grep -q "${changeBranch}" || git ls-remote --heads origin "${changeBranch}" | grep -q "${changeBranch}"; then
+                 echo "Found PR ${changeBranch} branch for repo, checking out for build..."
+                 git checkout "${changeBranch}"
+               else
+                  echo "No PR branch "${changeBranch}" found for repo, proceeding..."
+               fi
+            """
         }
     }
     // pass the required environment variables to the CI pipeline for QAIRT Tools