ai: fix image generation.

This commit is contained in:
thedragonsinn
2025-05-03 21:17:27 +05:30
parent a37f7014af
commit 42cc5c4f26

View File

@@ -69,17 +69,19 @@ def get_response_content(
parts[0]
except (AttributeError, IndexError, TypeError):
LOGGER.info(response)
return "Query failed... Try again", None
return "`Query failed... Try again`", None
try:
image_data = io.BytesIO(parts[0].inline_data.data)
image_data.name = "photo.jpg"
except (AttributeError, IndexError):
image_data = None
text = "\n".join([part.text for part in parts if part.text])
image_data = None
text = ""
sources = ""
for part in parts:
if part.text:
text += part + "\n"
if part.inline_data:
image_data = io.BytesIO(part.inline_data.data)
image_data.name = "photo.jpg"
if add_sources:
try:
hrefs = [