Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -1,8 +1,9 @@
 import pandas as pd
 import streamlit as st
 from my_model.tabs.run_inference import InferenceRunner
+from my_model.state_manager import StateManager
 
-class UIManager:
+class UIManager():
     """Manages the user interface for the Streamlit application."""
 
     def __init__(self):
@@ -18,6 +19,9 @@ class UIManager:
             "More Pages will follow .. ": self.display_placeholder
         }
 
+        state_manager = StateManager()
+        state_manager.initialize_state()
+
     
     def add_tab(self, tab_name, display_function):
         """Adds a new tab to the UI."""