Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import base64
 import pdfplumber
 
-st.title("Extract Tables from PDFs")
+st.title("PDF Table Extractor")
 
 input_pdf = st.file_uploader(label="Upload PDF here", type='pdf')
 
@@ -17,16 +17,21 @@ if input_pdf is not None:
         f.write(base64.b64decode(base64_pdf))
 
     pdf = pdfplumber.open("input.pdf")
-    page = pdf.pages[int(page_number) - 1]
-    tables = page.extract_tables()
+    num_pages = len(pdf.pages)
 
-    st.markdown("## Number of Tables")
-    st.write(len(tables))
+    if not page_number.isdigit() or int(page_number) <= 0 or int(page_number) > num_pages:
+        st.error(f"Invalid page number. Please enter a number between 1 and {num_pages}.")
+    else:
+        page = pdf.pages[int(page_number) - 1]
+        tables = page.extract_tables()
 
-    if tables:
-        option = st.selectbox(label="Select the table to be displayed", options=list(range(1, len(tables) + 1)))
+        st.markdown("## Number of Tables")
+        st.write(len(tables))
 
-        st.markdown("### Output Table")
-        st.dataframe(tables[option - 1])
+        if tables:
+            option = st.selectbox(label="Select the table to be displayed", options=list(range(1, len(tables) + 1)))
 
-    pdf.close()
+            st.markdown("### Output Table")
+            st.dataframe(tables[option - 1])
+
+    pdf.close()