Update handler.py

handler.py

@@ -6,6 +6,7 @@ import base64
 import io
 from transformers import BlipForConditionalGeneration, BlipProcessor
 import logging
+from io import BytesIO
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
@@ -23,36 +24,45 @@ class EndpointHandler():
         logging.error(f"----------This is an error message {str(data)}")
         input_data = data.get("inputs", {})
         logging.warning(f"------input_data-- {str(input_data)}")
-    #     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"} 
+        encoded_images = input_data.get("url")
+        print("url---",encoded_images)
+        # Convert image to bytes
+       # image = Image.open(encoded_images[0])
+        #url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg' 
+
+        # Send a GET request to the URL to get the image data
+        response = requests.get(encoded_images)
+
+        # Read the image data from the response
+        image_data = BytesIO(response.content)
+
+        #image_bytes = image.tobytes()
+       # img = Image.open(image_data)
+        #print("testing img--------------", img)
+        #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(encoded_images)}")
-            # byteImgIO = io.BytesIO()
+            logging.warning(f"---encoded_images-- provided in try block--- {str(encoded_images)}")
+            byteImgIO = io.BytesIO()
             
-            # byteImg = Image.open(encoded_images[0])
-            # byteImg.save(byteImgIO, "PNG")
-            # byteImgIO.seek(0)
-            # byteImg = byteImgIO.read()
+            byteImg = Image.open(image_data)
+            print("testing img---byteImg-----------", byteImg)
+            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
-            raw_images = [input_data]
-            logging.warning(f"---raw_images-raw_images----------- {str(raw_images)}")
             if not raw_images:
                 print("No valid images found.")
             processed_inputs = [
@@ -72,4 +82,4 @@ class EndpointHandler():
         except Exception as e:
             print(f"Error during processing: {str(e)}")
             logging.error(f"Error during processing: ----------------{str(e)}")
-            return {"captions": [], "error": str(e)}
+            return {"captions": [], "error": str(e)}