ttnn-visualizer 0.67.0__py3-none-any.whl → 0.68.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -177,19 +177,22 @@ def open_browser(host, port, instance_id=None):
 
     max_attempts = 10
     attempt = 0
-    server_ready = False
+    app_ready = False
 
-    print(f"Waiting for server to be ready at {url}...")
-    while attempt < max_attempts and not server_ready:
+    print(f"Waiting for application to be ready at {url}...")
+    while attempt < max_attempts and not app_ready:
         try:
             urlopen(url, timeout=1)
-            server_ready = True
+            app_ready = True
         except (URLError, ConnectionError, OSError):
             attempt += 1
+            logger.warning(f"Retrying {url}...")
             time.sleep(0.5)
 
-    if not server_ready:
-        print(f"❌ Server not ready after {max_attempts} attempts.")
+    if not app_ready:
+        print(
+            f"❌ Application not ready after {max_attempts} attempts - is the front end running?"
+        )
     else:
         print(f"Launching browser with url: {url}")
 

ttnn_visualizer/csv_queries.py

@@ -329,7 +329,9 @@ class OpsPerformanceQueries:
             performance_path.glob(f"{OpsPerformanceQueries.PERF_RESULTS_PREFIX}_*.csv")
         )
         if not perf_files:
-            raise FileNotFoundError("No performance results file found.")
+            raise FileNotFoundError(
+                f"No performance results file found at {performance_path}"
+            )
 
         # Use the latest file
         latest_file = max(perf_files, key=os.path.getctime)

ttnn_visualizer/decorators.py

@@ -76,8 +76,8 @@ def remote_exception_handler(func):
 
             # Return user-friendly error message about SSH keys
             user_message = (
-                "SSH authentication failed. This application requires SSH key-based authentication. "
-                "Please ensure your SSH public key is added to the authorized_keys file on the remote server. "
+                "SSH authentication failed. This application requires SSH key-based authentication."
+                " Please ensure your SSH public key is added to the authorized_keys file on the remote server."
                 "Password authentication is not supported."
             )
 
