From aba4b96f23ef38ca75195b3b78a88bb92035b4a9 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Mon, 12 Feb 2024 11:41:27 +0100
Subject: Add new Client API with Docs Use object urls for the preview of image
 uploads. Fix upload images in You provider Fix create image. It's now a
 single image. Improve system message for create images.

---
 g4f/Provider/BingCreateImages.py | 56 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 g4f/Provider/BingCreateImages.py

(limited to 'g4f/Provider/BingCreateImages.py')

diff --git a/g4f/Provider/BingCreateImages.py b/g4f/Provider/BingCreateImages.py
new file mode 100644
index 00000000..901b7787
--- /dev/null
+++ b/g4f/Provider/BingCreateImages.py
@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+import asyncio
+import os
+from typing import Generator
+
+from ..cookies import get_cookies
+from ..image import ImageResponse
+from ..errors import MissingRequirementsError, MissingAuthError
+from .bing.create_images import create_images, create_session, get_cookies_from_browser
+
+class BingCreateImages:
+    """A class for creating images using Bing."""
+
+    def __init__(self, cookies: dict[str, str] = {}, proxy: str = None) -> None:
+        self.cookies = cookies
+        self.proxy = proxy
+
+    def create(self, prompt: str) -> Generator[ImageResponse, None, None]:
+        """
+        Generator for creating imagecompletion based on a prompt.
+
+        Args:
+            prompt (str): Prompt to generate images.
+
+        Yields:
+            Generator[str, None, None]: The final output as markdown formatted string with images.
+        """
+        cookies = self.cookies or get_cookies(".bing.com", False)
+        if "_U" not in cookies:
+            login_url = os.environ.get("G4F_LOGIN_URL")
+            if login_url:
+                yield f"Please login: [Bing]({login_url})\n\n"
+            try:
+                self.cookies = get_cookies_from_browser(self.proxy)
+            except MissingRequirementsError as e:
+                raise MissingAuthError(f'Missing "_U" cookie. {e}')
+        yield asyncio.run(self.create_async(prompt))
+
+    async def create_async(self, prompt: str) -> ImageResponse:
+        """
+        Asynchronously creates a markdown formatted string with images based on the prompt.
+
+        Args:
+            prompt (str): Prompt to generate images.
+
+        Returns:
+            str: Markdown formatted string with images.
+        """
+        cookies = self.cookies or get_cookies(".bing.com", False)
+        if "_U" not in cookies:
+            raise MissingAuthError('Missing "_U" cookie')
+        proxy = self.proxy or os.environ.get("G4F_PROXY")
+        async with create_session(cookies, proxy) as session:
+            images = await create_images(session, prompt, proxy)
+            return ImageResponse(images, prompt, {"preview": "{image}?w=200&h=200"} if len(images) > 1 else {})
\ No newline at end of file
-- 
cgit v1.2.3