ptxboa_streamlit.py

# -*- coding: utf-8 -*-
"""
PtX BOA streamlit app, main file.

Execution:
>>> streamlit run  ptxboa_streamlit.py
"""

__version__ = "2.1.1"

import logging

import pandas as pd
import streamlit as st
import streamlit_antd_components as sac

from app.context_data import load_context_data
from app.layout_elements import display_footer
from app.sidebar import make_sidebar
from app.tab_certification_schemes import content_certification_schemes
from app.tab_costs import content_costs
from app.tab_country_fact_sheets import content_country_fact_sheets
from app.tab_deep_dive_countries import content_deep_dive_countries
from app.tab_info import content_info
from app.tab_input_data import content_input_data
from app.tab_literature import content_literature
from app.tab_market_scanning import content_market_scanning
from app.tab_optimization import content_optimization
from app.tab_sustainability import content_sustainability
from app.user_data import display_user_changes
from app.user_data_from_file import download_user_data, upload_user_data
from ptxboa import DEFAULT_CACHE_DIR, DEFAULT_DATA_DIR
from ptxboa.api import PtxboaAPI

# setup logging
# level can be changed on strartup with: --logger.level=LEVEL
loglevel = st.logger.get_logger(__name__).level
logger = logging.getLogger()  # do not use __name__ so we can resue it in submodules
logger.setLevel(loglevel)
if not logger.handlers:
    # only add one handler
    logger.handlers.append(logging.StreamHandler())
for handler in logger.handlers:
    handler.setFormatter(
        logging.Formatter(
            "[%(asctime)s %(levelname)7s] %(message)s", datefmt="%Y-%m-%d %H:%M:%S"
        )
    )


# app layout:

st.set_page_config(
    page_title="PtX Business Opportunity Analyser", page_icon="./data/favicon-16x16.png"
)

# Set the pandas display option to format floats with 2 decimal places
pd.set_option("display.float_format", "{:.2f}".format)

if "model_status" not in st.session_state:
    st.session_state["model_status"] = "not yet solved"


if "user_changes_df" not in st.session_state:
    st.session_state["user_changes_df"] = None

if "edit_input_data" not in st.session_state:
    st.session_state["edit_input_data"] = False

# https://discuss.streamlit.io/t/can-not-set-page-width-in-streamlit-1-5-0/21522/5
css = """
<style>
    section.main > div {max-width:80rem}
</style>
"""
st.markdown(css, unsafe_allow_html=True)

# https://discuss.streamlit.io/t/delete-red-bar-at-the-top-of-the-app/9658
hide_decoration_bar_style = """
    <style>
        header {visibility: hidden;}
    </style>
"""
st.markdown(hide_decoration_bar_style, unsafe_allow_html=True)

api = st.cache_resource(PtxboaAPI)(
    data_dir=DEFAULT_DATA_DIR,
    cache_dir=DEFAULT_CACHE_DIR,  # TODO: maybe disable in test environment?
)


st.title("PtX Business Opportunity Analyser")

with st.container():
    if st.session_state["edit_input_data"]:
        st.info("Data editing mode **ON**")
        with st.expander("Modified data"):
            if st.session_state["user_changes_df"] is not None:
                download_user_data()
            display_user_changes(api)

            with st.container():
                st.divider()
                st.markdown("##### Upload your data from a previous session")
                upload_user_data(api)
    else:
        placeholder = st.empty()

tabs = (
    "Info",
    "Costs",
    "Market scanning",
    "Deep-dive countries",
    "Optimization",
    "Input data",
    "Country fact sheets",
    "Certification schemes",
    "Sustainability",
    "Literature",
)

tabs_icons = {
    "Costs": "house-fill",
    "Info": "question-circle-fill",
}

# the "tab_key" is used to identify the sac.tabs element. Whenever a tab is switched
# programatically (e.g. via app.ptxboa.functions.move_to_tab), the "tab_key" entry is
# incremented by 1. This allows us to set the programatically set tab as the default
# `index` in `sac.tabs()`.
if "tab_key" not in st.session_state:
    st.session_state["tab_key"] = "tab_key_0"

# initializing "tab at first round
if st.session_state["tab_key"] not in st.session_state:
    st.session_state[st.session_state["tab_key"]] = "Costs"

sac.buttons(
    [sac.ButtonsItem(label=i, icon=tabs_icons.get(i, None)) for i in tabs],
    index=tabs.index(st.session_state[st.session_state["tab_key"]]),
    format_func="title",
    align="center",
    key=st.session_state["tab_key"],
)
st.divider()
# create sidebar:
make_sidebar(api)

# import agora color scale:
if "colors" not in st.session_state:
    colors = pd.read_csv("data/Agora_Industry_Colours.csv")
    st.session_state["colors"] = colors["Hex Code"].to_list()

if st.session_state[st.session_state["tab_key"]] in [
    "Market scanning",
    "Deep-dive countries",
    "Country fact sheets",
    "Certification schemes",
    "Sustainability",
    "Literature",
]:
    # import context data:
    cd = load_context_data()

# costs:
if st.session_state[st.session_state["tab_key"]] == "Costs":
    content_costs(api)

if st.session_state[st.session_state["tab_key"]] == "Market scanning":
    content_market_scanning(api, cd)

if st.session_state[st.session_state["tab_key"]] == "Input data":
    content_input_data(api)

if st.session_state[st.session_state["tab_key"]] == "Deep-dive countries":
    content_deep_dive_countries(api)

if st.session_state[st.session_state["tab_key"]] == "Country fact sheets":
    content_country_fact_sheets(cd, api)

if st.session_state[st.session_state["tab_key"]] == "Certification schemes":
    content_certification_schemes(cd)

if st.session_state[st.session_state["tab_key"]] == "Sustainability":
    content_sustainability(cd)

if st.session_state[st.session_state["tab_key"]] == "Literature":
    content_literature(cd)

if st.session_state[st.session_state["tab_key"]] == "Info":
    content_info()

if st.session_state[st.session_state["tab_key"]] == "Optimization":
    content_optimization(api)

display_footer()