added PDF Merger

Dockerfile

@@ -5,6 +5,7 @@ FROM python:3.11-slim
 WORKDIR /app
 RUN mkdir /app/uploads
 RUN mkdir /app/split
+RUN mkdir /app/merge
 RUN apt-get update
 
 # install dependencies

app.py

@@ -1,26 +1,32 @@
-from flask import Flask, render_template, request, redirect, url_for, jsonify, send_from_directory
-import os
-from PyPDF2 import PdfReader, PdfWriter
-
-from pathlib import Path
 import shutil
 import os
+from flask import Flask, render_template, request, redirect, jsonify, send_from_directory
+from PyPDF2 import PdfReader, PdfWriter
+from pathlib import Path
 
 app = Flask(__name__)
 
 UPLOAD_FOLDER = 'uploads'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
+
 @app.route('/split/<path:path>')
 def send_report(path):
     return send_from_directory('split', path)
 
-@app.route('/split', methods=['POST'])
+
-def split_file():
+@app.route('/merge/<path:path>')
+def send_merge(path):
+    return send_from_directory('merge', path)
+
+
+@app.route('/split_to_zip', methods=['POST'])
+def split_to_zip():
     if 'pdf' not in request.files:
         return redirect(request.url)
 
@@ -34,14 +40,11 @@ def split_file():
         filename = os.path.join(app.config['UPLOAD_FOLDER'], pdf_file.filename)
         pdf_file.save(filename)
 
-
         out_filenames = []
         Path("/tmp/split_pdf").mkdir(parents=True, exist_ok=True)
-        # Process the PDF file (e.g., extract text)
         with open(filename, 'rb') as pdf_file:
             pdf_reader = PdfReader(pdf_file)
             num_pages = len(pdf_reader.pages)
-            text = ''
             for page_num in range(num_pages):
 
                 writer = PdfWriter()
@@ -63,5 +66,50 @@ def split_file():
         # response.headers.add("Access-Control-Allow-Origin", "*")
         return response
 
+
+@app.route('/merge_to_pdf', methods=['POST'])
+def merge_to_pdf():
+    if 'pdf_1' not in request.files or 'pdf_2' not in request.files:
+        return redirect(request.url)
+
+    pdf_file_1 = request.files['pdf_1']
+    pdf_file_2 = request.files['pdf_2']
+
+    if pdf_file_1.filename == '' or pdf_file_2.filename == '':
+        return redirect(request.url)
+
+    if pdf_file_1:
+        filename_1 = os.path.join(app.config['UPLOAD_FOLDER'], pdf_file_1.filename)
+        pdf_file_1.save(filename_1)
+
+    if pdf_file_2:
+        filename_2 = os.path.join(app.config['UPLOAD_FOLDER'], pdf_file_2.filename)
+        pdf_file_2.save(filename_2)
+
+    if pdf_file_1 and pdf_file_2:
+
+        with open(filename_1, 'rb') as pdf_file_1, open(filename_2, 'rb') as pdf_file_2:
+            pdf_reader_1 = PdfReader(pdf_file_1)
+            pdf_reader_2 = PdfReader(pdf_file_2)
+
+            Path("/tmp/merge_pdf").mkdir(parents=True, exist_ok=True)
+            writer = PdfWriter()
+
+            for page_num in range(len(pdf_reader_1.pages)):
+                writer.add_page(pdf_reader_1.pages[page_num])
+
+            for page_num in range(len(pdf_reader_2.pages)):
+                writer.add_page(pdf_reader_2.pages[page_num])
+
+            out_filename = '/app/merge/merger.pdf'
+            with open(out_filename, 'wb') as outfile:
+                writer.write(outfile)
+
+            response = jsonify({"url": '/merge/merger.pdf', "name": 'merge.pdf'})
+            # response.headers.add("Access-Control-Allow-Origin", "*")
+            return response
+
+
 if __name__ == '__main__':
+
     app.run(debug=True)

static/js/api_handler.js

@@ -0,0 +1,53 @@
+function downloadURI(uri, name) {
+    var link = document.createElement("a");
+    link.download = name;
+    link.href = uri;
+    document.body.appendChild(link);
+    link.click();
+    document.body.removeChild(link);
+    delete link;
+}
+
+function uploadPDF_split() {
+    const fileInput = document.getElementById('split_pdfFile');
+    const file = fileInput.files[0];
+    const formData = new FormData();
+    formData.append('pdf', file);
+
+    const backendURL = '/split_to_zip';
+
+    fetch(backendURL, {
+        method: 'POST',
+        body: formData
+    })
+    .then(response => response.json())
+    .then(data => {
+        console.debug("data from Backend: ", data)
+        downloadURI(data["url"], data["name"]);
+    })
+    .catch(error => console.error(error));
+}
+
+
+function uploadPDF_merge() {
+    const fileInput_1 = document.getElementById('merge_pdfFile_1');
+    const file_1 = fileInput_1.files[0];
+    const fileInput_2 = document.getElementById('merge_pdfFile_2');
+    const file_2 = fileInput_2.files[0];
+    const formData = new FormData();
+    formData.append('pdf_1', file_1);
+    formData.append('pdf_2', file_2);
+
+    const backendURL = '/merge_to_pdf';
+
+    fetch(backendURL, {
+        method: 'POST',
+        body: formData
+    })
+    .then(response => response.json())
+    .then(data => {
+        console.debug("data from Backend: ", data)
+        downloadURI(data["url"], data["name"]);
+    })
+    .catch(error => console.error(error));
+}

templates/index.html

@@ -3,43 +3,17 @@
 <head>
     <title>PDF Splitter</title>
     <link rel="shortcut icon" href="/static/icons/filetype-pdf.svg" type="image/x-icon">
+    <script src="/static/js/api_handler.js"></script> 
 </head>
 <body>
     <h1>PDF Splitter</h1>
-    <input type="file" id="pdfFile" accept=".pdf"><br>
+    <input type="file" id="split_pdfFile" accept=".pdf"><br>
-    <button onclick="uploadPDF()">Split PDF into individual pages</button>
+    <button onclick="uploadPDF_split()">Split PDF into individual pages</button>
+
+    <h1>PDF Merger</h1>
+    <input type="file" id="merge_pdfFile_1" accept=".pdf"><br>
+    <input type="file" id="merge_pdfFile_2" accept=".pdf"><br>
+    <button onclick="uploadPDF_merge()">Merge the two selected PDFs</button>
     <div id="output"></div>
-
-    <script>
-        function downloadURI(uri, name) {
-            var link = document.createElement("a");
-            link.download = name;
-            link.href = uri;
-            document.body.appendChild(link);
-            link.click();
-            document.body.removeChild(link);
-            delete link;
-        }
-
-        function uploadPDF() {
-            const fileInput = document.getElementById('pdfFile');
-            const file = fileInput.files[0];
-            const formData = new FormData();
-            formData.append('pdf', file);
-
-            const backendURL = '/split';
-
-            fetch(backendURL, {
-                method: 'POST',
-                body: formData
-            })
-            .then(response => response.json())
-            .then(data => {
-                console.debug("data from Backend: ", data)
-                downloadURI(data["url"], data["name"]);
-            })
-            .catch(error => console.error(error));
-        }
-    </script>
 </body>
 </html>