ttnn-visualizer 0.39.0__py3-none-any.whl → 0.40.1__py3-none-any.whl

ttnn_visualizer/app.py

@@ -99,9 +99,6 @@ def extensions(app: flask.Flask):
         socketio.init_app(app)
     db.init_app(app)
 
-    app.config["SESSION_TYPE"] = "sqlalchemy"
-    app.config["SESSION_SQLALCHEMY"] = db
-
     if app.config["USE_WEBSOCKETS"]:
         register_handlers(socketio)
 

ttnn_visualizer/decorators.py

@@ -2,6 +2,7 @@
 #
 # SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
+import logging
 import re
 from ttnn_visualizer.enums import ConnectionTestStates
 
@@ -21,6 +22,8 @@ from ttnn_visualizer.exceptions import (
 )
 from ttnn_visualizer.instances import get_or_create_instance
 
+logger = logging.getLogger(__name__)
+
 
 def with_instance(func):
     @wraps(func)
@@ -44,7 +47,6 @@ def with_instance(func):
         if instance.instance_id not in session['instances']:
             session['instances'] = session.get('instances', []) + [instance.instance_id]
 
-
         return func(*args, **kwargs)
 
     return wrapper

ttnn_visualizer/file_uploads.py

@@ -9,6 +9,7 @@ import shlex
 import shutil
 import subprocess
 import tempfile
+import time
 from pathlib import Path
 
 from flask import current_app
