ttnn-visualizer 0.25.1__py3-none-any.whl → 0.26.0__py3-none-any.whl

ttnn_visualizer/file_uploads.py

@@ -43,6 +43,13 @@ def extract_report_name(files):
     unsplit_report_name = str(files[0].filename)
     return unsplit_report_name.split("/")[0]
 
+def extract_npe_name(files):
+    if not files:
+        return None
+
+    file_path = Path(files[0].filename)
+    return file_path.stem
+
 
 def save_uploaded_files(
     files,

ttnn_visualizer/models.py

@@ -182,6 +182,7 @@ class StatusMessage(SerializeableModel):
 class ActiveReport(SerializeableModel):
     report_name: Optional[str] = None
     profile_name: Optional[str] = None
+    npe_name: Optional[str] = None
 
 
 class RemoteReportFolder(SerializeableModel):
@@ -195,6 +196,7 @@ class TabSession(BaseModel):
     tab_id: str
     report_path: Optional[str] = None
     profiler_path: Optional[str] = None
+    npe_path: Optional[str] = None
     active_report: Optional[ActiveReport] = None
     remote_connection: Optional[RemoteConnection] = None
     remote_folder: Optional[RemoteReportFolder] = None
@@ -208,6 +210,7 @@ class TabSessionTable(db.Model):
     tab_id = Column(String, unique=True, nullable=False)
     report_path = Column(String)
     profiler_path = Column(String, nullable=True)
+    npe_path = Column(String, nullable=True)
     active_report = db.Column(MutableDict.as_mutable(JSON), nullable=False, default={})
     remote_connection = Column(JSON, nullable=True)
     remote_folder = Column(JSON, nullable=True)
@@ -219,13 +222,15 @@ class TabSessionTable(db.Model):
         active_report,
         remote_connection=None,
         remote_folder=None,
+        remote_profile_folder=None,
         report_path=None,
         profiler_path=None,
-        remote_profile_folder=None,
+        npe_path=None,
     ):
         self.tab_id = tab_id
         self.active_report = active_report
         self.report_path = report_path
+        self.npe_path = npe_path
         self.remote_connection = remote_connection
         self.remote_folder = remote_folder
         self.profiler_path = profiler_path
@@ -241,6 +246,7 @@ class TabSessionTable(db.Model):
             "remote_profile_folder": self.remote_profile_folder,
             "report_path": self.report_path,
             "profiler_path": self.profiler_path,
+            "npe_path": self.npe_path
         }
 
     def to_pydantic(self) -> TabSession:
@@ -250,6 +256,9 @@ class TabSessionTable(db.Model):
             profiler_path=(
                 str(self.profiler_path) if self.profiler_path is not None else None
             ),
