#from backend import ResultPipeline
import streamlit as st
from backend import InvoicePipeline
def main():
    st.set_page_config(page_title="Bill App")
    st.title("Bill Extractor")

    #This section handle upload of files
    files = st.file_uploader("Upload the file here:", type=["pdf"], accept_multiple_files=True)
    submit = st.button("Extract")

    # If user submitted the files, we need to call our pipeline
    if submit:
        with st.spinner("Please wait, while we processing your information..."):
            pipe = InvoicePipeline(files)
            df_results = pipe.run()
            st.write(df_results)
            convert_to_csv = df_results.to_csv(index = False).encode("utf-8")
            st.download_button(
                "Download",
                convert_to_csv,
                "bill.csv",
                "text/csv",
                key = "download-csv"
            
        )
         

# Calling the main function now
if __name__ == "__main__":
    main()