diff options
Diffstat (limited to 'g4f/image.py')
| -rw-r--r-- | g4f/image.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/g4f/image.py b/g4f/image.py index 5402f9c8..ed8af103 100644 --- a/g4f/image.py +++ b/g4f/image.py @@ -210,7 +210,9 @@ def format_images_markdown(images: Union[str, list], alt: str, preview: Union[st if not isinstance(preview, list): preview = [preview.replace('{image}', image) if preview else image for image in images] result = "\n".join( - f"[]({image})" for idx, image in enumerate(images) + #f"[]({image})" + f'[<img src="{preview[idx]}" width="200" alt="#{idx+1} {alt}">]({image})' + for idx, image in enumerate(images) ) start_flag = "<!-- generated images start -->\n" end_flag = "<!-- generated images end -->\n" @@ -259,6 +261,13 @@ class ImageResponse: def get_list(self) -> list[str]: return [self.images] if isinstance(self.images, str) else self.images +class ImagePreview(ImageResponse): + def __str__(self): + return "" + + def to_string(self): + return super().__str__() + class ImageRequest: def __init__( self, |