Update my_model/utilities/ui_manager.py

my_model/utilities/ui_manager.py

@@ -18,26 +18,34 @@ class UIManager:
             "More Pages will follow .. ": self.display_placeholder
         }
 
+    
     def add_tab(self, tab_name, display_function):
         """Adds a new tab to the UI."""
         self.tabs[tab_name] = display_function
 
+    
     def display_sidebar(self):
         """Displays the sidebar for navigation."""
+        
         st.sidebar.title("Navigation")
         selection = st.sidebar.radio("Go to", list(self.tabs.keys()))
         return selection
+        
 
     def display_selected_page(self, selection):
         """Displays the selected page based on user's choice."""
+        
         if selection in self.tabs:
             self.tabs[selection]()
 
+    
     def display_home(self):
         """Displays the Home page of the application."""
+        
         st.markdown("<h1 style='text-align: center;'>MultiModal Learning for Visual Question Answering using World Knowledge</h1>", unsafe_allow_html=True)
         st.write(" ")
         st.markdown("<h2 style='text-align: center;'>(Knowledge-Based Visual Question Answering)</h2>", unsafe_allow_html=True)
+        
         st.write("""\n\nThis application is an interactive element of the project as part of the dissertation for Masters degree in Artificial Intelligence at the [University of Bath](https://www.bath.ac.uk/). 
                     \n\n\n
                     Developed by: [Mohammed H AlHaj](https://www.linkedin.com/in/m7mdal7aj)
@@ -45,21 +53,27 @@ class UIManager:
                     Dissertation Supervisor: [Andreas Theophilou](https://researchportal.bath.ac.uk/en/persons/andreas-theophilou)
                     Further details will be updated later""")
 
+    
     def display_dataset_analysis(self):
         """Displays the Dataset Analysis page."""
+        
         st.title("OK-VQA Dataset Analysis")
         st.write("This is a Place Holder until the contents are uploaded.")
 
+    
     def display_finetuning_evaluation(self):
         """Displays the Finetuning and Evaluation Results page."""
         st.title("Finetuning and Evaluation Results")
         st.write("This is a Place Holder until the contents are uploaded.")
 
+    
     def display_run_inference(self):
         """Displays the Run Inference page."""
+        
         inference_runner = InferenceRunner()
         inference_runner.run_inference()
 
+    
     def display_dissertation_report(self):
         """Displays the Dissertation Report page."""
         st.title("Dissertation Report")
@@ -72,14 +86,17 @@ class UIManager:
             mime="application/octet-stream"
         )
 
+    
     def display_code(self):
         """Displays the Code page with a link to the project's code repository."""
+        
         st.title("Code")
-        st.markdown("You can view the code for this project on the HuggingFace Space file page.")
-        # Inform users they are leaving the app
+        st.markdown("You can view the code for this project on HuggingFace Space files page.")
         st.markdown("[View Code](https://huggingface.co/spaces/m7mdal7aj/Mohammed_Alhaj_PlayGround/tree/main)", unsafe_allow_html=True)
 
+    
     def display_placeholder(self):
         """Displays a placeholder for future content."""
+        
         st.title("Stay Tuned")
         st.write("This is a Place Holder until the contents are uploaded.")