@@ -95,7 +95,7 @@ def remote_exception_handler(func):
             current_app.logger.error(f"No projects: {str(err)}")
             raise RemoteConnectionException(
                 status=ConnectionTestStates.FAILED,
-                message=f"No projects found at remote location: {str(err)}",
+                message=f"No remote projects found: {str(err)}",
             )
         except NoValidConnectionsError as err:
             current_app.logger.warning(
@@ -132,6 +132,15 @@ def remote_exception_handler(func):
             raise RemoteConnectionException(
                 status=ConnectionTestStates.FAILED, message=message
             )
+        except Exception as err:
+            # Catch any other unhandled exceptions
+            current_app.logger.exception(
+                f"Unexpected error during remote operation for {connection}: {str(err)}"
+            )
+            raise RemoteConnectionException(
+                status=ConnectionTestStates.FAILED,
+                message=f"An unexpected error occurred: {str(err)}",
+            )
 
     return remote_handler
 

ttnn_visualizer/instances.py

@@ -19,14 +19,25 @@ logger = getLogger(__name__)
 from flask import current_app, jsonify
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 
+# Active report dictionary keys
+KEY_PROFILER_NAME = "profiler_name"
+KEY_PROFILER_LOCATION = "profiler_location"
+KEY_PERFORMANCE_NAME = "performance_name"
+KEY_PERFORMANCE_LOCATION = "performance_location"
+KEY_NPE_NAME = "npe_name"
+KEY_NPE_LOCATION = "npe_location"
+
 _sentinel = object()
 
 
 def update_existing_instance(
     instance_data,
     profiler_name,
+    profiler_location,
     performance_name,
+    performance_location,
     npe_name,
+    npe_location,
     remote_connection,
     remote_profiler_folder,
     remote_performance_folder,
@@ -39,19 +50,24 @@ def update_existing_instance(
 
     # First ifs are explicit deletes and elifs are updates
     if profiler_name == "":
-        active_report.pop("profiler_name", None)
-    elif profiler_name is not None:
-        active_report["profiler_name"] = profiler_name
+        active_report.pop(KEY_PROFILER_NAME, None)
+        active_report.pop(KEY_PROFILER_LOCATION, None)
+    elif profiler_name is not None and profiler_location is not None:
+        active_report[KEY_PROFILER_NAME] = profiler_name
+        active_report[KEY_PROFILER_LOCATION] = profiler_location
 
     if performance_name == "":
-        active_report.pop("performance_name", None)
-    elif performance_name is not None:
-
-        active_report["performance_name"] = performance_name
+        active_report.pop(KEY_PERFORMANCE_NAME, None)
+        active_report.pop(KEY_PERFORMANCE_LOCATION, None)
+    elif performance_name is not None and performance_location is not None:
+        active_report[KEY_PERFORMANCE_NAME] = performance_name
+        active_report[KEY_PERFORMANCE_LOCATION] = performance_location
     if npe_name == "":
-        active_report.pop("npe_name", None)
-    elif npe_name is not None:
-        active_report["npe_name"] = npe_name
+        active_report.pop(KEY_NPE_NAME, None)
+        active_report.pop(KEY_NPE_LOCATION, None)
+    elif npe_name is not None and npe_location is not None:
+        active_report[KEY_NPE_NAME] = npe_name
+        active_report[KEY_NPE_LOCATION] = npe_location
 
     instance_data.active_report = active_report
 
@@ -68,9 +84,9 @@ def update_existing_instance(
     if profiler_path is not _sentinel:
         instance_data.profiler_path = profiler_path
     else:
-        if active_report.get("profiler_name"):
+        if active_report.get(KEY_PROFILER_NAME):
             instance_data.profiler_path = get_profiler_path(
-                profiler_name=active_report["profiler_name"],
+                profiler_name=active_report[KEY_PROFILER_NAME],
                 current_app=current_app,
                 remote_connection=remote_connection,
             )
@@ -78,9 +94,9 @@ def update_existing_instance(
     if performance_path is not _sentinel:
         instance_data.performance_path = performance_path
     else:
-        if active_report.get("performance_name"):
+        if active_report.get(KEY_PERFORMANCE_NAME):
             instance_data.performance_path = get_performance_path(
-                performance_name=active_report["performance_name"],
+                performance_name=active_report[KEY_PERFORMANCE_NAME],
                 current_app=current_app,
                 remote_connection=remote_connection,
             )
@@ -88,9 +104,9 @@ def update_existing_instance(
     if npe_path is not _sentinel:
         instance_data.npe_path = npe_path
     else:
-        if active_report.get("npe_name"):
+        if active_report.get(KEY_NPE_NAME):
             instance_data.npe_path = get_npe_path(
-                npe_name=active_report["npe_name"], current_app=current_app
+                npe_name=active_report[KEY_NPE_NAME], current_app=current_app
             )
 
 
@@ -117,8 +133,11 @@ def commit_and_log_session(instance_data, instance_id):
 def create_new_instance(
     instance_id,
     profiler_name,
+    profiler_location,
     performance_name,
+    performance_location,
     npe_name,
+    npe_location,
     remote_connection,
     remote_profiler_folder,
     remote_performance_folder,
@@ -128,12 +147,18 @@ def create_new_instance(
     npe_path=_sentinel,
 ):
     active_report = {}
-    if profiler_name:
-        active_report["profiler_name"] = profiler_name
-    if performance_name:
-        active_report["performance_name"] = performance_name
-    if npe_name:
-        active_report["npe_name"] = npe_name
+
+    if profiler_name and profiler_location is not None:
+        active_report[KEY_PROFILER_NAME] = profiler_name
+        active_report[KEY_PROFILER_LOCATION] = profiler_location
+
+    if performance_name and performance_location is not None:
+        active_report[KEY_PERFORMANCE_NAME] = performance_name
+        active_report[KEY_PERFORMANCE_LOCATION] = performance_location
+
+    if npe_name and npe_location is not None:
+        active_report[KEY_NPE_NAME] = npe_name
+        active_report[KEY_NPE_LOCATION] = npe_location
 
     if clear_remote:
         remote_connection = None
@@ -147,7 +172,7 @@ def create_new_instance(
             profiler_path
             if profiler_path is not _sentinel
             else get_profiler_path(
-                active_report["profiler_name"],
+                active_report[KEY_PROFILER_NAME],
                 current_app=current_app,
                 remote_connection=remote_connection,
             )
@@ -178,8 +203,11 @@ def create_new_instance(
 def update_instance(
     instance_id,
     profiler_name=None,
+    profiler_location=None,
     performance_name=None,
+    performance_location=None,
     npe_name=None,
+    npe_location=None,
     remote_connection=None,
     remote_profiler_folder=None,
     remote_performance_folder=None,
@@ -195,8 +223,11 @@ def update_instance(
             update_existing_instance(
                 instance_data,
                 profiler_name,
+                profiler_location,
                 performance_name,
+                performance_location,
                 npe_name,
+                npe_location,
                 remote_connection,
                 remote_profiler_folder,
                 remote_performance_folder,
@@ -209,8 +240,11 @@ def update_instance(
             instance_data = create_new_instance(
                 instance_id,
                 profiler_name,
+                profiler_location,
                 performance_name,
+                performance_location,
                 npe_name,
+                npe_location,
                 remote_connection,
                 remote_profiler_folder,
                 remote_performance_folder,
@@ -231,8 +265,11 @@ def update_instance(
 def get_or_create_instance(
     instance_id,
     profiler_name=None,
+    profiler_location=None,
     performance_name=None,
+    performance_location=None,
     npe_name=None,
+    npe_location=None,
     remote_connection=None,
     remote_profiler_folder=None,
 ):
@@ -265,16 +302,22 @@ def get_or_create_instance(
         # Update the instance if any new data is provided
         if (
             profiler_name
+            or profiler_location is not None
             or performance_name
+            or performance_location is not None
             or npe_name
+            or npe_location is not None
             or remote_connection
             or remote_profiler_folder
         ):
             update_instance(
                 instance_id=instance_id,
                 profiler_name=profiler_name,
+                profiler_location=profiler_location,
                 performance_name=performance_name,
+                performance_location=performance_location,
                 npe_name=npe_name,
+                npe_location=npe_location,
                 remote_connection=remote_connection,
                 remote_profiler_folder=remote_profiler_folder,
             )
@@ -359,9 +402,9 @@ def create_instance_from_local_paths(profiler_path, performance_path):
     instance_data = InstanceTable(
         instance_id=create_random_instance_id(),
         active_report={
-            "profiler_name": profiler_name,
-            "performance_name": performance_name,
-            "npe_name": None,
+            KEY_PROFILER_NAME: profiler_name,
+            KEY_PERFORMANCE_NAME: performance_name,
+            KEY_NPE_NAME: None,
         },
         profiler_path=f"{_profiler_path}/db.sqlite" if profiler_path else None,
         performance_path=performance_path if performance_path else None,

ttnn_visualizer/models.py

@@ -24,6 +24,11 @@ class BufferType(enum.Enum):
     TRACE = 4
 
 
+class ReportLocation(enum.Enum):
+    LOCAL = "local"
+    REMOTE = "remote"
+
+
 @dataclasses.dataclass
 class Operation(SerializeableDataclass):
     operation_id: int
@@ -200,8 +205,11 @@ class StatusMessage(SerializeableModel):
 
 class ActiveReports(SerializeableModel):
     profiler_name: Optional[str] = None
+    profiler_location: Optional[ReportLocation] = None
     performance_name: Optional[str] = None
+    performance_location: Optional[ReportLocation] = None
     npe_name: Optional[str] = None
+    npe_location: Optional[ReportLocation] = None
 
 
 class RemoteReportFolder(SerializeableModel):

ttnn_visualizer/sftp_operations.py

@@ -620,13 +620,21 @@ def read_remote_file(
 
 @remote_exception_handler
 def check_remote_path_for_reports(remote_connection):
-    remote_profiler_paths = find_folders_by_files(
-        remote_connection, remote_connection.profilerPath, [TEST_CONFIG_FILE]
-    )
+    remote_profiler_paths = []
+    if remote_connection.profilerPath:
+        remote_profiler_paths = find_folders_by_files(
+            remote_connection, remote_connection.profilerPath, [TEST_CONFIG_FILE]
+        )
+    else:
+        logger.info("No profiler path configured; skipping check")
 
-    remote_performance_paths = find_folders_by_files(
-        remote_connection, remote_connection.performancePath, [TEST_PROFILER_FILE]
-    )
+    remote_performance_paths = []
+    if remote_connection.performancePath:
+        remote_performance_paths = find_folders_by_files(
+            remote_connection, remote_connection.performancePath, [TEST_PROFILER_FILE]
+        )
+    else:
+        logger.info("No performance path configured; skipping check")
 
     errors = []
     if not remote_profiler_paths and remote_connection.profilerPath:
@@ -682,6 +690,9 @@ def find_folders_by_files(
     remote_connection: RemoteConnection, root_folder: str, file_names: List[str]
 ) -> List[str]:
     """Given a remote path, return a list of top-level folders that contain any of the specified files."""
+    if not root_folder:
+        return []
+
     matched_folders: List[str] = []
 
     # Build SSH command to find directories in root_folder
@@ -746,8 +757,23 @@ def find_folders_by_files(
             # This line should never be reached as handle_ssh_subprocess_error raises an exception
             return []
         else:
-            logger.error(f"Error finding folders: {e.stderr}")
-            return []
+            stderr = e.stderr.lower() if e.stderr else ""
+            # Check for permission denied errors
+            if "permission denied" in stderr:
+                error_msg = (
+                    f"Permission denied accessing '{root_folder}'. "
+                    f"The user '{remote_connection.username}' does not have read access to this directory. "
+                    "Please check directory permissions on the remote server or choose a different path."
+                )
+                logger.error(f"Error finding folders: {e.stderr}")
+                raise RemoteConnectionException(
+                    message=error_msg,
+                    status=ConnectionTestStates.FAILED,
+                    detail=e.stderr.strip() if e.stderr else None,
+                )
+            else:
+                logger.error(f"Error finding folders: {e.stderr}")
+                return []
     except subprocess.TimeoutExpired:
         logger.error(f"Timeout finding folders in: {root_folder}")
         return []
@@ -761,23 +787,28 @@ def get_remote_performance_folders(
     remote_connection: RemoteConnection,
 ) -> List[RemoteReportFolder]:
     """Return a list of remote folders containing a profile_log_device file."""
-    if remote_connection.performancePath is None:
+    performance_paths = []
+
+    if remote_connection.performancePath:
+        performance_paths = find_folders_by_files(
+            remote_connection, remote_connection.performancePath, [TEST_PROFILER_FILE]
+        )
+    else:
         error = "Performance path is not configured for this connection"
         logger.error(error)
         raise NoProjectsException(status=ConnectionTestStates.FAILED, message=error)
 
-    performance_paths = find_folders_by_files(
-        remote_connection, remote_connection.performancePath, [TEST_PROFILER_FILE]
-    )
     if not performance_paths:
-        error = f"No profiler paths found at {remote_connection.performancePath}"
+        error = f"Performance path: {remote_connection.performancePath}"
         logger.info(error)
         raise NoProjectsException(status=ConnectionTestStates.FAILED, message=error)
+
     remote_folder_data = []
     for path in performance_paths:
         remote_folder_data.append(
             get_remote_performance_folder(remote_connection, path)
         )
+
     return remote_folder_data
 
 
@@ -786,19 +817,29 @@ def get_remote_profiler_folders(
     remote_connection: RemoteConnection,
 ) -> List[RemoteReportFolder]:
     """Return a list of remote folders containing a config.json file."""
-    remote_config_paths = find_folders_by_files(
-        remote_connection, remote_connection.profilerPath, [TEST_CONFIG_FILE]
-    )
-    if not remote_config_paths:
-        error = f"No projects found at {remote_connection.profilerPath}"
+    profiler_paths = []
+
+    if remote_connection.profilerPath:
+        profiler_paths = find_folders_by_files(
+            remote_connection, remote_connection.profilerPath, [TEST_CONFIG_FILE]
+        )
+    else:
+        error = f"No profiler reports found at {remote_connection.profilerPath}"
+        logger.info(error)
+        raise NoProjectsException(status=ConnectionTestStates.FAILED, message=error)
+
+    if not profiler_paths:
+        error = f"Profiler path: {remote_connection.profilerPath}"
         logger.info(error)
         raise NoProjectsException(status=ConnectionTestStates.FAILED, message=error)
+
     remote_folder_data = []
-    for config_path in remote_config_paths:
+    for path in profiler_paths:
         remote_folder = get_remote_profiler_folder_from_config_path(
-            remote_connection, str(Path(config_path).joinpath(TEST_CONFIG_FILE))
+            remote_connection, str(Path(path).joinpath(TEST_CONFIG_FILE))
         )
         remote_folder_data.append(remote_folder)
+
     return remote_folder_data
 
 

ttnn_visualizer/static/assets/{allPaths-DiBJY9fQ.js → allPaths-BZLVCrxa.js}

@@ -1 +1 @@
-import{I as s}from"./index-voJy5fZe.js";import{I as r}from"./index-BZITDwoa.js";import{p as n,I as c}from"./index-BO1FGYnE.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};
+import{I as s}from"./index-voJy5fZe.js";import{I as r}from"./index-BZITDwoa.js";import{p as n,I as c}from"./index-ryWOUYjC.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};

ttnn_visualizer/static/assets/allPathsLoader-7-MAALae.js

@@ -0,0 +1,2 @@
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-BZLVCrxa.js","assets/index-voJy5fZe.js","assets/index-BZITDwoa.js","assets/index-ryWOUYjC.js","assets/index-DS75jGv-.css"])))=>i.map(i=>d[i]);
+import{_ as e}from"./index-ryWOUYjC.js";const s=async(t,a)=>{const{getIconPaths:o}=await e(async()=>{const{getIconPaths:r}=await import("./allPaths-BZLVCrxa.js");return{getIconPaths:r}},__vite__mapDeps([0,1,2,3,4]));return o(t,a)};export{s as allPathsLoader};