Create utils.py

utils.py

@@ -0,0 +1,20 @@
+import fitz  # PyMuPDF
+import cv2
+import numpy as np
+import os
+import uuid
+
+def convert2img(pdf_file):
+    doc = fitz.open(stream=pdf_file.read(), filetype="pdf")
+    page = doc.load_page(0)
+    pix = page.get_pixmap(dpi=150)
+    img = np.frombuffer(pix.samples, dtype=np.uint8).reshape(pix.height, pix.width, pix.n)
+    if img.shape[2] == 4:
+        img = cv2.cvtColor(img, cv2.COLOR_RGBA2BGR)
+    return img
+
+def save_image(img):
+    filename = f"{uuid.uuid4().hex}.png"
+    path = os.path.join("/tmp", filename)
+    cv2.imwrite(path, img)
+    return filename