Spaces:
Build error
Build error
File size: 2,484 Bytes
9afab0a |
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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import streamlit as st
import pytesseract
from pdf2image import convert_from_path
import os
import tempfile
import base64
# Set page config
st.set_page_config(page_title="PDF Text Extractor", layout="wide")
# Function to convert PDF pages to images
def pdf_to_images(pdf_path, dpi=300):
return convert_from_path(pdf_path, dpi=dpi)
# Function to extract text from images
def extract_text_from_images(images, progress_bar):
extracted_text = ""
for i, image in enumerate(images):
text = pytesseract.image_to_string(image)
extracted_text += f"Page {i+1}:\n{text}\n\n"
progress_bar.progress((i + 1) / len(images))
return extracted_text
# Function to create a download link
def get_download_link(text, filename, link_text):
b64 = base64.b64encode(text.encode()).decode()
return f'<a href="data:file/txt;base64,{b64}" download="{filename}">{link_text}</a>'
# Main Streamlit app
def main():
st.title("PDF Text Extractor")
st.sidebar.header("Settings")
dpi = st.sidebar.slider("DPI (Higher for better quality, slower processing)", 100, 600, 300, 50)
uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
if uploaded_file is not None:
st.success("File successfully uploaded!")
# Save uploaded file temporarily
with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
tmp_file.write(uploaded_file.getvalue())
tmp_file_path = tmp_file.name
try:
with st.spinner("Converting PDF to images..."):
images = pdf_to_images(tmp_file_path, dpi)
st.info(f"Total pages: {len(images)}")
progress_bar = st.progress(0)
status_text = st.empty()
with st.spinner("Extracting text from images..."):
status_text.text("Extracting text... (This may take a while)")
extracted_text = extract_text_from_images(images, progress_bar)
status_text.text("Text extraction complete!")
st.subheader("Extracted Text")
st.text_area("", extracted_text, height=300)
st.markdown(get_download_link(extracted_text, "extracted_text.txt", "Download Extracted Text"), unsafe_allow_html=True)
except Exception as e:
st.error(f"An error occurred: {str(e)}")
finally:
# Clean up the temporary file
os.unlink(tmp_file_path)
if __name__ == "__main__":
main() |