+            npe_path=(
+                str(self.npe_path) if self.npe_path is not None else None
+            ),
             active_report=(
                 (ActiveReport(**self.active_report) if self.active_report else None)
                 if isinstance(self.active_report, dict)

ttnn_visualizer/sessions.py

@@ -6,7 +6,7 @@ from logging import getLogger
 
 from flask import request
 
-from ttnn_visualizer.utils import get_report_path, get_profiler_path
+from ttnn_visualizer.utils import get_report_path, get_profiler_path, get_npe_path
 from ttnn_visualizer.models import (
     TabSessionTable,
 )
@@ -23,6 +23,7 @@ def update_existing_tab_session(
     session_data,
     report_name,
     profile_name,
+    npe_name,
     remote_connection,
     remote_folder,
     remote_profile_folder,
@@ -34,6 +35,8 @@ def update_existing_tab_session(
         active_report["report_name"] = report_name
     if profile_name:
         active_report["profile_name"] = profile_name
+    if npe_name:
+        active_report["npe_name"] = npe_name
 
     session_data.active_report = active_report
 
@@ -48,7 +51,7 @@ def update_existing_tab_session(
         clear_remote_data(session_data)
 
     update_paths(
-        session_data, active_report, remote_connection, report_name, profile_name
+        session_data, active_report, remote_connection
     )
 
 
@@ -73,7 +76,7 @@ def commit_and_log_session(session_data, tab_id):
 
 
 def update_paths(
-    session_data, active_report, remote_connection, report_name, profile_name
+    session_data, active_report, remote_connection
 ):
     if active_report.get("profile_name"):
         session_data.profiler_path = get_profiler_path(
@@ -89,11 +92,18 @@ def update_paths(
             remote_connection=remote_connection,
         )
 
+    if active_report.get("npe_name"):
+        session_data.npe_path = get_npe_path(
+            npe_name=active_report["npe_name"],
+            current_app=current_app
+        )
+
 
 def create_new_tab_session(
     tab_id,
     report_name,
     profile_name,
+    npe_name,
     remote_connection,
     remote_folder,
     remote_profile_folder,
@@ -104,6 +114,8 @@ def create_new_tab_session(
         active_report["report_name"] = report_name
     if profile_name:
         active_report["profile_name"] = profile_name
+    if npe_name:
+        active_report["npe_name"] = npe_name
 
     if clear_remote:
         remote_connection = None
@@ -134,6 +146,7 @@ def update_tab_session(
     tab_id,
     report_name=None,
     profile_name=None,
+    npe_name=None,
     remote_connection=None,
     remote_folder=None,
     remote_profile_folder=None,
@@ -147,6 +160,7 @@ def update_tab_session(
                 session_data,
                 report_name,
                 profile_name,
+                npe_name,
                 remote_connection,
                 remote_folder,
                 remote_profile_folder,
@@ -157,6 +171,7 @@ def update_tab_session(
                 tab_id,
                 report_name,
                 profile_name,
+                npe_name,
                 remote_connection,
                 remote_folder,
                 remote_profile_folder,
@@ -175,6 +190,7 @@ def get_or_create_tab_session(
     tab_id,
     report_name=None,
     profile_name=None,
+    npe_name=None,
     remote_connection=None,
     remote_folder=None,
 ):
@@ -198,11 +214,12 @@ def get_or_create_tab_session(
             db.session.commit()
 
         # Update the session if any new data is provided
-        if report_name or profile_name or remote_connection or remote_folder:
+        if report_name or profile_name or npe_name or remote_connection or remote_folder:
             update_tab_session(
                 tab_id=tab_id,
                 report_name=report_name,
                 profile_name=profile_name,
+                npe_name=npe_name,
                 remote_connection=remote_connection,
                 remote_folder=remote_folder,
             )

ttnn_visualizer/settings.py

@@ -19,7 +19,7 @@ class DefaultConfig(object):
 
     # Path Settings
     REPORT_DATA_DIRECTORY = Path(__file__).parent.absolute().joinpath("data")
-    VERSION = "0.14.2"
+    VERSION = "0.26.0"
     LOCAL_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("local")
     REMOTE_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("remote")
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))

ttnn_visualizer/static/assets/{allPaths-BsZhgQ3T.js → allPaths-egCeGMYd.js}

@@ -1 +1 @@
-import{I as n}from"./index-BVMreIQm.js";import{I as e}from"./index-Do7YB6C4.js";import{p as r,I as s}from"./index-Fd4wdmZt.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};
+import{I as n}from"./index-BVMreIQm.js";import{I as e}from"./index-Do7YB6C4.js";import{p as r,I as s}from"./index-rIoXC5rS.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};

ttnn_visualizer/static/assets/{allPathsLoader-DgFQ6cmn.js → allPathsLoader-ChZ_kotC.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-BsZhgQ3T.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-Fd4wdmZt.js","assets/index-C6f3UdbP.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-Fd4wdmZt.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-BsZhgQ3T.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-egCeGMYd.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-rIoXC5rS.js","assets/index-S-_ELv4m.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-rIoXC5rS.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-egCeGMYd.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};