| import tabula | |
| import pandas as pd | |
| import streamlit as st | |
| import os | |
| os.system('wget https://download.oracle.com/java/17/latest/jdk-17_linux-x64_bin.deb') | |
| os.system('sudo dpkg -i jdk-17_linux-x64_bin.deb') | |
| os.environ["JAVA_HOME"] = "/usr/lib/jvm/jdk-17/bin" | |
| st.title("TableOCR") | |
| st.markdown( | |
| """ | |
| <style> | |
| .css-1jc7ptx, .e1ewe7hr3, .viewerBadge_container__1QSob, | |
| .styles_viewerBadge__1yB5_, .viewerBadge_link__1S137, | |
| .viewerBadge_text__1JaDK { | |
| display: none; | |
| } | |
| </style> | |
| """, | |
| unsafe_allow_html=True | |
| ) | |
| with st.form(key='my_form'): | |
| file = st.file_uploader("Upload a file", type="pdf", accept_multiple_files=False) | |
| page_no = st.number_input("Enter page number", min_value=1, value=1) | |
| submit_button = st.form_submit_button(label='Submit') | |
| if submit_button and file is not None and page_no is not None: | |
| with st.spinner("Converting PDF page to image..."): | |
| tables = tabula.read_pdf(file, pages=page_no, multiple_tables=True) | |
| table_df = tables[0] if tables else pd.DataFrame() | |
| st.write("Scroll down to download the output file.") | |
| st.table(table_df) | |
| table_df.to_excel("output.xlsx", index=False) | |
| st.markdown( | |
| f'<a href="output.xlsx" download="output.xlsx">Click here to download the output file</a>', | |
| unsafe_allow_html=True | |
| ) |