Update app.py

app.py

@@ -1,29 +1,32 @@
 import streamlit as st
 import os
 import base64
-import tabula
-
+import pdfplumber
 
 st.title("Extract Tables from PDFs")
 
-input_pdf = st.file_uploader(label="Upload PDF here",type='pdf')
+input_pdf = st.file_uploader(label="Upload PDF here", type='pdf')
 
 st.markdown("### Page Number")
 
-page_number = st.text_input("Enter the page # from where you want the table", value=1)
+page_number = st.text_input("Enter the page # from where you want the table", value='1')
 
 if input_pdf is not None:
-    with open("input.pdf","wb") as f:
+    with open("input.pdf", "wb") as f:
         base64_pdf = base64.b64encode(input_pdf.read()).decode('utf-8')
         f.write(base64.b64decode(base64_pdf))
-        
-    tables = tabula.read_pdf("input.pdf", pages=page_number, multiple_tables=True)
-    
+
+    pdf = pdfplumber.open("input.pdf")
+    page = pdf.pages[int(page_number) - 1]
+    tables = page.extract_tables()
+
     st.markdown("## Number of Tables")
     st.write(len(tables))
-    
+
     if tables:
         option = st.selectbox(label="Select the table to be displayed", options=list(range(1, len(tables) + 1)))
-        
+
         st.markdown("### Output Table")
         st.dataframe(tables[option - 1])
+
+    pdf.close()