ttnn-visualizer 0.62.0__py3-none-any.whl → 0.63.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -25,7 +25,7 @@ from ttnn_visualizer.exceptions import (
 )
 from ttnn_visualizer.instances import create_instance_from_local_paths
 from ttnn_visualizer.settings import Config, DefaultConfig
-from ttnn_visualizer.utils import create_path_resolver
+from ttnn_visualizer.utils import find_gunicorn_path
 from werkzeug.debug import DebuggedApplication
 from werkzeug.middleware.proxy_fix import ProxyFix
 
@@ -115,7 +115,6 @@ def extensions(app: flask.Flask):
     :param app: Flask application instance
     :return: None
     """
-
     flask_static_digest.init_app(app)
     if app.config["USE_WEBSOCKETS"]:
         socketio.init_app(app)
@@ -124,14 +123,9 @@ def extensions(app: flask.Flask):
     if app.config["USE_WEBSOCKETS"]:
         register_handlers(socketio)
 
-    # Create the tables within the application context
     with app.app_context():
         db.create_all()
 
-    # For automatically reflecting table data
-    # with app.app_context():
-    #    db.reflect()
-
     return None
 
 
@@ -198,63 +192,91 @@ def parse_args():
     parser.add_argument(
         "--tt-metal-home", help="Specify a TT-Metal home path", default=None
     )
+    parser.add_argument(
+        "-d",
+        "--daemon",
+        action="store_true",
+        help="Run the server as a daemon process",
+    )
     return parser.parse_args()
 
 
+def display_mode_info_without_db(config):
+    """Display mode information using only config, without initializing database."""
+    # Determine if we're in TT-Metal mode
+    tt_metal_home = config.TT_METAL_HOME
+    is_tt_metal_mode = tt_metal_home is not None
+
+    if is_tt_metal_mode:
+        print("🚀 TT-METAL MODE: Working directly with tt-metal generated directory")
+        print(f"   TT_METAL_HOME: {tt_metal_home}")
+
+        profiler_base = Path(tt_metal_home) / "generated" / "ttnn" / "reports"
+        performance_base = Path(tt_metal_home) / "generated" / "profiler" / "reports"
+
+        print(f"   Profiler reports: {profiler_base}")
+        print(f"   Performance reports: {performance_base}")
+
+        # Validate setup
+        if not Path(tt_metal_home).exists():
+            print(
+                f"   ⚠️  Warning: TT_METAL_HOME directory does not exist: {tt_metal_home}"
+            )
+        elif not (Path(tt_metal_home) / "generated").exists():
+            print(f"   ⚠️  Warning: TT-Metal generated directory not found")
+        elif not profiler_base.exists():
+            print(
+                f"   ⚠️  Warning: Profiler reports directory not found: {profiler_base}"
+            )
+        elif not performance_base.exists():
+            print(
+                f"   ⚠️  Warning: Performance reports directory not found: {performance_base}"
+            )
+        else:
+            print(f"   ✓ TT-Metal setup is valid")
+    else:
+        print(
+            "📁 UPLOAD/SYNC MODE: Using local data directory for uploaded/synced reports"
+        )
+        print(f"   Local directory: {config.LOCAL_DATA_DIRECTORY}")
+        print(f"   Remote directory: {config.REMOTE_DATA_DIRECTORY}")
+
+
 def main():
 
     run_command = sys.argv[0].split("/")
     if run_command[-1] == "ttnn-visualizer":
         os.environ.setdefault("FLASK_ENV", "production")
 
-    config = cast(DefaultConfig, Config())
     args = parse_args()
-    instance_id = None
 
-    if args.profiler_path or args.performance_path:
-        app = create_app()
-        app.app_context().push()
-        try:
-            session = create_instance_from_local_paths(
-                profiler_path=args.profiler_path,
-                performance_path=args.performance_path,
-            )
-        except InvalidReportPath:
-            sys.exit("Invalid report path")
-        except InvalidProfilerPath:
-            sys.exit("Invalid profiler path")
-
-        instance_id = session.instance_id
+    config = cast(DefaultConfig, Config())
+    instance_id = None
 
+    # Display mode information first (using config only, no DB needed)
     if args.tt_metal_home:
         config.TT_METAL_HOME = args.tt_metal_home
 
-    # Display mode information
-    app = create_app()
-    with app.app_context():
-        resolver = create_path_resolver(app)
-        mode_info = resolver.get_mode_info()
+    display_mode_info_without_db(config)
 
-        if mode_info["mode"] == "tt_metal":
-            print(
-                "🚀 TT-METAL MODE: Working directly with tt-metal generated directory"
-            )
-            print(f"   TT_METAL_HOME: {mode_info['tt_metal_home']}")
-            print(f"   Profiler reports: {mode_info['profiler_base']}")
-            print(f"   Performance reports: {mode_info['performance_base']}")
-
-            # Validate setup
-            is_valid, message = resolver.validate_tt_metal_setup()
-            if is_valid:
-                print(f"   ✓ {message}")
-            else:
-                print(f"   ⚠️  Warning: {message}")
-        else:
-            print(
-                "📁 UPLOAD/SYNC MODE: Using local data directory for uploaded/synced reports"
-            )
-            print(f"   Local directory: {mode_info['local_dir']}")
-            print(f"   Remote directory: {mode_info['remote_dir']}")
+    # If profiler/performance paths are provided, create an instance
+    # This requires DB access, so we create the app temporarily
+    if args.profiler_path or args.performance_path:
+        app = create_app()
+        with app.app_context():
+            try:
+                session = create_instance_from_local_paths(
+                    profiler_path=args.profiler_path,
+                    performance_path=args.performance_path,
+                )
+                instance_id = session.instance_id
+            except InvalidReportPath:
+                sys.exit("Invalid report path")
+            except InvalidProfilerPath:
+                sys.exit("Invalid profiler path")
+
+        # Clean up this temporary app - workers will create their own
+        del app
 
     # Check if DEBUG environment variable is set
     debug_mode = os.environ.get("DEBUG", "false").lower() == "true"
@@ -263,8 +285,19 @@ def main():
         for key, value in config.to_dict().items():
             print(f"{key}={value}")
 
+    # Warn if there's a gunicorn config file in current directory
+    if Path("gunicorn.conf.py").exists():
+        logger.warning(
+            "Found gunicorn.conf.py in current directory - this may override environment settings"
+        )
+
+    gunicorn_cmd, gunicorn_warning = find_gunicorn_path()
+
+    if gunicorn_warning:
+        print(gunicorn_warning)
+
     gunicorn_args = [
-        "gunicorn",
+        gunicorn_cmd,
         "-t",
         config.GUNICORN_TIMEOUT,
         "-k",
@@ -279,7 +312,10 @@ def main():
     if debug_mode:
         gunicorn_args.insert(1, "--reload")
 
-    if config.LAUNCH_BROWSER_ON_START:
+    if args.daemon:
+        gunicorn_args.insert(1, "--daemon")
+
+    if config.LAUNCH_BROWSER_ON_START and not args.daemon:
         flask_env = os.getenv("FLASK_ENV", "development")
         port = config.PORT if flask_env == "production" else config.DEV_SERVER_PORT
         host = config.HOST if flask_env == "production" else config.DEV_SERVER_HOST

ttnn_visualizer/csv_queries.py

@@ -463,7 +463,7 @@ class OpsPerformanceReportQueries:
 
         try:
             perf_report.generate_perf_report(
-                csv_file,
+                [csv_file],
                 signpost,
                 ignore_signposts,
                 cls.DEFAULT_MIN_PERCENTAGE,
@@ -497,7 +497,12 @@ class OpsPerformanceReportQueries:
                 op_id = index + 2  # Match IDs with row numbers in ops perf results csv
                 if not any(s["op_code"] == op_code for s in signposts):
                     captured_signposts.add(op_code)
-                    signposts.append({"id": op_id, "op_code": op_code})
+                    signposts.append(
+                        {
+                            "id": op_id,
+                            "op_code": op_code,
+                        }
+                    )
 
         report = []
 

ttnn_visualizer/settings.py

@@ -6,6 +6,7 @@ import os
 from pathlib import Path
 
 from dotenv import load_dotenv
+from sqlalchemy.pool import NullPool
 from ttnn_visualizer.utils import str_to_bool
 
 load_dotenv()
@@ -55,13 +56,17 @@ class DefaultConfig(object):
     USE_WEBSOCKETS = str_to_bool(os.getenv("USE_WEBSOCKETS", "true"))
 
     # SQL Alchemy Settings
-    SQLALCHEMY_DATABASE_URI = (
-        f"sqlite:///{os.path.join(APP_DATA_DIRECTORY, f'ttnn_{DB_VERSION}.db')}"
-    )
+    # Build database path - use absolute path to avoid any ambiguity
+    _db_file_path = str(Path(APP_DATA_DIRECTORY) / f"ttnn_{DB_VERSION}.db")
+    SQLALCHEMY_DATABASE_URI = f"sqlite:///{_db_file_path}"
     SQLALCHEMY_ENGINE_OPTIONS = {
-        "pool_size": 10,  # Adjust pool size as needed (default is 5)
-        "max_overflow": 20,  # Allow overflow of the pool size if necessary
-        "pool_timeout": 30,  # Timeout in seconds before giving up on getting a connection
+        # SQLite-specific settings for multi-process/worker environments
+        # NullPool: Each worker gets its own connection, avoiding file locking issues
+        # This is critical for gunicorn's multi-worker mode with SQLite
+        "poolclass": NullPool,
+        "connect_args": {
+            "check_same_thread": False,  # Allow SQLite to be used across threads in gevent
+        },
     }
     SQLALCHEMY_TRACK_MODIFICATIONS = False
 

ttnn_visualizer/static/assets/{allPaths-BCkFOWKv.js → allPaths-D6ZQWDaz.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-Bixx4Hi-.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-DhooSFaT.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-KltlmKXl.js

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