Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -5,7 +5,8 @@ from my_model.tabs.run_inference import InferenceRunner
 from my_model.tabs.results import run_demo
 from my_model.tabs.home import run_home
 from my_model.state_manager import StateManager
-from my_model.tabs.dataset_analysis import run_dataset_analyzer
+from my_model.tabs.dataset_analysis import run_dataset_analyzer  run_model_arch
+from my_model.tabs.model_arch import run_model_arch
 
 class UIManager():
     """Manages the user interface for the Streamlit application."""
@@ -16,6 +17,7 @@ class UIManager():
         self.tabs = {
             "Home": self.display_home,
             "Dataset Analysis": self.display_dataset_analysis,
+            "Model Architecture": self.display_model_arch()
             "Results": self.display_results,
             "Run Inference": self.display_run_inference,
             "Dissertation Report": self.display_dissertation_report,
@@ -80,6 +82,15 @@ class UIManager():
         
         run_demo()
 
+    def display_model_arch(self):
+        """Displays Model Architecture page."""
+        
+        st.title("Model Architecture")
+        st.write("This page shows the detailed Model Architecture.")
+        st.write("\n")
+        
+        run_model_arch()
+
     
     def display_run_inference(self):
         """Displays the Run Inference page."""