simplified

handler.py

@@ -3,22 +3,10 @@ import base64
 import io
 import tempfile
 from PIL import Image
-from pdf2image import convert_from_path
 import logging
 
-def process_image(pdfdata):
-    prompt = "detect the accident diagram"
-    # PDF input -> convert to images -> find crash diagram
-    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
-        temp_file.write(pdfdata)
-        temp_file.flush()
-        print("temporary name:", temp_file.name)
-        images = convert_from_path(temp_file.name)
-
-    crash_img = None
-    for img in images:
-        crash_img = img.crop((0,0, 200, 200))
-        break
+def process_image(img):
+    crash_img = img.crop((0,0, 200, 200))
 
     if crash_img:
         img_io = io.BytesIO()
@@ -43,9 +31,10 @@ class EndpointHandler:
      def __call__(self, data: Any) -> List[List[Dict[str, str]]]:
          logging.warning("inside __call__")
          inputs = data.pop("inputs", data)
-         pdfdata = inputs.pop("filedata", inputs)
-         if isinstance(pdfdata, str):
-             logging.warning("decoding pdfdata")
-             pdfdata = base64.b64decode(pdfdata)
+         imagedata = inputs.pop("imagedata", inputs)
+         if isinstance(imagedata, str):
+            logging.warning("decoding pdfdata")
+            image_bytes = base64.b64decode(imagedata)
+            img = Image.open(io.BytesIO(image_bytes))
 
-         return process_image(pdfdata)
+            return process_image(img)

requirements.txt

@@ -1,2 +1 @@
-pdf2image==1.17.0
 pillow==10.4.0