Allow images to be embedded from document

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main.py b/src/main.py
index 88d2a4d..51fc8c5 100644
--- a/src/main.py
+++ b/src/main.py
@@ -25,7 +25,7 @@ async def explain(ctx, *args):
 
     try:
         search_content = scrape_lexicanum.get_search_response(config, query)
-        page_header, page_content = scrape_lexicanum.get_page_content(config, search_content[0])
+        page_header, page_content, page_img_link = scrape_lexicanum.get_page_content(config, search_content[0])
         
         embed = discord.Embed(
                     title=page_header,
@@ -38,6 +38,9 @@ async def explain(ctx, *args):
         embed.add_field(name="Overview", value=page_content)
         embed.add_field(name="You May Have Meant", value=string_results)
 
+        if page_img_link:
+            embed.set_image(url=page_img_link)
+
         await ctx.send(embed=embed)
     except scrape_lexicanum.WikiError as e:
         await ctx.send(f"{e}")