File size: 3,698 Bytes
ec6dd69 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 |
from menu import display_pages_menu
import streamlit as st
from utils import set_algorithm_name, get_pdf_iframe
import sys
import logging
logging.basicConfig(stream=sys.stdout, level=logging.INFO, format="%(message)s")
keep = "keep the extracted value"
remove = "remove this column"
def transpose_current_table():
st.session_state.tables[
st.session_state["algorithm_name"]
] = st.session_state.tables[st.session_state["algorithm_name"]].transpose()
def set_headers(algorithm_name: str) -> None:
for header in st.session_state.tables[algorithm_name].columns.values.tolist():
if st.session_state["widget" + str(header)] == remove:
st.session_state.tables[algorithm_name].drop(columns=[header], inplace=True)
if st.session_state["widget" + str(header)] == keep:
pass
else:
st.session_state.tables[algorithm_name].rename(
columns={header: st.session_state["widget" + str(header)]},
inplace=True,
)
header_list = [
keep,
"jurisdiction",
"profit_before_tax",
"tax_accrued",
"tax_paid",
"employees",
"unrelated_revenues",
"related_revenues",
"stated_capital",
"accumulated_earnings",
"tangible_assets",
"total_revenues",
remove,
]
st.set_page_config(layout="wide") # page_icon="📈"
st.title("Country by Country Tax Reporting analysis : Headers")
st.subheader(
"This page will allow you to modify the headers and to remove columns",
)
display_pages_menu()
if "tables" not in st.session_state:
st.markdown(
"# !! Don't change the page while the algorithms are runing, else they will start again"
)
if (
st.session_state.get("validate_selected_pages", False)
and "pdf_after_page_validation" in st.session_state
):
col1, col2 = st.columns(2)
with col1:
st.markdown(
get_pdf_iframe(st.session_state["pdf_after_page_validation"]),
unsafe_allow_html=True,
)
with col2:
index = (
list(st.session_state.tables.keys()).index(
st.session_state["algorithm_name"],
)
if "algorithm_name" in st.session_state
else 0
)
st.session_state["algorithm_name"] = st.selectbox(
"Choose the extracted table you want to see",
list(st.session_state.tables.keys()),
index=index,
on_change=set_algorithm_name,
args=("selectbox1",),
key="selectbox1",
)
st.markdown(
"Table shape :"
+ str(st.session_state.tables[st.session_state["algorithm_name"]].shape)
)
st.button("Transpose table", on_click=transpose_current_table)
with st.form(key="my_form"):
for header in st.session_state.tables[
st.session_state["algorithm_name"]
].columns.values.tolist():
st.selectbox(
"Choose the value of the following extracted header : "
+ str(header),
header_list,
key="widget" + str(header),
)
submitted = st.form_submit_button(
label="Submit",
on_click=set_headers,
args=(st.session_state["algorithm_name"],),
)
if submitted:
st.switch_page("pages/5_Clean_Tables.py")
st.markdown("# Current extraction")
st.markdown("The extracted table is displaye below")
st.dataframe(
st.session_state.tables[st.session_state["algorithm_name"]],
width=900,
height=900,
)
|