"""
Streamlit UI for the DIA quantification - precursor ions module - diaPASEF.
"""
import logging
from typing import Any, Dict, Type
import pages.texts.proteobench_builder as pbb
import streamlit as st
from pages.base_pages.banner import display_banner
from pages.base_pages.quant import QuantUIObjects
from pages.pages_variables.Quant.lfq_DIA_ion_diaPASEF_variables import (
VariablesDIAQuantdiaPASEF,
)
from pages.texts.generic_texts import WebpageTexts
from proteobench.io.parsing.parse_settings import ParseSettingsBuilder
from proteobench.modules.quant.quant_lfq_ion_DIA_diaPASEF import (
DIAQuantIonModulediaPASEF,
)
[docs]
class StreamlitUI:
"""
Streamlit UI for the DIA quantification - precursor ions module - diaPASEF.
"""
def __init__(self):
"""
Initialize the Streamlit UI for the DIA quantification - precursor ions module - diaPASEF.
"""
self.variables_dia_quant: VariablesDIAQuantdiaPASEF = VariablesDIAQuantdiaPASEF()
self.texts: Type[WebpageTexts] = WebpageTexts
self.user_input: Dict[str, Any] = dict()
pbb.proteobench_page_config()
if self.variables_dia_quant.submit not in st.session_state:
st.session_state[self.variables_dia_quant.submit] = False
try:
token = st.secrets["gh"]["token"]
except KeyError:
token = ""
self.ionmodule: DIAQuantIonModulediaPASEF = DIAQuantIonModulediaPASEF(token=token)
self.parsesettingsbuilder = ParseSettingsBuilder(
module_id=self.ionmodule.module_id, parse_settings_dir=self.variables_dia_quant.parse_settings_dir
)
self.quant_uiobjects = QuantUIObjects(
self.variables_dia_quant,
self.ionmodule,
self.parsesettingsbuilder,
page_name="Quant LFQ DIA ion diaPASEF",
)
self._main_page()
def _main_page(self) -> None:
"""
Set up the main page layout for the Streamlit application.
"""
# Create tabs
(
tab_results_all,
tab_submission_details,
tab_indepth_plots,
tab_results_new,
tab_compare_workflows,
tab_public_submission,
) = st.tabs(
[
"View Public Results",
"Upload New Results (Private)",
"View Single Result",
"View Public + New Results",
"Compare Two Results",
"Submit New Results",
]
)
# Tab 1: Results (All Data)
with tab_results_all:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_all_data_results_main()
# Tab 2: Submission Details
with tab_submission_details:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_submission_form()
# Tab 2.5: in-depth plots current data
with tab_indepth_plots:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_indepth_plots()
# Tab 3: Results (New Submissions)
with tab_results_new:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_all_data_results_submitted()
# Tab: Compare Workflows
with tab_compare_workflows:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_workflow_comparison()
# Tab 4: Public Submission
with tab_public_submission:
st.title(self.variables_dia_quant.title)
st.link_button(
"Go to module documentation",
url=self.variables_dia_quant.doc_url,
type="secondary",
help="link to the module documentation",
)
display_banner(self.variables_dia_quant)
self.quant_uiobjects.display_public_submission_ui()
if __name__ == "__main__":
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
StreamlitUI() # Instantiate and run the extended UI class
