Spaces:
Sleeping
Sleeping
| # from backend import ResultPipeline | |
| import streamlit as st | |
| from backend import InvoicePipeline | |
| def main(): | |
| # Just for setting up the title & head bar | |
| st.set_page_config(page_title = "Bill App") | |
| st.title("Bill Extractor") | |
| # This section will handle uploadation of files | |
| files = st.file_uploader("Upload the files here..", type = ["pdf"], accept_multiple_files = True) | |
| submit = st.button("Extract") | |
| # If a user has submitted the fieles, we need to call our pipeline | |
| if submit: | |
| with st.spinner("Please wait, while we are 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, | |
| "bills.csv", | |
| "text/csv", | |
| key = "download-csv" | |
| ) | |
| # Calling the main function | |
| if __name__ == "__main__": | |
| main() |