Update handler.py

handler.py

@@ -21,27 +21,34 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         logging.error(f"----------This is an error message {str(data)}")
-        input_data = data.get("inputs", {})
-        logging.warning(f"------input_data-- {str(input_data)}")
-       # encoded_images = input_data.get("images")
-        #logging.warning(f"---encoded_images----- {str(encoded_images)}")
-        if not input_data:
-            #logging.warning(f"---encoded_images--not provided in if block--- {str(encoded_images)}")
-            return {"captions": [], "error": "No images provided"} 
+        raw_images = data.get("inputs", {})
+    #     logging.warning(f"------input_data-- {str(input_data)}")
+    #     encoded_images = input_data.get("images")
+    #     # Convert image to bytes
+    #     image = Image.open(encoded_images[0])
+    #     image_bytes = image.tobytes()
+    #     logging.warning(f"---image_bytes----- {str(image_bytes)}")
+    #     # Encode image bytes as base64
+    #     image_base64 = base64.b64encode(image_bytes).decode("utf-8")
+    #     #logging.warning(f"---encoded_images----- {str(image_base64)}")
+    #  #   print("000--------", image_base64)
+    #     if not encoded_images:
+    #         logging.warning(f"---encoded_images--not provided in if block--- {str(encoded_images)}")
+    #         return {"captions": [], "error": "No images provided"} 
         try:
-            logging.warning(f"---encoded_images-- provided in try block--- {str(input_data)}")
-            byteImgIO = io.BytesIO()
+            # logging.warning(f"---encoded_images-- provided in try block--- {str(encoded_images)}")
+            # byteImgIO = io.BytesIO()
             
-            byteImg = Image.open(input_data)
-            byteImg.save(byteImgIO, "PNG")
-            byteImgIO.seek(0)
-            byteImg = byteImgIO.read()
+            # byteImg = Image.open(encoded_images[0])
+            # byteImg.save(byteImgIO, "PNG")
+            # byteImgIO.seek(0)
+            # byteImg = byteImgIO.read()
 
 
-            # Non test code
-            dataBytesIO = io.BytesIO(byteImg)
-            raw_images =[Image.open(dataBytesIO)]
-            logging.warning(f"----raw_images----0--- {str(raw_images)}")
+            # # Non test code
+            # dataBytesIO = io.BytesIO(byteImg)
+            # raw_images =[Image.open(dataBytesIO)]
+            # logging.warning(f"----raw_images----0--- {str(raw_images)}")
             # Check if any images were successfully decoded
             if not raw_images:
                 print("No valid images found.")