UltraCoder/vk_chat_bot
1
0
Fork 0
Code Issues 4 Pull requests Projects Releases Packages Activity Actions

Конвертация в JPEG выделена в отдельную функцию.

Browse source 
Качество сжатия JPEG увеличено до 95.
This commit is contained in:
Kirill Kirilenko 2026-03-01 22:55:53 +03:00
parent 456d63ad0c
commit 11367c6c86
1 changed files with 12 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
ai_agent.py
View file

@ -92,6 +92,13 @@ def _get_resolution_for_aspect_ratio(aspect_ratio: str) -> Tuple[int, int]:
return aspect_ratio_resolution_map.get(aspect_ratio, (1280, 1024)) return aspect_ratio_resolution_map.get(aspect_ratio, (1280, 1024))
def _convert_image_to_jpeg(image: bytes) -> bytes:
img = Image.open(BytesIO(image)).convert("RGB")
output = BytesIO()
img.save(output, format='JPEG', quality=95, optimize=True)
return output.getvalue()
class AiAgent: class AiAgent:
def __init__(self, def __init__(self,
openrouter_token: str, openrouter_model: str, openrouter_token: str, openrouter_model: str,
@ -307,14 +314,10 @@ class AiAgent:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(image_url) as response: async with session.get(image_url) as response:
if response.status == 200: if response.status == 200:
image_bytes = await response.read() image = await response.read()
if not image_url.endswith(".jpg"): if not image_url.endswith(".jpg"):
image = Image.open(BytesIO(image_bytes)).convert("RGB") image = _convert_image_to_jpeg(image)
output = BytesIO() return Ok(image)
image.save(output, format="JPEG", quality=80, optimize=True)
image_bytes = output.getvalue()
return Ok(image_bytes)
else: else:
raise RuntimeError(f"Не удалось загрузить изображение ({response.status}).") raise RuntimeError(f"Не удалось загрузить изображение ({response.status}).")
@ -362,13 +365,8 @@ class AiAgent:
try: try:
outputs = await self.replicate_client.async_run(REPLICATE_MODEL, input=arguments) outputs = await self.replicate_client.async_run(REPLICATE_MODEL, input=arguments)
image_bytes = await outputs[0].aread() image = _convert_image_to_jpeg(await outputs[0].aread())
return Ok(image)
image = Image.open(BytesIO(image_bytes)).convert("RGB")
output = BytesIO()
image.save(output, format="JPEG", quality=80, optimize=True)
image_bytes = output.getvalue()
return Ok(image_bytes)
except Exception as e: except Exception as e:
print(f"Ошибка генерации изображения: {e}") print(f"Ошибка генерации изображения: {e}")
return Err(str(e)) return Err(str(e))