Update my_model/utilities/st_utils.py

my_model/utilities/st_utils.py

@@ -1,30 +1,71 @@
 import streamlit as st
+from my model.
 
 class UIManager:
-    def __init__(self, state_manager):
-        self.state_manager = state_manager
-
-    def display_sidebar_navigation(self):
+    def __init__(self):
+        self.tabs = {
+            "Home": self.display_home,
+            "Dataset Analysis": self.display_dataset_analysis,
+            "Finetuning and Evaluation Results": self.display_finetuning_evaluation,
+            "Run Inference": self.display_run_inference,
+            "Dissertation Report": self.display_dissertation_report,
+            "Code": self.display_code,
+            "More Pages will follow .. ": self.display_placeholder
+        }
+
+    def add_tab(self, tab_name, display_function):
+        self.tabs[tab_name] = display_function
+
+    def display_sidebar(self):
         st.sidebar.title("Navigation")
-        return st.sidebar.radio("Go to", ["Home", "Dataset Analysis", "Evaluation Results", "Run Inference", "Dissertation Report"])
+        selection = st.sidebar.radio("Go to", list(self.tabs.keys()))
+        st.sidebar.write("More Pages will follow .. ")
+        return selection
+
+    def display_selected_page(self, selection):
+        if selection in self.tabs:
+            self.tabs[selection]()
 
-    def display_home_page(self):
+    def display_home(self):
         st.title("MultiModal Learning for Knowledge-Based Visual Question Answering")
-        st.write("Home page content goes here...")
+        st.write("""This application is an interactive element of the project and prepared by Mohammed Alhaj as part of the dissertation for Masters degree in Artificial Intelligence at the University of Bath. 
+                    Further details will be updated later""")
+
+    def display_dataset_analysis(self):
+        st.title("OK-VQA Dataset Analysis")
+        st.write("This is a Place Holder until the contents are uploaded.")
+
+    def display_finetuning_evaluation(self):
+        st.title("Finetuning and Evaluation Results")
+        st.write("This is a Place Holder until the contents are uploaded.")
+
+    def display_run_inference(self):
+        
+        run_inference()
 
     def display_dissertation_report(self):
         st.title("Dissertation Report")
         st.write("Click the link below to view the PDF.")
-        with open("Files/Dissertation Report.pdf", "rb") as file:
-            st.download_button(label="Download PDF", data=file, file_name="Dissertation_Report.pdf", mime="application/octet-stream")
-
-    def display_evaluation_results(self):
-        st.title("Evaluation Results")
+        st.download_button(
+            label="Download PDF",
+            data=open("Files/Dissertation Report.pdf", "rb"),
+            file_name="example.pdf",
+            mime="application/octet-stream"
+        )
+
+    def display_code(self):
+        st.title("Code")
+        st.markdown("You can view the code for this project on the Hugging Face Space file page.")
+        st.markdown("[View Code](https://huggingface.co/spaces/m7mdal7aj/Mohammed_Alhaj_PlayGround/tree/main)", unsafe_allow_html=True)
+
+    def display_placeholder(self):
+        st.title("Stay Tuned")
         st.write("This is a Place Holder until the contents are uploaded.")
 
-    def display_dataset_analysis(self):
-        st.title("OK-VQA Dataset Analysis")
-        st.write("This is a Place Holder until the contents are uploaded.")
+
+
+
+