hjxdl 0.1.78__py3-none-any.whl → 0.1.80__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.78'
-__version_tuple__ = version_tuple = (0, 1, 78)
+__version__ = version = '0.1.80'
+__version_tuple__ = version_tuple = (0, 1, 80)

hdl/utils/llm/vis.py

@@ -2,6 +2,7 @@ from pathlib import Path
 import json
 import base64
 from io import BytesIO
+import requests
 
 import torch
 import numpy as np
@@ -25,6 +26,43 @@ __all__ = [
 
 HF_HUB_PREFIX = "hf-hub:"
 
+import requests
+import base64
+from io import BytesIO
+from PIL import Image
+
+
+def imgurl_to_base64(image_url: str):
+    """Converts an image from a URL to base64 format.
+
+    Args:
+        image_url (str): The URL of the image.
+
+    Returns:
+        str: The image file converted to base64 format with appropriate MIME type.
+    """
+    # Send a GET request to fetch the image from the URL
+    response = requests.get(image_url)
+
+    # Ensure the request was successful
+    if response.status_code == 200:
+        # Read the image content from the response
+        img_data = response.content
+
+        # Load the image using PIL to determine its format
+        img = Image.open(BytesIO(img_data))
+        img_format = img.format.lower()  # Get image format (e.g., jpeg, png)
+
+        # Determine the MIME type based on the format
+        mime_type = f"image/{img_format}"
+
+        # Convert the image data to base64
+        img_base64 = f"data:{mime_type};base64," + base64.b64encode(img_data).decode('utf-8')
+
+        return img_base64
+    else:
+        raise Exception(f"Failed to retrieve image from {image_url}, status code {response.status_code}")
+
 
 def imgfile_to_base64(img_dir: str):
     """Converts an image file to base64 format, supporting multiple formats.
@@ -175,10 +213,10 @@ class ImgHandler:
         images_fixed = []
         for img in images:
             if isinstance(img, str):
-                if Path(img).is_file():
-                    images_fixed.append(Image.open(img))
-                elif img.startswith("data:image"):
+                if img.startswith("data:image"):
                     images_fixed.append(imgbase64_to_pilimg(img))
+                elif Path(img).is_file():
+                    images_fixed.append(Image.open(img))
             elif isinstance(img, Image.Image):
                 images_fixed.append(img)
             else:

{hjxdl-0.1.78.dist-info → hjxdl-0.1.80.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hjxdl
-Version: 0.1.78
+Version: 0.1.80
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu

{hjxdl-0.1.78.dist-info → hjxdl-0.1.80.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=NwEVa6X7Fv67GF45wK1zLhWgrV9Uc642OSmWE-DZQw4,413
+hdl/_version.py,sha256=vi0UG5RBPxIXCmwjdgS_LYJzAu3zej2dH5Q8A1c13hg,413
 hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
 hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -131,12 +131,12 @@ hdl/utils/llm/chat.py,sha256=sk7Lw5Oa30k-l2fnJknkMmTc5zkBeEKsR981aeFhH5s,11907
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
 hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
-hdl/utils/llm/vis.py,sha256=mXJxKHETUKDydauuvv2FM9aQaxvEImA6S2wl4P4rIqI,14528
+hdl/utils/llm/vis.py,sha256=2ZxOE8zvGASg_QputCgnvFcKKln8KsfeJI0SLXCLos8,15697
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
 hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
-hjxdl-0.1.78.dist-info/METADATA,sha256=SZdVCIjTKkNCR60Ngi3VZ-DFzu6vvTKy0VuIW0fEXsA,903
-hjxdl-0.1.78.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-hjxdl-0.1.78.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.78.dist-info/RECORD,,
+hjxdl-0.1.80.dist-info/METADATA,sha256=S2Mz7p9uRyhNYpdFImolX_VAc2coeIn7zOe9E2z4e08,903
+hjxdl-0.1.80.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+hjxdl-0.1.80.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.80.dist-info/RECORD,,