Hemasagar's picture
Create app.py
c76a309 verified
raw
history blame
1.11 kB
import streamlit as st
from dotenv import load_dotenv
from utils import *
def main():
load_dotenv()
st.set_page_config(page_title="Invoice Extraction Bot")
st.title("Audio_to_text & Invoice Extraction Bot...💁 ")
st.subheader("I can help you in extracting text and invoice data and Aduio2text")
# Upload the Invoices (pdf files)...
pdf = st.file_uploader("Upload invoices here for now, only PDF files allowed and will accept other formate as well", 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()