app.py
|
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 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import streamlit as st from pdf2image import convert_from_bytes from PIL import Image from PyPDF2 import PdfReader def read_pdf_page(file, page_number): pdfReader = PdfReader(file) page = pdfReader.pages[page_number] return page.extract_text() def on_text_area_change(): st.session_state.page_text = st.session_state.my_text_area def main(): st.set_page_config(page_title="PDF Upload and Display") st.title("PDF Upload and Display") # PDF file upload pdf_file = st.file_uploader("Upload a PDF file", type=["pdf"]) if pdf_file: # Create a selectbox to choose the page number pdfReader = PdfReader(pdf_file) page_numbers = list(range(1, len(pdfReader.pages)+1)) selected_page = st.selectbox("Select a page", page_numbers) selected_page -= 1 # Convert the selected page to an image images = convert_from_bytes(pdf_file.getvalue()) image = images[selected_page] # Create two columns to display the image and text col1, col2 = st.columns(2) # Display the image in the first column col1.image(image, caption=f"Page {selected_page + 1}") # Display the content of the selected page in the second column # if "page_text" not in st.session_state: # st.session_state.page_text = col2.text_area("Page Text", height=800, value=read_pdf_page(pdf_file, selected_page), key="my_text_area", on_change=on_text_area_change) if __name__ == '__main__': main() |
