main.py

from typing import Dict, Type

import streamlit as st

from src import pages

from src.utils import Page, get_file_content_as_string, provide_state

PAGE_MAP: Dict[str, Type[Page]] = {
    "Audio Analysis": pages.AudioAnalysis,
    "Page 1": pages.Page1,
    "Page 2": pages.Page2
}


@provide_state()
def main(state=None):
    st.set_page_config(page_title="Sonar Analyzer", layout="centered")
    readme_text = st.markdown(get_file_content_as_string("docs/instructions.md"))

    st.sidebar.header("Sonar Analyzer")
    current_page = st.sidebar.selectbox("Menu", ["Home"] + list(PAGE_MAP))

    if current_page == "Home":
        st.sidebar.success('To continue select an option aboce.')
    else:
        readme_text.empty()
        PAGE_MAP.get(current_page, pages.Default)(state=state).write()


if __name__ == "__main__":
    main()