import sort

main.py

@@ -1,16 +1,15 @@
+import math
 import os
-import sys
 import platform
-from fpdf import FPDF
-
-from PIL import Image
-from tkinter import filedialog
-from tkinter import Tk, Button, Label, Toplevel
-from tkinter.filedialog import askdirectory, askopenfiles
-from pypdf import PdfWriter, PdfReader, Transformation
-import math
 import random
 import subprocess
+import sys
+from tkinter import Button, Label, Tk, Toplevel, filedialog
+from tkinter.filedialog import askdirectory, askopenfiles
+
+from fpdf import FPDF
+from PIL import Image
+from pypdf import PdfReader, PdfWriter, Transformation
 
 
 def get_os():

test.py

@@ -1,14 +1,14 @@
 import os
+
 import fitz
 
 
 def add_watermask():
-    file_path = './pdfs/girl.pdf'
+    file_path = "./pdfs/girl.pdf"
-    water_path = './waters/logo.png'
+    water_path = "./waters/logo.png"
     baseaname = os.path.basename(file_path)
     name, suffix = os.path.splitext(baseaname)
-    out_path = os.path.join(os.path.dirname(file_path),
+    out_path = os.path.join(os.path.dirname(file_path), name + "_watered" + suffix)
-                            name + '_watered' + suffix)
     pdf_file = fitz.open(file_path)
     page = pdf_file[0]
     print(page.bound())