How to Extract Text and Images from a PDF File

PDFs (Portable Document Format) is a widely used file format for documents, eBooks, and other files. However, extracting text and images from a PDF can be challenging, especially when dealing with large PDF files or complex document layouts. In this tutorial, we’ll show you how to use Python to extract text and images from a PDF file.

Contact me

Find the GitHub repository of this tutorial at https://github.com/shamim-akhtar/extract-pdf-text-images.

Prerequisites

Before we start, you’ll need to install Python on your computer. If you haven’t installed Python, download it from the official website (https://www.python.org/downloads/). You’ll also need to install the following libraries:

PyPDF2: A library for working with PDF files.
fitz: A Python wrapper for the MuPDF PDF rendering library.

You can install both libraries using pip in your command prompt or terminal:

pip install PyPDF2 fitz

Extracting Text from a PDF

Let’s start by extracting the text from a PDF file using Python. Here’s the code:

import PyPDF2
import os

# Set the input and output paths
input_path = 'C:/path/to/input.pdf'
output_path = 'C:/path/to/output.txt'

# Open the PDF file
with open(input_path, 'rb') as pdf_file:
    # Read the PDF file
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)

    # Extract text from all pages
    text = ''
    for page in pdf_reader.pages:
        text += page.extract_text()

# Save the text to a file
with open(output_path, 'w', encoding='utf-8') as output_file:
    output_file.write(text)

Let’s go over the code step by step:

First, we import the PyPDF2 and os libraries.
We set the input and output paths. Replace the input_path variable with the file path of the PDF file that you want to extract text. Replace the output_path variable with the file path where you want to save the extracted text.
We open the PDF file using Python’s open function and the 'rb' mode, which stands for read binary.
We create a PdfFileReader object using PyPDF2. This object represents the PDF file that we just opened.
We use a for loop to iterate over all the pages in the PDF file.
For each page, we use the extract_text method to extract the text.
We append the extracted text to the text variable.
After all the pages have been processed, we save the extracted text to a file using Python’s open function and the 'w' mode, which stands for write.
We write the text variable to the output file using the write method.

That’s it! The extracted text should now be saved to the file specified by the output_path variable.

Extracting Images from a PDF

Next, let’s look at how to extract images from a PDF file using Python. Here’s the code:

import fitz
import os

# Set the input and output paths
input_path = 'C:/path/to/input.pdf'
output_path = 'C:/path/to/images'

# Open the PDF file with fitz
pdf_doc = fitz.open(input_path)

# Get a list of images on the first page of the PDF
first_page_images = pdf_doc.get_page_images(0)

# Save each image to a file
for image in first_page_images

Code Explanation

The code is a Python script (Jupyter Notebook) that extracts text and images from a PDF file. It imports three libraries: fitz, os, and PyPDF2. It then sets the input and output paths, opens the PDF file and reads it using PyPDF2, and extracts the text from the first page of the PDF. The extracted text is saved to a file in the specified output folder. The script then opens the PDF file again using fitz, gets a list of images on the first page of the PDF, saves each image to a file in the specified output folder, and prints the number of images detected on the first page of the PDF.

To use the code, you must replace the input_path variable with the PDF file path you want to extract text and images. You should also set the output_path variable to the folder where you want the output files to be saved. After running the Python script, it will extract the text and images from the PDF and save them to files in the specified output folder.

Note that the code is only designed to extract text and images from the first page of the PDF. To extract text and images from other pages, you must modify the code accordingly.

Prerequisites

Extracting Text from a PDF

Extracting Images from a PDF

Code Explanation

Read My Other Tutorials