import streamlit as st

# setting page config. for centered mode
st.set_page_config(layout="centered")


from utils.footer import cust_footer
import docx2txt
import requests
import pdfplumber

# function to run the enter button
def run_function(documents):
    data = ""
    if documents is not None:
        for document in documents:
            document_details = {
                "filename": document.name,
                "filetype": document.type,
                "filesize": document.size
            }
            st.write(document_details)

            # Extract content from the txt file
            if document.type == "text/plain":
                # Read as bytes
                data += str(document.read(), "utf-8")

            # Extract content from the pdf file
            elif document.type == "application/pdf":
                # using PyPDF2
                # data += read_pdf(document)

                # using pdfplumber
                try:
                    with pdfplumber.open(document) as pdf:
                        all_text = ""
                        for page in pdf.pages:
                            text = page.extract_text()
                            all_text += text + "\n"
                        data += all_text
                except requests.exceptions.RequestException as e:
                    st.write("None")

            # Extract content from the docx file
            elif document.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
                data += docx2txt.process(document)

        # Display the extracted text content from file
        st.text_area("Extracted Text", value=data, height=200)
        # return extract status, and the data extracted
        return True, data
    
        

    else:
        st.error("Error: An error occurred while fetching content.")
        # return extract status, and the data extracted
        return False, data


def main():
    
    st.subheader("Extract Data from Documents")

    documents = st.file_uploader(
        "", type=["pdf", "txt", "docx"], accept_multiple_files=True
    )

    if "button_enter_doc" not in st.session_state:
        st.session_state.button_enter_doc = False

    if "extracted_doc" not in st.session_state:
        st.session_state.extracted_doc = False
    data = ""

    
    if st.button("Enter"):
        st.session_state.button_enter_doc = True
    
    

    # the enter button
    if st.session_state.button_enter_doc:
        # check if it is a sitemap or not
        if not documents:
            documents = None
        else:
            for doc in documents:
                if doc.name.split(".")[-1].lower() not in ["pdf", "txt", "docx"]:
                    # if documents is not the relevant type
                    st.error("Unsupported file: " + doc.name)

        st.session_state.extracted_doc, data = run_function(documents)

        if st.session_state.extracted_doc:
            col1, col2 = st.columns([0.5, 0.5])
            with col1:
                saved_button = False

                if st.download_button(
                    label="Save",
                    data=data
                ):
                    saved_button = True

            with col2:
                if st.button("Clear"):
                    st.session_state.button_enter_doc = False
                    st.session_state.extracted_doc = False
                    st.experimental_rerun()

            if saved_button:
                # Confirmation message
                st.success(f"File saved successfully.")

        else:
            st.warning("Data not extracted")
            if st.button("clear"):
                st.session_state.button_enter_doc = False
                st.session_state.extracted_doc = False
                st.experimental_rerun()
            
            
                


    

    # Add a success message to the sidebar
    st.sidebar.success("Select a page above.")

    # importing the custom footer from utils
    cust_footer()


if __name__ == "__main__":
    main()