hjxdl 0.2.90__py3-none-any.whl → 0.2.92__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.90'
-__version_tuple__ = version_tuple = (0, 2, 90)
+__version__ = version = '0.2.92'
+__version_tuple__ = version_tuple = (0, 2, 92)

hdl/utils/llm/chat.py

@@ -6,14 +6,16 @@ import subprocess
 from typing import Generator
 import re
 
-
 from openai import OpenAI
+from PIL import Image
+
 from ..desc.template import FN_TEMPLATE, COT_TEMPLATE, OD_TEMPLATE
 from ..desc.func_desc import TOOL_DESC
-from .vis import draw_and_plot_boxes_from_json, to_img
+from .vis import draw_and_plot_boxes_from_json, to_img, to_base64
 # import json
 # import traceback
 
+
 def parse_fn_markdown(markdown_text, params_key="params"):
     """
     Parses a markdown text to extract function name and parameters.
@@ -585,8 +587,7 @@ class OpenAI_M():
         Returns:
             str: A JSON string containing the results of the object detection.
         """
-        if isinstance(image, str):
-            image = to_img(image)
+        image = to_base64(image)
         json_str = self.invoke(
             prompt=self.od_desc,
             images=[image]

hdl/utils/llm/vis.py

@@ -30,6 +30,19 @@ from redis.commands.search.query import Query
 HF_HUB_PREFIX = "hf-hub:"
 
 def to_img(img_str):
+    """
+    Convert an image source string to a PIL Image object.
+    The function supports three types of image sources:
+    1. Base64 encoded image strings starting with "data:image".
+    2. URLs starting with "http".
+    3. Local file paths.
+    Args:
+        img_str (str): The image source string. It can be a base64 encoded string, a URL, or a local file path.
+    Returns:
+        PIL.Image.Image: The converted image as a PIL Image object.
+    Raises:
+        ValueError: If the image source string is not valid or the image cannot be loaded.
+    """
     if img_str.startswith("data:image"):
         img = imgbase64_to_pilimg(img_str)
     elif img_str.startswith("http"):
@@ -45,6 +58,30 @@ def to_img(img_str):
     return img
 
 
+def to_base64(img):
+    """
+    Convert an image to its base64 representation.
+    Parameters:
+    img (PIL.Image.Image or str): The image to convert. This can be a PIL Image object,
+                                  a base64 string starting with "data:image", a URL starting
+                                  with "http", or a file path to an image.
+    Returns:
+    str: The base64 representation of the image.
+    Raises:
+    TypeError: If the input is not a PIL Image object or a string.
+    """
+    if isinstance(img, Image.Image):
+        img_base64 = pilimg_to_base64(img)
+    elif isinstance(img, str):
+        if img.startswith("data:image"):
+            img_base64 = img
+        elif img.startswith("http"):
+            img_base64 = imgurl_to_base64(img)
+        elif Path(img).is_file():
+            img_base64 = imgfile_to_base64(img)
+    return img_base64
+
+
 def imgurl_to_base64(image_url: str):
     """Converts an image from a URL to base64 format.
 
@@ -177,6 +214,8 @@ def draw_and_plot_boxes_from_json(
     # except FileNotFoundError:
     #     print(f"Image file not found at {image_path}. Please check the path.")
     #     return None
+    if not isinstance(image, Image.Image):
+        image = to_img(image)
     img = image
 
     draw = ImageDraw.Draw(img)

{hjxdl-0.2.90.dist-info → hjxdl-0.2.92.dist-info}/METADATA

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

{hjxdl-0.2.90.dist-info → hjxdl-0.2.92.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=dsW-sD-GWSmrnvsjSNdPvgFmdtr0NxczFjTqn2lljPU,413
+hdl/_version.py,sha256=kkEuGTn2hmLTPDGbYFNi0KFL-GFSm6bEFKYCUrPee_8,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
@@ -128,18 +128,18 @@ hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 hdl/utils/general/glob.py,sha256=Zuf7WHU0UdUPOs9UrhxmrCiMC8GrHxQU6n3mTThv6yc,1120
 hdl/utils/general/runners.py,sha256=x7QBolp3MrqNV6L4rB6Ueybr26bqkRFZTuXhY0SwyLk,3061
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=FykxgrUlolJvG5NN0sDlTmVf2F2kDWp1M1iodVNFeUg,24850
+hdl/utils/llm/chat.py,sha256=_yqgWFhsxtYfNNm8pb5b-HIl950ZHL841JM5qm7zd0Y,24848
 hdl/utils/llm/chatgr.py,sha256=5F5PJHe8vz3iCfi4TT54DCLRi1UeJshECdVtgvvvao0,3696
 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=fuXdEFfUpW4KR7TKmIBEuMmo3ZdZxfd79a_qfTM0tPk,24739
+hdl/utils/llm/vis.py,sha256=GkL7UMQxosTekhOZXSj_FTKGmZ1v5WciXOabn55FnWQ,26242
 hdl/utils/llm/visrag.py,sha256=0i-VrxqgiV-J7R3VPshu9oc7-rKjFJOldYik3HDXj6M,10176
 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.2.90.dist-info/METADATA,sha256=zXS1Zo2oeR0MqDgUOwvYGU_RxOLS75wdfsjRNlz4q64,836
-hjxdl-0.2.90.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-hjxdl-0.2.90.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.90.dist-info/RECORD,,
+hjxdl-0.2.92.dist-info/METADATA,sha256=3KY26P4eZG7AreivWqsYxq1Fs1J2GU_BjkGKB-WF2sI,836
+hjxdl-0.2.92.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+hjxdl-0.2.92.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.2.92.dist-info/RECORD,,