summaryrefslogtreecommitdiffstats
path: root/g4f/image.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-02-11 09:26:02 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-02-11 09:26:02 +0100
commit9aeae65b9b0e7322f63186c0df061de06ca642ee (patch)
treea2436c8874e24e95dbb9d821b6d461cc3d2f3100 /g4f/image.py
parentChange python version to 3.12 in unitests (diff)
downloadgpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar.gz
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar.bz2
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar.lz
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar.xz
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.tar.zst
gpt4free-9aeae65b9b0e7322f63186c0df061de06ca642ee.zip
Diffstat (limited to 'g4f/image.py')
-rw-r--r--g4f/image.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/g4f/image.py b/g4f/image.py
index 93922c2e..01d6ae50 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -187,7 +187,7 @@ def to_base64_jpg(image: Image, compression_rate: float) -> str:
image.save(output_buffer, format="JPEG", quality=int(compression_rate * 100))
return base64.b64encode(output_buffer.getvalue()).decode()
-def format_images_markdown(images, alt: str, preview: str = None) -> str:
+def format_images_markdown(images: Union[str, list], alt: str, preview: Union[str, list] = None) -> str:
"""
Formats the given images as a markdown string.
@@ -202,9 +202,10 @@ def format_images_markdown(images, alt: str, preview: str = None) -> str:
if isinstance(images, str):
images = f"[![{alt}]({preview.replace('{image}', images) if preview else images})]({images})"
else:
+ if not isinstance(preview, list):
+ preview = [preview.replace('{image}', image) if preview else image for image in images]
images = [
- f"[![#{idx+1} {alt}]({preview.replace('{image}', image) if preview else image})]({image})"
- for idx, image in enumerate(images)
+ f"[![#{idx+1} {alt}]({preview[idx]})]({image})" for idx, image in enumerate(images)
]
images = "\n".join(images)
start_flag = "<!-- generated images start -->\n"