import streamlit as st
from utils import *



def main():
  

    st.set_page_config(page_title="pdf invoice Extraction Bot")
    st.title("I am Bot. How can I help you today?... ")
    st.subheader("I can help you extract text from pdfs. You can download CSV as an output and Aduio2Text.")


    # Upload the Invoices (pdf files)...
    pdf = st.file_uploader("Upload pdf invoices here for now, only PDF files allowed for Now", type=["pdf"],accept_multiple_files=True)

    submit=st.button("Extract Data")

    if submit:
        with st.spinner('Wait for it...'):
            df=create_docs(pdf)
            st.write(df.head())

            data_as_csv= df.to_csv(index=False).encode("utf-8")
            
            st.download_button(
                "Download data as CSV", 
                data_as_csv, 
                "benchmark-tools.csv",
                "text/csv",
                key="download-tools-csv",
            )
        st.success("Hope I was able to save your time❤️")



#Invoking main function
if __name__ == '__main__':
    main()