app.py

"""
Main page for the OpenMS Template App.

This module sets up and displays the Streamlit app for the OpenMS Template App.
It includes:
- Setting the app title.
- Displaying a description.
- Providing a download button for the Windows version of the app.

Usage:
Run this script to launch the OpenMS Template App.

Note:
- If run in local mode, the CAPTCHA control is not applied.
- If not in local mode, CAPTCHA control is applied to verify the user.

Returns:
    None
"""

import sys
import streamlit as st
from pathlib import Path
from src.captcha_ import captcha_control
from src.common import page_setup, save_params
from st_pages import Page, show_pages

params = page_setup(page="main")


def flashdeconvPages():
    show_pages([
        Page("app.py", "FLASHViewer", "🏠"),
        Page("pages/FLASHDeconvWorkflow.py", "Workflow", "⚙️"),
        Page("pages/FileUpload.py", "File Upload", "📁"),
        Page("pages/SequenceInput.py", "Sequence Input", "🧵"),
        Page("pages/LayoutManager.py", "Layout Manager", "📝️"),
        Page("pages/FLASHDeconvViewer.py", "Viewer", "👀"),
        Page("pages/FLASHDeconvDownload.py", "Download", "⬇️"),
    ])


def flashtagPages():
    show_pages([
        Page("app.py", "FLASHViewer", "🏠"),
        Page("pages/FLASHTaggerWorkflow.py", "Workflow", "⚙️"),
        Page("pages/FileUploadTagger.py", "File Upload", "📁"),
        Page("pages/LayoutManagerTagger.py", "Layout Manager", "📝️"),
        Page("pages/FLASHTaggerViewer.py", "Viewer", "👀"),
        Page("pages/FLASHTaggerDownload.py", "Download", "⬇️"),
    ])


def flashquantPages():
    show_pages([
        Page("app.py", "FLASHViewer", "🏠"),
        Page("pages/FileUpload_FLASHQuant.py", "File Upload", "📁"),
        Page("pages/FLASHQuantViewer.py", "Viewer", "👀"),
    ])


page_names_to_funcs = {
    "FLASHTagger": flashtagPages,
    "FLASHDeconv": flashdeconvPages,
    "FLASHQuant": flashquantPages,
}


def onToolChange():
    if 'changed_tool_name' in st.session_state:
        match st.session_state.changed_tool_name:
            case 'FLASHDeconv':
                st.session_state['tool_index'] = 0
            case 'FLASHTagger':
                st.session_state['tool_index'] = 1
            case 'FLASHQuant':
                st.session_state['tool_index'] = 2
        st.rerun()  # reload the page to sync the change


def main():
    """
    Display main page content.
    """

    # sidebar to toggle between tools
    if 'tool_index' not in st.session_state:
        page_names_to_funcs['FLASHDeconv']()
        st.session_state['tool_index'] = 0

    # main content
    st.markdown('#### FLASHViewer visualizes outputs from FLASH\* tools.')

    st.info("""
        **💡 How to run FLASHViewer**
        1. Go to the **⚙️ Workflow** page through the sidebar and run your analysis.\
            OR, go to the **📁 File Upload** page through the sidebar and upload FLASHDeconv output files (\*_annotated.mzML & \*_deconv.mzML)
        2. Click the **👀 Viewer** page on the sidebar to view the results in detail.
            
            **\***Download of results is supported.only for FLASHDeconv
        """)

    # when entered into other page, key is resetting (emptied) - thus set the value with index
    st.selectbox("Choose a tool", ['FLASHDeconv', 'FLASHTagger', 'FLASHQuant'], index=st.session_state.tool_index,
                 on_change=onToolChange(), key='changed_tool_name')
    page_names_to_funcs[st.session_state.changed_tool_name]()


    if Path("OpenMS-App.zip").exists():
        st.text("")
        st.text("")
        st.markdown("""
        Download the latest version for Windows by clicking the button below.
        """)
        with open("OpenMS-App.zip", "rb") as file:
            st.download_button(
                label="Download for Windows",
                data=file,
                file_name="OpenMS-App.zip",
                mime="archive/zip",
                type="primary",
            )
            
    save_params(params)


# Check if the script is run in local mode (e.g., "streamlit run app.py local")
if "local" in sys.argv:
    # In local mode, run the main function without applying captcha
    main()

# If not in local mode, assume it's hosted/online mode
else:
    show_pages([
        Page("app.py", "FLASHViewer", "🏠"),
    ])

    # WORK LIKE MULTIPAGE APP
    if "controllo" not in st.session_state or st.session_state["controllo"] is False:
        # Apply captcha control to verify the user
        captcha_control()

    else:
        # Run the main function
        main()