f

app.py

@@ -32,9 +32,15 @@ app.logger.setLevel(logging.DEBUG)
 # Create a queue to hold the files for processing
 file_queue = queue.Queue()
 
+image_queue = queue.Queue()
+
+lock = threading.Lock()
+
 # Directory to save uploaded files (you can modify this path)
 UPLOAD_FOLDER = './uploads'
+IMAGE_FOLDER = './images'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+os.makedirs(IMAGE_FOLDER, exist_ok=True)
 
 # Printer configuration
 printer_model = "QL-1100"
@@ -45,7 +51,6 @@ printer = 'usb://0x04f9:0x20a7'
 def convert_pdf_to_images(pdf_path):
     """Converts a PDF to a PIL Image"""
     try:
-        # Convert PDF to image (get the first page)
         images = pdf2image.convert_from_path(pdf_path)
         if images:
             return images
@@ -55,23 +60,19 @@ def convert_pdf_to_images(pdf_path):
         return None
 
 # Function to print address labels
-def print_address_label(images, pdf_path):
+def print_address_label(image, pdf_path):
     try:
-        processed_images = []
-        for image in images:
         if 'address' in pdf_path.lower():
             target_width = 1660
             target_height = 1164
             image = image.convert("RGB")
             image = image.resize((target_width, target_height), Image.LANCZOS)
             image = image.rotate(90, expand=True)
-                processed_images.append(image)
         else:
             target_width = 1164
             target_height = 1660
             image = image.convert("RGB")
             image = image.resize((target_width, target_height), Image.LANCZOS)
-                processed_images.append(image)
         
         qlr = BrotherQLRaster(printer_model)
         qlr.exception_on_warning = True
@@ -79,7 +80,7 @@ def print_address_label(images, pdf_path):
         app.logger.info("Converting image to printer instructions...")
         instructions = convert(
             qlr=qlr,
-            images=processed_images,
+            images=[image],
             label='102x152',
             threshold=70.0,
             dither=False,
@@ -103,8 +104,19 @@ def print_address_label(images, pdf_path):
     except Exception as e:
         app.logger.error(f"Error during printing: {str(e)}")
 
+def save_images(images, pdf_path):
+    uuid = str(int(time.time()))
+    for i,image in enumerate(images):
+        if 'address' in pdf_path.lower():
+            image.save(os.path.join(IMAGE_FOLDER, f"address_label_{uuid}_{i}.png"), "PNG")
+        elif 'packing' in pdf_path.lower():
+            image.save(os.path.join(IMAGE_FOLDER, f"shipping_label_{uuid}_{i}.png"), "PNG")
+        else:
+            image.save(os.path.join(IMAGE_FOLDER, f"unknown_{uuid}_{i}.png"), "PNG")
+        image_queue.put((image, pdf_path))
+
 # Worker thread that processes files from the queue
-def process_queue():
+def file_worker():
     while True:
         # Wait until a file is available in the queue
         file_path = file_queue.get()
@@ -114,15 +126,25 @@ def process_queue():
         app.logger.info(f"Processing file: {file_path}")
         images = convert_pdf_to_images(file_path)
         if images:
-            print_address_label(images, file_path)
+            save_images(images, file_path)
         else:
             app.logger.error(f"Failed to convert PDF to images: {file_path}")
         # Signal that the task is done
         file_queue.task_done()
 
+def printer_worker():
+    while True:
+        image, path = image_queue.get()
+        if image is None:
+            break
+        with lock:
+            print_address_label(image, path)
+        image_queue.task_done()
+
 # Start the worker thread
-thread = threading.Thread(target=process_queue, daemon=True)
-thread.start()
+threading.Thread(target=file_worker, daemon=True).start()
+# Start the printer worker thread
+threading.Thread(target=printer_worker, daemon=True).start()
 
 @app.route('/upload', methods=['POST'])
 def upload_file():