fix: exclude app.py from exported server package, add torch init

nodes/server_manager.py

@@ -148,7 +148,7 @@ async def start(self, port=None, host=None):
                 self.host = host
 
             # Get the path to the ComfyStream server directory and script
-            server_dir = Path(__file__).parent.parent / "src" / "comfystream" / "server"
+            server_dir = Path(__file__).parent.parent / "server"
             server_script = server_dir / "app.py"
             logging.info(f"Server script: {server_script}")
 

src/comfystream/server/app.py → server/app.py

@@ -4,6 +4,11 @@
 import logging
 import os
 import sys
+import torch
+
+# Initialize CUDA before any other imports to prevent core dump.
+if torch.cuda.is_available():
+    torch.cuda.init()
 
 from aiohttp import web
 from aiortc import (

src/comfystream/__init__.py

@@ -1,16 +1,13 @@
 from .client import ComfyStreamClient
 from .pipeline import Pipeline
 from .server.utils import temporary_log_level
-from .server.app import VideoStreamTrack, AudioStreamTrack
 from .server.utils import FPSMeter
 from .server.metrics import MetricsManager, StreamStatsManager
 
 __all__ = [
     'ComfyStreamClient',
     'Pipeline',
     'temporary_log_level',
-    'VideoStreamTrack',
-    'AudioStreamTrack',
     'FPSMeter',
     'MetricsManager',
     'StreamStatsManager'