hjxdl 0.2.11__py3-none-any.whl → 0.2.13__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.11'
-__version_tuple__ = version_tuple = (0, 2, 11)
+__version__ = version = '0.2.13'
+__version_tuple__ = version_tuple = (0, 2, 13)

hdl/utils/llm/visrag.py

@@ -7,8 +7,12 @@ import gradio as gr
 import os
 import numpy as np
 import json
+# import base64
+# import io
 from transformers import AutoModel, AutoTokenizer
-from hdl.utils.llm.chat import OpenAI_M
+
+from .chat import OpenAI_M
+from .vis import imgfile_to_base64
 
 def get_image_md5(img: Image.Image):
     img_byte_array = img.tobytes()
@@ -104,11 +108,22 @@ def retrieve_gradio(knowledge_base, query, topk, cache_dir=None, model=None, tok
 
     return images_topk
 
+# def convert_image_to_base64(image):
+#     """Convert a PIL Image to a base64 encoded string."""
+#     buffered = io.BytesIO()
+#     image.save(buffered, format="PNG")
+#     image_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+#     return image_base64
+
 def answer_question(images, question, gen_model):
-    images_ = [Image.open(image[0]).convert('RGB') for image in images]
+    # Convert images to base64
+    # images_base64 = [convert_image_to_base64(Image.open(image[0]).convert('RGB')) for image in images]
+    images_base64 = [imgfile_to_base64(image) for image in images]
+
+    # Pass base64-encoded images to gen_model.chat
     answer = gen_model.chat(
         prompt=question,
-        images=images_,
+        images=images_base64,  # Use the base64 images
         stream=False
     )
     return answer

{hjxdl-0.2.11.dist-info → hjxdl-0.2.13.dist-info}/METADATA

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

{hjxdl-0.2.11.dist-info → hjxdl-0.2.13.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=zHgAwrkDTK_-ggjmpfUbyN4DWYMgD-GquJ0wB1j66Z4,413
+hdl/_version.py,sha256=_zyE7p8pmjayqCVwnrfAUFK0j2fzgvPDn81vjk9iKCc,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
@@ -133,12 +133,12 @@ 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=2pI0439GWi_BEVfQJtY29Y72FkUa8jEvBeqMlwy7xkc,15716
-hdl/utils/llm/visrag.py,sha256=Nkt39OvltWvrYrHBw2eNgCMkArYtJI_RPttMi2se-Ns,7872
+hdl/utils/llm/visrag.py,sha256=diOO5eOeIwVQ5lIAgWTSHF70wqhyOVTpGahGEEQaLc4,8408
 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.11.dist-info/METADATA,sha256=Vllm_K5eqawd7WpTCegnb3KWBEKV1igVAj3FALI1-K0,836
-hjxdl-0.2.11.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-hjxdl-0.2.11.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.11.dist-info/RECORD,,
+hjxdl-0.2.13.dist-info/METADATA,sha256=2Ljze-IXByM5cSRDYqtjGCVH_Rd4X6eHSQ5CQpP63Is,836
+hjxdl-0.2.13.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+hjxdl-0.2.13.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.2.13.dist-info/RECORD,,