@@ -69,9 +70,11 @@ def save_uploaded_files(
     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 folder_name: The name to use for the directory.
+    :param base_directory: The base directory for saving the files.
+    :param parent_folder_name: The name to use for the directory.
     """
+    saved_paths = []
+
     if current_app.config["MALWARE_SCANNER"]:
         scanned_files = scan_uploaded_files(files, base_directory, parent_folder_name)
 
@@ -82,6 +85,7 @@ def save_uploaded_files(
             logger.info(f"Saving uploaded file (clean): {dest_path}")
 
             shutil.move(temp_path, dest_path)
+            saved_paths.append(dest_path)
     else:
         for file in files:
             dest_path = construct_dest_path(file, base_directory, parent_folder_name)
@@ -96,18 +100,23 @@ def save_uploaded_files(
 
             logger.info(f"Saving uploaded file: {dest_path}")
             file.save(dest_path)
+            saved_paths.append(dest_path)
 
-            # Update the modified time of the parent directory (for sorting purposes)
-            os.utime(dest_path.parent, None)
+    for saved_path in saved_paths:
+        # Update the modified time of the parent directory (for sorting purposes)
+        os.utime(saved_path.parent, None)
 
     # Update the modified time of the uploaded directory
     if parent_folder_name:
         uploaded_dir = Path(base_directory) / parent_folder_name
     else:
         uploaded_dir = Path(base_directory)
+
     if uploaded_dir.exists():
         os.utime(uploaded_dir, None)
 
+    return saved_paths
+
 
 def scan_uploaded_files(
     files,
@@ -146,9 +155,13 @@ def scan_uploaded_files(
 
 
 def construct_dest_path(file, target_directory, folder_name):
+    prefix = f"{int(time.time())}_" if current_app.config["SERVER_MODE"] else ""
+
     if folder_name:
-        dest_path = Path(target_directory) / folder_name / str(file.filename)
+        prefixed_folder_name = f"{prefix}{folder_name}"
+        dest_path = Path(target_directory) / prefixed_folder_name / str(file.filename)
     else:
-        dest_path = Path(target_directory) / str(file.filename)
+        prefixed_filename = f"{prefix}{file.filename}"
+        dest_path = Path(target_directory) / prefixed_filename
 
     return dest_path

ttnn_visualizer/instances.py

@@ -23,6 +23,9 @@ from flask import jsonify, current_app
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 
 
+_sentinel = object()
+
+
 def update_existing_instance(
     instance_data,
     profiler_name,
@@ -32,6 +35,9 @@ def update_existing_instance(
     remote_profiler_folder,
     remote_performance_folder,
     clear_remote,
+    profiler_path=_sentinel,
+    performance_path=_sentinel,
+    npe_path=_sentinel,
 ):
     active_report = instance_data.active_report or {}
 
@@ -63,9 +69,34 @@ def update_existing_instance(
     if clear_remote:
         clear_remote_data(instance_data)
 
-    update_paths(
-        instance_data, active_report, remote_connection
-    )
+    if profiler_path is not _sentinel:
+        instance_data.profiler_path = profiler_path
+    else:
+        if active_report.get("profiler_name"):
+            instance_data.profiler_path = get_profiler_path(
+                profiler_name=active_report["profiler_name"],
+                current_app=current_app,
+                remote_connection=remote_connection,
+            )
+
+    if performance_path is not _sentinel:
+        instance_data.performance_path = performance_path
+    else:
+        if active_report.get("performance_name"):
+            instance_data.performance_path = get_performance_path(
+                performance_name=active_report["performance_name"],
+                current_app=current_app,
+                remote_connection=remote_connection,
+            )
+
+    if npe_path is not _sentinel:
+        instance_data.npe_path = npe_path
+    else:
+        if active_report.get("npe_name"):
+            instance_data.npe_path = get_npe_path(
+                npe_name=active_report["npe_name"],
+                current_app=current_app
+            )
 
 
 def clear_remote_data(instance_data):
@@ -88,30 +119,6 @@ def commit_and_log_session(instance_data, instance_id):
     )
 
 
-def update_paths(
-    instance_data, active_report, remote_connection
-):
-    if active_report.get("performance_name"):
-        instance_data.performance_path = get_performance_path(
-            performance_name=active_report["performance_name"],
-            current_app=current_app,
-            remote_connection=remote_connection,
-        )
-
-    if active_report.get("profiler_name"):
-        instance_data.profiler_path = get_profiler_path(
-            profiler_name=active_report["profiler_name"],
-            current_app=current_app,
-            remote_connection=remote_connection,
-        )
-
-    if active_report.get("npe_name"):
-        instance_data.npe_path = get_npe_path(
-            npe_name=active_report["npe_name"],
-            current_app=current_app
-        )
-
-
 def create_new_instance(
     instance_id,
     profiler_name,
@@ -121,6 +128,9 @@ def create_new_instance(
     remote_profiler_folder,
     remote_performance_folder,
     clear_remote,
+    profiler_path=_sentinel,
+    performance_path=_sentinel,
+    npe_path=_sentinel,
 ):
     active_report = {}
     if profiler_name:
@@ -138,7 +148,7 @@ def create_new_instance(
     instance_data = InstanceTable(
         instance_id=instance_id,
         active_report=active_report,
-        profiler_path=get_profiler_path(
+        profiler_path=profiler_path if profiler_path is not _sentinel else get_profiler_path(
             active_report["profiler_name"],
             current_app=current_app,
             remote_connection=remote_connection,
@@ -151,6 +161,13 @@ def create_new_instance(
             remote_performance_folder.model_dump() if remote_performance_folder else None
         ),
     )
+
+    if performance_path is not _sentinel:
+        instance_data.performance_path = performance_path
+
+    if npe_path is not _sentinel:
+        instance_data.npe_path = npe_path
+
     db.session.add(instance_data)
     return instance_data
 
@@ -164,6 +181,9 @@ def update_instance(
     remote_profiler_folder=None,
     remote_performance_folder=None,
     clear_remote=False,
+    profiler_path=_sentinel,
+    performance_path=_sentinel,
+    npe_path=_sentinel,
 ):
     try:
         instance_data = get_or_create_instance(instance_id)
@@ -178,6 +198,9 @@ def update_instance(
                 remote_profiler_folder,
                 remote_performance_folder,
                 clear_remote,
+                profiler_path,
+                performance_path,
+                npe_path,
             )
         else:
             instance_data = create_new_instance(
@@ -189,6 +212,9 @@ def update_instance(
                 remote_profiler_folder,
                 remote_performance_folder,
                 clear_remote,
+                profiler_path,
+                performance_path,
+                npe_path,
             )
 
         commit_and_log_session(instance_data, instance_id)

ttnn_visualizer/requirements.txt

@@ -11,7 +11,6 @@ python-dotenv==1.0.1
 flask-sqlalchemy
 flask-socketio
 gevent==24.10.2
-flask-session
 pandas==2.2.3
 wheel
 build

ttnn_visualizer/settings.py

@@ -23,6 +23,9 @@ class DefaultConfig(object):
         if o
     ]
     BASE_PATH = os.getenv("BASE_PATH", "/")
+    MAX_CONTENT_LENGTH = (
+        None if not (v := os.getenv("MAX_CONTENT_LENGTH")) else int(v)
+    )
 
     # Path Settings
     DB_VERSION = "0.29.0"  # App version when DB schema last changed

ttnn_visualizer/static/assets/{allPaths-BYcRoLEw.js → allPaths-DSA-bqy2.js}

@@ -1 +1 @@
-import{I as n}from"./index-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-Ba6DlrXF.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-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-C3j0nlYa.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-CNKgAYcC.js → allPathsLoader-oycPcHBu.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-BYcRoLEw.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-Ba6DlrXF.js","assets/index-B9wn2kZo.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-Ba6DlrXF.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-BYcRoLEw.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-DSA-bqy2.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-C3j0nlYa.js","assets/index-C1rJBrMl.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-C3j0nlYa.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-DSA-bqy2.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};