File size: 1,065 Bytes
6b0f4ab bafa6f7 6b0f4ab bafa6f7 6b0f4ab bafa6f7 6b0f4ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
# 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() |