# This is a script that replaces the address or any other text from PDF with another string
# in this use case, two lines need to be find and modify separetly, so two conditionals are made
# the more lines, the script needs to be adapted to have more conditionals. 


import fitz  # PyMuPDF
import os  # we need this to read all files in a directory

# These are the available styles that the flitz library can take
# ---------------------------------------------------------------
# Times-Roman, Times-Italic, Times-Bold, Times-BoldItalic
# Helvetica, Helvetica-Oblique, Helvetica-Bold, Helvetica-BoldOblique
# Courier, Courier-Oblique, Courier-Bold, Courier-BoldOblique
# Symbol, ZapfDingbats
# ---------------------------------------------------------------

# get the list of all pdf files in the replace directory
files = [f for f in os.listdir("./replace") if f.endswith(".pdf")]

for file_name in files:
    file_path = "./replace/" + file_name
    doc = fitz.open(file_path)

    for page_num in range(len(doc)):
        page = doc[page_num]
        areas_line1 = page.search_for("32 N Gould St") # This the first line that the script will look for
        areas_line2 = page.search_for("Sheridan, WY 82801") # This the second line that the script will look for
        areas = areas_line1 + areas_line2  # combine areas from both lines

        for area in areas:
            # Add rectangle with the same color as the background to hide old text, Ik its cheap, but it is what it is
            rect = fitz.Rect(area)
            shape = page.new_shape()
            shape.draw_rect(rect)
            shape.finish(fill=(1, 1, 1), color=(1, 1, 1), stroke_opacity=0)
            shape.commit()

        # Add new text, broken into two parts
        if areas_line1:  # if there is a first line to replace
            x1, y1, x2, y2 = areas_line1[0]  # coordinates for the first line of text

            # These are the parameters of the new text
            # --------------------------------------------------------------------
            fontsize = 11  # Change as needed to match  the 

            font = "Helvetica-Bold"  # Style, change as needed

            color = (0, 0, 0)  # black, change as needed

            padding = 9 # This is the amount of padding that both lines will have, change as needed
            # --------------------------------------------------------------------

            text1 = "5337 N. Socrum Loop Road Unit 406" # First line that will match first find
            # Insert the first line of text
            page.insert_text((x1, y1 + padding), text1, fontsize=fontsize, color=color, fontname=font)



# ----------------------------------------------------------------------------



        if areas_line2:  # if there is a second line to replace
            x1, y1, x2, y2 = areas_line2[0]  # coordinates for the second line of text

            text2 = "Lakeland, FL 33809" # Second line that will match second find

            # Insert the second line of text
            page.insert_text((x1, y1 + padding), text2, fontsize=fontsize, color=color, fontname=font)

    # save the modified pdf to newfiles directory with the same file name
    doc.save("./newfiles/" + file_name)

# This script has no validation for repeated file names, so make sure to handle files correctly