From 335c971f6a9cd071d18f9fffeb76df4eda8876d5 Mon Sep 17 00:00:00 2001
From: H Lohaus <hlohaus@users.noreply.github.com>
Date: Fri, 13 Dec 2024 22:20:58 +0100
Subject: Add multiple images support (#2478)

* Add multiple images support

* Add multiple images support in gui

* Support multiple images in legacy client and in the api
Fix some model names in provider model list

* Fix unittests

* Add vision and providers docs
---
 g4f/image.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'g4f/image.py')

diff --git a/g4f/image.py b/g4f/image.py
index 97dfdea3..06151ef8 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -266,6 +266,7 @@ def to_bytes(image: ImageType) -> bytes:
         image.seek(0)
         return bytes_io.getvalue()
     else:
+        image.seek(0)
         return image.read()
 
 def to_data_uri(image: ImageType) -> str:
@@ -283,7 +284,7 @@ async def copy_images(
     images: list[str],
     cookies: Optional[Cookies] = None,
     proxy: Optional[str] = None
-):
+) -> list[str]:
     ensure_images_dir()
     async with ClientSession(
         connector=get_connector(proxy=proxy),
-- 
cgit v1.2.3