From d583a30e4815cfd3ecec66351ecc0dff275a6e8f Mon Sep 17 00:00:00 2001
From: khaangnguyeen <khaangnguyeen@gmail.com>
Date: Mon, 3 Mar 2025 16:31:21 +0700
Subject: [PATCH] Update util.py

---
 easyapi/util.py | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/easyapi/util.py b/easyapi/util.py
index c9eba68..2da1ec9 100644
--- a/easyapi/util.py
+++ b/easyapi/util.py
@@ -54,11 +54,26 @@ def image_to_base64(pli_image, pnginfo=None):
 
 
 def read_image_from_url(image_url):
-    s = requests.Session()
-    s.keep_alive = False
-    response = s.get(image_url, verify=False)
-    img = Image.open(io.BytesIO(response.content))
-    return img
+    try:
+        # Create a new session and disable keep-alive if desired
+        session = requests.Session()
+        session.keep_alive = False
+
+        # Get the image content from the URL
+        response = session.get(image_url, stream=True, verify=False)
+        response.raise_for_status()  # Ensure we got a valid response
+
+        # Convert the response content into a BytesIO object
+        image_bytes = io.BytesIO(response.content)
+        
+        # Open the image using PIL and force loading the image data
+        img = Image.open(image_bytes)
+        img.load()  # Ensure the image is fully loaded
+        
+        return img
+    except Exception as e:
+        print(f"Error reading image from URL {image_url}: {e}")
+        return None
 
 
 def hex_to_rgba(hex_color):