File size: 1,040 Bytes
1108a3a |
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 34 35 36 37 38 |
import streamlit as st
from utils import *
def main():
st.set_page_config(page_title="Invoice Extractor")
st.title("Invoice Extractor Bot...")
st.subheader("I can help you in extracting invoice data")
# Upload the Invoices (pdf files)
pdf = st.file_uploader("Upload invoices here", type=["pdf"], accept_multiple_files=True)
submit = st.button("Extract Data")
if submit:
with st.spinner("Loading invoices..."):
df = create_docs(pdf)
# st.write('managed to extract data from pdf')
# st.write(df)
st.write(df.head())
# Saving the dataframe as CSV file
data_as_csv = df.to_csv(index=False).encode('utf-8')
st.download_button(
"Download CSV",
data_as_csv,
"invoice_extracted_data.csv",
"text/csv",
key="download-tools-csv",
)
st.success("Hope I was able to save your time <3")
if __name__ == '__main__':
main()
|