refactoring

main.py

@@ -33,7 +33,7 @@ from core.config import get_config, setup_logging, Config
 from core.data_provider import DataProvider
 
 # Import checkpoint management
-from utils.checkpoint_manager import get_checkpoint_manager
+from NN.training.model_manager import create_model_manager
 from utils.training_integration import get_training_integration
 
 logger = logging.getLogger(__name__)
@@ -77,7 +77,7 @@ async def run_web_dashboard():
         
         # Load model registry for integrated pipeline
         try:
-            from models import get_model_registry
+            from NN.training.model_manager import create_model_manager
             model_registry = {}  # Use simple dict for now
             logger.info("[MODELS] Model registry initialized for training")
         except ImportError:
@@ -85,7 +85,7 @@ async def run_web_dashboard():
             logger.warning("Model registry not available, using empty registry")
         
         # Initialize checkpoint management
-        checkpoint_manager = get_checkpoint_manager()
+        checkpoint_manager = create_model_manager()
         training_integration = get_training_integration()
         logger.info("Checkpoint management initialized for training pipeline")
         
@@ -163,7 +163,7 @@ def start_web_ui(port=8051):
         
         # Load model registry for enhanced features
         try:
-            from models import get_model_registry
+            from NN.training.model_manager import create_model_manager
             model_registry = {}  # Use simple dict for now
         except ImportError:
             model_registry = {}