summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/nexra/NexraImageURL.py
diff options
context:
space:
mode:
authorkqlio67 <kqlio67@users.noreply.github.com>2024-10-11 08:33:30 +0200
committerkqlio67 <kqlio67@users.noreply.github.com>2024-10-11 08:33:30 +0200
commita9bc67362f2be529fe9165ebb13347195ba1ddcf (patch)
tree1a91836eaa94f14c18ad5d55f687ff8a2118c357 /g4f/Provider/nexra/NexraImageURL.py
parentfeat(g4f/Provider/__init__.py): add new providers and update imports (diff)
downloadgpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar.gz
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar.bz2
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar.lz
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar.xz
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.tar.zst
gpt4free-a9bc67362f2be529fe9165ebb13347195ba1ddcf.zip
Diffstat (limited to 'g4f/Provider/nexra/NexraImageURL.py')
-rw-r--r--g4f/Provider/nexra/NexraImageURL.py46
1 files changed, 0 insertions, 46 deletions
diff --git a/g4f/Provider/nexra/NexraImageURL.py b/g4f/Provider/nexra/NexraImageURL.py
deleted file mode 100644
index 13d70757..00000000
--- a/g4f/Provider/nexra/NexraImageURL.py
+++ /dev/null
@@ -1,46 +0,0 @@
-from __future__ import annotations
-from aiohttp import ClientSession
-import json
-from ...typing import AsyncResult, Messages
-from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
-from ..helper import format_prompt
-from ...image import ImageResponse
-
-class NexraImageURL(AsyncGeneratorProvider, ProviderModelMixin):
- label = "Image Generation Provider"
- api_endpoint = "https://nexra.aryahcr.cc/api/image/complements"
- models = ['dalle', 'dalle2', 'dalle-mini', 'emi', 'sdxl-turbo', 'prodia']
-
- @classmethod
- async def create_async_generator(
- cls,
- model: str,
- messages: Messages,
- proxy: str = None,
- **kwargs
- ) -> AsyncResult:
- headers = {
- "Content-Type": "application/json",
- }
-
- async with ClientSession(headers=headers) as session:
- prompt = format_prompt(messages)
- data = {
- "prompt": prompt,
- "model": model,
- "response": "url"
- }
-
- async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response:
- response.raise_for_status()
- response_text = await response.text()
-
- cleaned_response = response_text.lstrip('_')
- response_json = json.loads(cleaned_response)
-
- images = response_json.get("images")
- if images and len(images) > 0:
- image_response = ImageResponse(images[0], alt="Generated Image")
- yield image_response
- else:
- yield "No image URL found."