Source code for webinterface.pages.3_Quant_LFQ_DIA_ion_AIF

"""
Streamlit UI for the DIA quantification - precursor ions module - AIF.
"""

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_AIF_variables import VariablesDIAQuant
from pages.texts.generic_texts import WebpageTexts

from proteobench.io.parsing.parse_settings import ParseSettingsBuilder
from proteobench.modules.quant.quant_lfq_ion_DIA_AIF import DIAQuantIonModuleAIF




class StreamlitUI:
    """
    Streamlit UI for the DIA quantification - precursor ions module - AIF.
    """

    def __init__(self):
        """
        Initialize the Streamlit UI for the DIA quantification - precursor ions module - AIF.
        """
        self.variables_dia_quant: VariablesDIAQuant = VariablesDIAQuant()
        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: DIAQuantIonModuleAIF = DIAQuantIonModuleAIF(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 AIF",
        )

        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_public_submission,
        ) = st.tabs(
            [
                "Public Benchmark Runs",
                "Submit New Data",
                "Results In-Depth",
                "Results New Data",
                "Public Submission",
            ]
        )

        # 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 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