#import appStore.target as target_extraction
#import appStore.netzero as netzero
#import appStore.sector as sector
#import appStore.adapmit as adapmit
#import appStore.ghg as ghg
#import appStore.policyaction as policyaction
#import appStore.conditional as conditional
#import appStore.indicator as indicator
import appStore.doc_processing as processing
from utils.uploadAndExample import add_upload
from PIL import Image
import streamlit as st

####################################### Dashboard ######################################################

# App 

st.set_page_config(page_title = 'Vulnerable Groups Identification', 
                   initial_sidebar_state='expanded', layout="wide") 

with st.sidebar:
    # upload and example doc
    choice = st.sidebar.radio(label = 'Select the Document',
                            help = 'You can upload the document \
                            or else you can try a example document', 
                            options = ('Upload Document', 'Try Example'), 
                            horizontal = True)
    add_upload(choice)

with st.container():
        st.markdown("<h2 style='text-align: center; color: black;'> Vulnerable Groups Identification </h2>", unsafe_allow_html=True)
        st.write(' ')

with st.expander("ℹ️ - About this app", expanded=False):
    st.write(
        """
        The Vulnerable Groups Identification App is an open-source\
        digital tool which aims to assist policy analysts and \
        other users in extracting and filtering relevant \
        information from public documents.
        """)
    st.write('**Definitions**')

    st.caption("""
            - **Place holder**: Place holder \
            Place holder \
            Place holder \
            Place holder \ 
            Place holder
              """)
    #c1, c2, c3 =  st.columns([12,1,10])
    #with c1:
        #image = Image.open('docStore/img/flow.jpg') 
        #st.image(image)
    #with c3:
        #st.write("""
        #    What happens in the background?
            
        #    - Step 1: Once the document is provided to app, it undergoes *Pre-processing*.\
        #    In this step the document is broken into smaller paragraphs \
        #    (based on word/sentence count).
        #    - Step 2: The paragraphs are fed to **Target Classifier** which detects if
        #    the paragraph contains any *Target* related information or not.
        #    - Step 3: The paragraphs which are detected containing some target \
        #    related information are then fed to multiple classifier to enrich the 
        #    Information Extraction.
    
        #    The Step 2 and 3 are repated then similarly for Action and  Policies & Plans.
        #    """)
                  
    #st.write("")


apps = [processing.app, target_extraction.app, netzero.app, ghg.app,
        policyaction.app, conditional.app, sector.app, adapmit.app,indicator.app]

multiplier_val =1/len(apps)
if st.button("Analyze Document"):
    prg = st.progress(0.0)
    for i,func in enumerate(apps):
        func()
        prg.progress((i+1)*multiplier_val)

    
if 'key1' in st.session_state:
    with st.sidebar:
        topic = st.radio(
                        "Which category you want to explore?",
                        ('Target', 'Action', 'Policies/Plans'))
    
    if topic == 'Target':
        target_extraction.target_display()
    elif topic == 'Action':
        policyaction.action_display()
    else: 
        policyaction.policy_display()
    # st.write(st.session_state.key1)

#st.title("Identify references to vulnerable groups.")

#st.write("""Vulnerable groups encompass various communities and individuals who are disproportionately affected by the impacts of climate change
#due to their socioeconomic status, geographical location, or inherent characteristics. By incorporating the needs and perspectives of these groups 
#into national climate policies, governments can ensure equitable outcomes, promote social justice, and strive to build resilience within the most marginalized populations, 
#fostering a more sustainable and inclusive society as we navigate the challenges posed by climate change.This app allows you to identify whether a text contains any 
#references to vulnerable groups, for example when talking about policy documents.""")

# Document upload
#uploaded_file = st.file_uploader("Upload your file here")

# Create text input box
#input_text = st.text_area(label='Please enter your text here', value="This policy has been implemented to support women.")

#st.write('Prediction:', model(input_text))

######################################### Model #########################################################

# Load the model
#model = SetFitModel.from_pretrained("leavoigt/vulnerable_groups")

# Define the classes
#id2label = {
 #   0: 'Agricultural communities',
 #   1: 'Children and Youth',
 #   2: 'Coastal communities',
 #   3: 'Drought-prone regions',
 #   4: 'Economically disadvantaged communities',
 #   5: 'Elderly population',
 #   6: 'Ethnic minorities and indigenous people',
 #   7: 'Informal sector workers',
 #   8: 'Migrants and Refugees',
 #   9: 'Other',
 #   10: 'People with Disabilities',
 #   11: 'Rural populations',
 #   12: 'Sexual minorities (LGBTQI+)',
 #   13: 'Urban populations',
 #   14: 'Women'}


### Process document to paragraphs 
# Source: https://blog.jcharistech.com/2021/01/21/how-to-save-uploaded-files-to-directory-in-streamlit-apps/

# Store uploaded file temporarily in directory to get file path (necessary for processing)
# def save_uploadedfile(upl_file):
#      with open(os.path.join("tempDir",upl_file.name),"wb") as f:
#          f.write(upl_file.getbuffer())
#      return st.success("Saved File:{} to tempDir".format(upl_file.name))

# if uploaded_file is not None: 
#     # Save the file 
#     file_details = {"FileName": uploaded_file.name, "FileType": uploaded_file.type}
#     save_uploadedfile(uploaded_file)

#     #Get the file path