import gradio as gr
import fitz
import os


def append_suffix_to_filename(filepath, suffix):
    directory, filename = os.path.split(filepath)
    filename_without_extension, extension = os.path.splitext(filename)
    return os.path.join(directory, filename_without_extension + suffix + extension)


def remove_password(input_files, password):
    output_files = []
    for file in input_files:
        output_path = append_suffix_to_filename(file.name, '_open')
        try:
            doc = fitz.open(file.name)
            if doc.is_encrypted:
                doc.authenticate(password)
            doc.save(output_path)
            print(output_path)
            doc.close()
            output_files.append(output_path)
        except Exception as e:
            print(e)
            pass
    return output_files


def pdf_password_remover(files, password):
    output_files = []
    for file in files:
        output_files.extend(remove_password([file], password))
    return output_files


input_components = [
    gr.Files(label="Upload PDF(s)"),
    gr.Textbox(label="Password", type="text")
]

output_component = [gr.Files(label="Download Unlocked PDF(s)")]

gr.Interface(
    fn=pdf_password_remover,
    inputs=input_components,
    outputs=output_component,
    title="PDF Password Remover",
    description="Upload password-protected PDF files and input a single password to unlock them."
).launch(debug=True)