Spaces:
Running
Running
| import streamlit as st | |
| import streamlit.components.v1 as components | |
| import pandas as pd | |
| 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.model_arch import run_model_arch | |
| class UIManager(): | |
| """ | |
| Manages the user interface for the Streamlit application. | |
| This class handles the creation and navigation of various tabs in the Streamlit app. It provides methods to add new | |
| tabs, display the sidebar, and render the content of the selected tab. | |
| Attributes: | |
| tabs (dict): A dictionary mapping tab names to their corresponding display functions. | |
| """ | |
| def __init__(self)-> None: | |
| """ | |
| Initializes the UIManager with predefined tabs. | |
| This method sets up the initial tabs for the application and initializes the state manager. | |
| """ | |
| 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, | |
| "Code": self.display_code | |
| } | |
| state_manager = StateManager() | |
| state_manager.initialize_state() | |
| def add_tab(self, tab_name: str, display_function: callable) -> None: | |
| """ | |
| Adds a new tab to the UI. | |
| Args: | |
| tab_name (str): The name of the new tab. | |
| display_function (callable): The function to be called when the tab is selected. | |
| Returns: | |
| None | |
| """ | |
| self.tabs[tab_name] = display_function | |
| def display_sidebar(self) -> str: | |
| """ | |
| Displays the sidebar for navigation. | |
| This method creates a sidebar with navigation options and returns the user's selection. | |
| Returns: | |
| str: The name of the selected tab. | |
| """ | |
| st.sidebar.image("Files/logo.jpg") | |
| st.sidebar.title("Navigation") | |
| selection = st.sidebar.radio("Go to", list(self.tabs.keys()), disabled=st.session_state['loading_in_progress']) | |
| st.sidebar.image("Files/mm.jpeg", use_column_width=True) | |
| st.sidebar.markdown( | |
| """ | |
| <div style="text-align: center;"> | |
| <a href="https://www.linkedin.com/in/m7mdal7aj" style="font-weight: bold; text-decoration: none;">Mohammed H AlHaj</a> | |
| </div> | |
| """, | |
| unsafe_allow_html=True | |
| ) | |
| return selection | |
| def display_selected_page(self, selection: str) -> None: | |
| """ | |
| Displays the selected page based on the user's choice. | |
| Args: | |
| selection (str): The name of the selected tab. | |
| Returns: | |
| None | |
| """ | |
| if selection in self.tabs: | |
| self.tabs[selection]() | |
| def display_home(self) -> None: | |
| """ | |
| Displays the Home page of the application. | |
| Returns: | |
| None | |
| """ | |
| run_home() | |
| def display_dataset_analysis(self) -> None: | |
| """ | |
| Displays the Dataset Analysis page. | |
| This page provides an overview of various KB-VQA datasets and analyses of the OK-VQA dataset. | |
| Returns: | |
| None | |
| """ | |
| st.title("Dataset Analysis") | |
| st.write("""This page shows an overview of some of the KB-VQA datasets, and various analysis of | |
| the [OK-VQA Dataset](https://okvqa.allenai.org/) that this KB-VQA model was fine-tuned | |
| and evaluated on.""") | |
| run_dataset_analyzer() | |
| def display_results(self) -> None: | |
| """ | |
| Displays the Evaluation Results page. | |
| This page demonstrates the model evaluation results and analyses in an interactive way. | |
| Returns: | |
| None | |
| """ | |
| st.title("Evaluation Results & Analyses") | |
| st.write("This page demonstrates the model evaluation results and analyses in an interactive way.") | |
| st.write("\n") | |
| run_demo() | |
| def display_model_arch(self) -> None: | |
| """ | |
| Displays the Model Architecture page. | |
| This page provides detailed information about the model architecture. | |
| Returns: | |
| None | |
| """ | |
| st.title("Model Architecture") | |
| st.write("This page shows the detailed Model Architecture.") | |
| st.write("\n") | |
| run_model_arch() | |
| def display_run_inference(self) -> None: | |
| """ | |
| Displays the Run Inference page. | |
| This page allows users to run inference using the fine-tuned model on the OK-VQA dataset. | |
| Returns: | |
| None | |
| """ | |
| st.title("Run Inference") | |
| st.write("""Please note that this is not a general purpose model, it is specifically trained on | |
| [OK-VQA Dataset](https://okvqa.allenai.org/) and desgined to give short and direct answers to the | |
| given questions about the given image.\n""") | |
| inference_runner = InferenceRunner() | |
| inference_runner.run_inference() | |
| def display_dissertation_report(self) -> None: | |
| """ | |
| Displays the Dissertation Report page. | |
| This page provides a link to download the dissertation report PDF. | |
| Returns: | |
| None | |
| """ | |
| st.title("Dissertation Report") | |
| st.write("Click the link below to view the PDF.") | |
| # Error handling for file access should be considered here | |
| 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) -> None: | |
| """ | |
| Displays the Code page with a link to the project's code repository. | |
| Returns: | |
| None | |
| """ | |
| st.title("Code") | |
| 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_KB-VQA/tree/main)", unsafe_allow_html=True) | |
| def display_placeholder(self) -> None: | |
| """ | |
| Displays a placeholder for future content. | |
| Returns: | |
| None | |
| """ | |
| st.title("Stay Tuned") | |
| st.write("This is a Place Holder until the contents are uploaded.") | |