ttnn-visualizer 0.33.2__py3-none-any.whl → 0.34.0__py3-none-any.whl

ttnn_visualizer/file_uploads.py

@@ -4,11 +4,12 @@
 
 from pathlib import Path
 import logging
+import re
 
 logger = logging.getLogger(__name__)
 
 
-def validate_files(files, required_files, pattern=None):
+def validate_files(files, required_files, pattern=None, folder_name=None):
     """Validate uploaded files against required file names and an optional pattern."""
     found_files = set()
 
@@ -19,7 +20,7 @@ def validate_files(files, required_files, pattern=None):
             pattern and file_path.name.startswith(pattern)
         ):
             found_files.add(file_path.name)
-            if len(file_path.parents) != 2:
+            if not folder_name and len(file_path.parents) != 2:
                 logger.warning(
                     f"File {file.filename} is not under a single parent folder."
                 )
@@ -43,25 +44,25 @@ def extract_profiler_name(files):
     unsplit_profiler_name = str(files[0].filename)
     return unsplit_profiler_name.split("/")[0]
 
+
 def extract_npe_name(files):
     if not files:
         return None
 
-    file_path = Path(files[0].filename)
-    return file_path.stem
+    return re.sub(r"\.(json|npeviz\.zst)$", "", files[0].filename)
 
 
 def save_uploaded_files(
     files,
     target_directory,
-    profiler_name=None,
+    folder_name=None,
 ):
     """
     Save uploaded files to the target directory.
 
     :param files: List of files to be saved.
     :param target_directory: The base directory for saving the files.
-    :param profiler_name: The report name to use for the directory.
+    :param folder_name: The name to use for the directory.
     """
     for file in files:
         current_file_name = str(file.filename)
@@ -69,7 +70,10 @@ def save_uploaded_files(
 
         file_path = Path(current_file_name)
 
-        destination_file = Path(target_directory).joinpath(str(file_path))
+        if folder_name:
+            destination_file = Path(target_directory) / folder_name / str(file_path)
+        else:
+            destination_file = Path(target_directory) / str(file_path)
 
         logger.info(f"Writing file to {destination_file}")
 

ttnn_visualizer/requirements.txt

@@ -18,6 +18,7 @@ build
 PyYAML==6.0.2
 python-dotenv==1.0.1
 tt-perf-report==1.0.6
+zstd==1.5.7.0
 
 # Dev dependencies
 mypy

ttnn_visualizer/sessions.py

@@ -276,7 +276,7 @@ def init_sessions(app):
 
 
 def create_random_instance_id():
-    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=45))
+    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=10))
 
 
 def create_instance_from_local_paths(profiler_path, performance_path):

ttnn_visualizer/settings.py

@@ -16,18 +16,16 @@ class DefaultConfig(object):
     DEBUG = bool(str_to_bool(os.getenv("FLASK_DEBUG", "false")))
     TESTING = False
     PRINT_ENV = True
-    SERVER_MODE = str_to_bool(os.getenv("SERVER_MODE", "false"))
 
     # Path Settings
     DB_VERSION = "0.29.0"  # App version when DB schema last changed
-    REPORT_DATA_DIRECTORY = os.getenv("REPORT_DATA_DIRECTORY", Path(__file__).parent.absolute().joinpath("data"))
+    REPORT_DATA_DIRECTORY = Path(__file__).parent.absolute().joinpath("data")
     LOCAL_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("local")
     REMOTE_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("remote")
     PROFILER_DIRECTORY_NAME = "profiler-reports"
     PERFORMANCE_DIRECTORY_NAME = "performance-reports"
     NPE_DIRECTORY_NAME = "npe-reports"
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))
-    APP_DATA_DIRECTORY = os.path.join("APP_DATA_DIRECTORY", APPLICATION_DIR)
     STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
     SEND_FILE_MAX_AGE_DEFAULT = 0
 
@@ -42,7 +40,7 @@ class DefaultConfig(object):
 
     # SQL Alchemy Settings
     SQLALCHEMY_DATABASE_URI = (
-        f"sqlite:///{os.path.join(APP_DATA_DIRECTORY, f'ttnn_{DB_VERSION}.db')}"
+        f"sqlite:///{os.path.join(APPLICATION_DIR, f'ttnn_{DB_VERSION}.db')}"
     )
     SQLALCHEMY_ENGINE_OPTIONS = {
         "pool_size": 10,  # Adjust pool size as needed (default is 5)

ttnn_visualizer/static/assets/{allPaths-C1NlZ4tc.js → allPaths-D0uOIsnD.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-C2xbrqJV.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-CHS7htir.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-B5Ah5bqG.js → allPathsLoader-B1iveKGB.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-C1NlZ4tc.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-C2xbrqJV.js","assets/index-Dhl0up3F.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-C2xbrqJV.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-C1NlZ4tc.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-D0uOIsnD.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-CHS7htir.js","assets/index-CJJgzCye.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-CHS7htir.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-D0uOIsnD.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};