Images to PDF

main.py

#!/bin/python3

import os
import sys
import shutil
import glob
from PIL import Image, ImageEnhance
import re
from fpdf import FPDF

origPath = 'orig'
finalPath = 'final'
split = 454

THRESH = 125
CONTRAST = 1.4

if(os.path.exists(finalPath)):
    shutil.rmtree(finalPath)
os.mkdir(finalPath)

imgs = glob.glob(origPath+'/*.JPG')
for img in imgs:
    basename = os.path.basename(img)
    print(basename)
    imgId = re.search('[0-9]{4}', basename)
    imgId = int(imgId.group(0))
    print(imgId)
    workingImg = Image.open(img)

    workingImg = workingImg.rotate(271, expand=True)
    workingImg = workingImg.crop((607, 855, 2726, 3646))

    width, height = workingImg.size
    newsize = (1000, int(1000 * (height/width)))
    workingImg = workingImg.resize(newsize)

    width, height = workingImg.size

    workingImg = ImageEnhance.Contrast(workingImg).enhance(CONTRAST)
    # workingImg = workingImg.convert('L')
    def fn(x): return 255 if x > THRESH else 0
    workingImg = workingImg.convert('L').point(fn, mode='1')

    finalBasename = basename
    if(imgId >= split):
        newPage = str(imgId - 38)
        finalBasename = 'IMG_'+newPage+'a.JPG'
    workingImg.save(finalPath+'/'+finalBasename, dpi=(150, 150))

pdf = FPDF(unit="pt", format=[width, height])

imgs = glob.glob(finalPath+'/*.JPG')
imgs.sort()
for img in imgs:
    pdf.add_page()
    pdf.image(img, 0, 0, width, height)
pdf.output(
    "October-Surprise-Decide-For-Yourself-Prevailing-Winds-Research.pdf", "F")