Update taskweaver/ext_role/image_reader/image_reader.py

Co-authored-by: Copilot <[email protected]>

Author: liqul

taskweaver/ext_role/image_reader/image_reader.py

@@ -23,10 +23,16 @@ def local_image_to_data_url(image_path):
     if mime_type is None:
         mime_type = "application/octet-stream"  # Default MIME type if none is found
 
-    # Read and encode the image file
-    with open(image_path, "rb") as image_file:
-        base64_encoded_data = base64.b64encode(image_file.read()).decode("utf-8")
-
+    try:
+        # Read and encode the image file
+        with open(image_path, "rb") as image_file:
+            base64_encoded_data = base64.b64encode(image_file.read()).decode("utf-8")
+    except FileNotFoundError:
+        print(f"Error: The file {image_path} does not exist.")
+        return None
+    except IOError:
+        print(f"Error: The file {image_path} could not be read.")
+        return None
     # Construct the data URL
     return f"data:{mime_type};base64,{base64_encoded_data}"