mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-24 16:50:22 +08:00
c0d31c2abb
- Modified g4f/providers/response.py to ensure format_images_markdown returns the result directly without additional flags in the 'format_images_markdown' function.
- Updated g4f/gui/server/api.py to add 'tempfiles' parameter with default empty list to '_create_response_stream' method.
- Changed or added code in API response handling to iterate over 'tempfiles' and attempt to remove each file after response completion, with exception handling (try-except block with logger.exception).
- Adjusted g4f/Tools/files.py to fix tempfile creation: corrected the 'suffix' parameter in 'get_tempfile' to use 'suffix' directly instead of splitting.
- In g4f/tools/media.py, changed 'render_part' function to handle 'text' key properly, checking 'part.get("text")' and returning a dictionary with 'type': 'text' and 'text': value, if present.
104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from aiohttp import ClientSession
|
|
import time
|
|
import asyncio
|
|
|
|
from ..typing import AsyncResult, Messages
|
|
from ..providers.response import ImageResponse
|
|
from ..image import use_aspect_ratio
|
|
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
|
|
|
|
|
|
class ImageLabs(AsyncGeneratorProvider, ProviderModelMixin):
|
|
url = "https://editor.imagelabs.net"
|
|
api_endpoint = "https://editor.imagelabs.net/txt2img"
|
|
|
|
working = True
|
|
supports_stream = False
|
|
supports_system_message = False
|
|
supports_message_history = False
|
|
|
|
default_model = 'sdxl-turbo'
|
|
default_image_model = default_model
|
|
image_models = [default_image_model]
|
|
models = image_models
|
|
|
|
@classmethod
|
|
async def create_async_generator(
|
|
cls,
|
|
model: str,
|
|
messages: Messages,
|
|
proxy: str = None,
|
|
# Image
|
|
prompt: str = None,
|
|
negative_prompt: str = "",
|
|
aspect_ratio: str = "1:1",
|
|
width: int = None,
|
|
height: int = None,
|
|
extra_body: dict = {},
|
|
**kwargs
|
|
) -> AsyncResult:
|
|
extra_body = use_aspect_ratio({
|
|
"width": width,
|
|
"height": height,
|
|
**extra_body
|
|
}, aspect_ratio)
|
|
|
|
headers = {
|
|
'accept': '*/*',
|
|
'accept-language': 'en-US,en;q=0.9',
|
|
'cache-control': 'no-cache',
|
|
'content-type': 'application/json',
|
|
'origin': cls.url,
|
|
'referer': f'{cls.url}/',
|
|
'x-requested-with': 'XMLHttpRequest',
|
|
'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36'
|
|
}
|
|
|
|
async with ClientSession(headers=headers) as session:
|
|
prompt = messages[-1]["content"] if prompt is None else prompt
|
|
|
|
# Generate image
|
|
payload = {
|
|
"prompt": prompt,
|
|
"seed": str(int(time.time())),
|
|
"subseed": str(int(time.time() * 1000)),
|
|
"attention": 0,
|
|
"tiling": False,
|
|
"negative_prompt": negative_prompt,
|
|
"reference_image": "",
|
|
"reference_image_type": None,
|
|
"reference_strength": 30,
|
|
**extra_body
|
|
}
|
|
|
|
async with session.post(f'{cls.url}/txt2img', json=payload, proxy=proxy) as generate_response:
|
|
generate_data = await generate_response.json()
|
|
task_id = generate_data.get('task_id')
|
|
|
|
# Poll for progress
|
|
while True:
|
|
async with session.post(f'{cls.url}/progress', json={"task_id": task_id}, proxy=proxy) as progress_response:
|
|
progress_data = await progress_response.json()
|
|
|
|
# Check for completion or error states
|
|
if progress_data.get('status') == 'Done' or progress_data.get('final_image_url'):
|
|
# Yield ImageResponse with the final image URL
|
|
yield ImageResponse(
|
|
urls=[progress_data.get('final_image_url')],
|
|
alt=prompt
|
|
)
|
|
break
|
|
|
|
# Check for queue or error states
|
|
if 'error' in progress_data.get('status', '').lower():
|
|
raise Exception(f"Image generation error: {progress_data}")
|
|
|
|
# Wait between polls
|
|
await asyncio.sleep(1)
|
|
|
|
@classmethod
|
|
def get_model(cls, model: str) -> str:
|
|
return cls.default_model
|