app.py

import os
import io
import json
import zipfile
from flask import Flask, render_template, request, send_file
from pypdf import PdfReader, PdfWriter

app = Flask(__name__)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/process', methods=['POST'])
def process():
    try:
        # Get metadata (JSON string mapping filenames to settings)
        metadata_str = request.form.get('metadata')
        if not metadata_str:
            return "No metadata provided", 400
        
        metadata = json.loads(metadata_str)
        # metadata is a dict: { "original_filename": { "newName": "...", "password": "..." } }

        # Create a ZIP file in memory
        memory_file = io.BytesIO()
        with zipfile.ZipFile(memory_file, 'w') as zf:
            
            uploaded_files = request.files.getlist('files')
            
            for file in uploaded_files:
                original_name = file.filename
                if original_name not in metadata:
                    continue
                
                settings = metadata[original_name]
                new_name = settings.get('newName') or original_name
                password = settings.get('password')
                
                # Ensure new_name has .pdf extension
                if not new_name.lower().endswith('.pdf'):
                    new_name += '.pdf'

                # Process PDF
                try:
                    reader = PdfReader(file)
                    writer = PdfWriter()

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

                    # Encrypt if password provided
                    if password:
                        writer.encrypt(password)

                    # Write to a bytes buffer
                    pdf_bytes = io.BytesIO()
                    writer.write(pdf_bytes)
                    pdf_bytes.seek(0)

                    # Add to ZIP
                    zf.writestr(new_name, pdf_bytes.getvalue())
                    
                except Exception as e:
                    print(f"Error processing {original_name}: {e}")
                    # Optionally handle error (skip file or add error log)

        memory_file.seek(0)
        return send_file(
            memory_file,
            mimetype='application/zip',
            as_attachment=True,
            download_name='encrypted_pdfs.zip'
        )

    except Exception as e:
        return str(e), 500

if __name__ == '__main__':
    app.run(debug=True, port=5000)