from pypdf import PdfReader, PdfWriter
import tkinter as tk
from tkinter import filedialog
import os

root = tk.Tk()
root.withdraw()  # Скрыть основное окно Tkinter

file_path = filedialog.askopenfilename()  # Открываем окно выбора файла
output_directory = filedialog.askdirectory()  # Открываем окно выбора директории для сохранения

original_filename = os.path.basename(file_path)
new_filename = os.path.join(output_directory, original_filename.split(".")[0] + "_compress.pdf")
try:
    quality = int(input("Введите желаемое quality: число не более 100: "))
except:
    print("ливни нахуй")

reader = PdfReader(file_path)
writer = PdfWriter()

for page in reader.pages:
    writer.add_page(page)

for page in writer.pages:
    for img in page.images:
        img.replace(img.image, quality=quality)

with open(new_filename, "wb") as f:
    writer.write(f)