Spaces:
Running
Running
| from PIL import Image | |
| import os | |
| import PyPDF2 | |
| def merge_pdfs_from_folder(folder_path): | |
| """ | |
| Merges all PDF files in a given folder into a single PDF and saves it in the same folder. | |
| :param folder_path: Path to the folder containing PDF files. | |
| """ | |
| folder_name = os.path.basename(os.path.normpath(folder_path)) # Get folder name | |
| folder_name = folder_name.replace("output", "").strip("_- ") # Remove "output" and clean up | |
| output_filename = f"merged_{folder_name}.pdf" if folder_name else "merged.pdf" | |
| output_path = os.path.join(folder_path, output_filename) | |
| merger = PyPDF2.PdfMerger() | |
| # Get a list of all PDF files in the folder, sorted by name | |
| pdf_files = sorted([f for f in os.listdir(folder_path) if f.endswith(".pdf")]) | |
| if not pdf_files: | |
| print("No PDF files found in the folder.") | |
| return | |
| for pdf in pdf_files: | |
| pdf_path = os.path.join(folder_path, pdf) | |
| merger.append(pdf_path) | |
| print(f"Added: {pdf}") | |
| merger.write(output_path) | |
| merger.close() | |
| print(f"Merged PDF saved as: {output_path}") | |
| def images_to_pdf(image_folder, output_pdf): | |
| """Converts all images in a folder to a single PDF.""" | |
| images = [] | |
| for file in sorted(os.listdir(image_folder)): | |
| if file.lower().endswith(('png', 'jpg', 'jpeg', 'bmp', 'gif')): | |
| img_path = os.path.join(image_folder, file) | |
| img = Image.open(img_path).convert('RGB') | |
| images.append(img) | |
| if images: | |
| images[0].save(output_pdf, save_all=True, append_images=images[1:]) | |
| print(f"PDF saved as {output_pdf}") | |
| else: | |
| print("No images found in the folder.") | |
| def process_main_folder(main_folder, output_folder): | |
| """Processes each subfolder in the main folder and creates a PDF for each.""" | |
| if not os.path.exists(output_folder): | |
| os.makedirs(output_folder) | |
| for subfolder in sorted(os.listdir(main_folder)): | |
| subfolder_path = os.path.join(main_folder, subfolder) | |
| if os.path.isdir(subfolder_path): | |
| output_pdf = os.path.join(output_folder, f"{subfolder}.pdf") | |
| images_to_pdf(subfolder_path, output_pdf) | |
| def rename_folder_path(folder_path): | |
| """Renames the folder by adding 'output-' as a prefix if not already present.""" | |
| folder_name = os.path.basename(folder_path) # Get only the folder name | |
| parent_dir = os.path.dirname(folder_path) # Get the parent directory path | |
| # Ensure 'output-' is not duplicated | |
| new_folder_name = folder_name if folder_name.startswith("output-") else f"output-{folder_name}" | |
| return os.path.join(parent_dir, new_folder_name) # Return the new full path | |
| def main(main_folder): | |
| output_folder = rename_folder_path(main_folder) | |
| process_main_folder(main_folder, output_folder) | |
| merge_pdfs_from_folder(output_folder) | |
| return output_folder | |
| if __name__ == "__main__": | |
| main_folder = 'boarding-diary-uncensored' | |
| main(main_folder) |