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

ttnn_visualizer/__init__.py

@@ -1,4 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 

ttnn_visualizer/app.py

@@ -1,8 +1,9 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import argparse
+import json
 import logging
 import os
 import subprocess
@@ -55,16 +56,29 @@ def create_app(settings_override=None):
 
     middleware(app)
 
-    app.register_blueprint(api)
+    app.register_blueprint(api, url_prefix=f"{app.config['BASE_PATH']}/api")
 
     extensions(app)
 
     if flask_env == "production":
-
-        @app.route("/", defaults={"path": ""})
+        @app.route(f"{app.config['BASE_PATH']}/", defaults={"path": ""})
         @app.route("/<path:path>")
         def catch_all(path):
-            return app.send_static_file("index.html")
+            js_config = {
+                "SERVER_MODE": app.config["SERVER_MODE"],
+                "BASE_PATH": app.config["BASE_PATH"],
+            }
+            js = f"window.TTNN_VISUALIZER_CONFIG = {json.dumps(js_config)};"
+
+            with open(os.path.join(app.static_folder, "index.html")) as f:
+                html = f.read()
+
+                html_with_config = html.replace(
+                    "/* SERVER_CONFIG */",
+                    js,
+                )
+
+            return flask.Response(html_with_config, mimetype="text/html")
 
     return app
 

ttnn_visualizer/csv_queries.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc.
+# SPDX-FileCopyrightText: © 2025 Tenstorrent Inc.
 import csv
 import os
 import tempfile

ttnn_visualizer/decorators.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import re
 from ttnn_visualizer.enums import ConnectionTestStates
@@ -42,7 +42,8 @@ def with_instance(func):
             session['instances'] = []
 
         if instance.instance_id not in session['instances']:
-            session['instances'].append(instance.instance_id)
+            session['instances'] = session.get('instances', []) + [instance.instance_id]
+
 
         return func(*args, **kwargs)
 
@@ -115,3 +116,16 @@ def remote_exception_handler(func):
             )
 
     return remote_handler
+
+
+def local_only(f):
+    from flask import current_app
+
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        if current_app.config["SERVER_MODE"]:
+            abort(403, description="Endpoint not accessible")
+
+        return f(*args, **kwargs)
+
+    return decorated_function

ttnn_visualizer/enums.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import enum
 

ttnn_visualizer/exceptions.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 from http import HTTPStatus
 

ttnn_visualizer/extensions.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 from flask_socketio import SocketIO
 from flask_static_digest import FlaskStaticDigest

ttnn_visualizer/file_uploads.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc.
+# SPDX-FileCopyrightText: © 2025 Tenstorrent Inc.
 
 import logging
 import os

ttnn_visualizer/instances.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import json
 import random

ttnn_visualizer/models.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import dataclasses
 import enum

ttnn_visualizer/queries.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import json
 from typing import Generator, Dict, Any, Union

ttnn_visualizer/remote_sqlite_setup.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import re
 

ttnn_visualizer/serializers.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import dataclasses
 from collections import defaultdict

ttnn_visualizer/settings.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import os
 from pathlib import Path
@@ -22,7 +22,7 @@ class DefaultConfig(object):
         o for o in os.getenv("ALLOWED_ORIGINS", "http://localhost:5173,http://localhost:8000").split(",")
         if o
     ]
-    API_PATH = os.getenv("API_PATH", "/")
+    BASE_PATH = os.getenv("BASE_PATH", "/")
 
     # Path Settings
     DB_VERSION = "0.29.0"  # App version when DB schema last changed

ttnn_visualizer/sftp_operations.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import json
 import logging

ttnn_visualizer/sockets.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import threading
 import time

ttnn_visualizer/ssh_client.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import paramiko
 import os

ttnn_visualizer/static/assets/{allPaths-Bl5zsQD8.js → allPaths-BYcRoLEw.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-DCEh1A_0.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-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};

ttnn_visualizer/static/assets/{allPathsLoader-62UMr_jj.js → allPathsLoader-CNKgAYcC.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-Bl5zsQD8.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-DCEh1A_0.js","assets/index-BJweZJCB.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-DCEh1A_0.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-Bl5zsQD8.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-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};