From b6833973d09a5ba47511ab1939ae97add46d80f9 Mon Sep 17 00:00:00 2001
From: nexy7574 <me@nexy7574.co.uk>
Date: Wed, 10 Jan 2024 16:20:03 +0000
Subject: [PATCH] Attach image to embed if provided

---
 src/cogs/ollama.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py
index f89a313..da73ece 100644
--- a/src/cogs/ollama.py
+++ b/src/cogs/ollama.py
@@ -266,6 +266,11 @@ class Ollama(commands.Cog):
                 inline=False
             )
             embed.set_footer(text="Using server %r" % server, icon_url=server_config.get("icon_url"))
+            if image_data:
+                if (image.height / image.width) >= 1.5:
+                    embed.set_image(url=image.url)
+                else:
+                    embed.set_thumbnail(url=image.url)
             view = OllamaView(ctx)
             try:
                 await ctx.edit(embed=embed, view=view)