File size: 4,321 Bytes
38167d4 f0f4b86 38167d4 ba3027e 38167d4 ba3027e d6bb045 38167d4 8a5ab3e 38167d4 ba3027e 38167d4 8a5ab3e 38167d4 ba3027e cb8b3fe 810a2b0 38167d4 8a5ab3e 38167d4 ba3027e 8a5ab3e 38167d4 8a5ab3e 82d483e ba3027e 8a5ab3e 116ed6e 58b59bc 116ed6e bca7fc0 3afdc18 58b59bc 41361e7 6c04fe8 38167d4 8a5ab3e 38167d4 ba3027e 8a5ab3e 38167d4 8a5ab3e 38167d4 ba3027e e0f05ce 38167d4 8a5ab3e 38167d4 ba3027e e0f05ce cb8b3fe e8eea71 e0f05ce f0f4b86 38167d4 8a5ab3e 38167d4 ba3027e e0f05ce 38167d4 ba3027e 38167d4 8a5ab3e 38167d4 ba3027e 8a5ab3e 38167d4 8a5ab3e 38167d4 8a5ab3e 38167d4 ba3027e 8a5ab3e 38167d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import pandas as pd
import streamlit as st
from my_model.tabs.run_inference import InferenceRunner
class UIManager:
"""Manages the user interface for the Streamlit application."""
def __init__(self):
"""Initializes the UIManager with predefined tabs."""
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):
"""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.text('')
st.markdown("<h2 style='text-align: center;'>(Knowledge-Based Visual Question Answering)</h2>", unsafe_allow_html=True)
st.text('')
st.text('')
st.text('')
st.write("""\n\n\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\nDeveloped by: [Mohammed H AlHaj](https://www.linkedin.com/in/m7mdal7aj) | Dissertation Supervisor: [Andreas Theophilou](https://researchportal.bath.ac.uk/en/persons/andreas-theophilou)
\n\nFurther 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."""
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.")
st.write("\n")
inference_runner = InferenceRunner()
inference_runner.run_inference()
def display_dissertation_report(self):
"""Displays the Dissertation Report page."""
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):
"""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 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.")
|