handle sage images

Author: NullDev

src/service/sageExecution.js

@@ -30,7 +30,14 @@ const executeSage = async function(interaction){
 
     return client.askSage(code)
         .then(res => {
-            embed.description = "Input:\n```py\n" + code + "\n```\nOutput:\n```\n" + (res?.result?.["text/plain"]?.trim() || res?.stdout?.trim() || "No Output") + "\n```";
+            embed.description = "Input:\n```py\n" + code + "\n```\nOutput:\n```\n" + (
+                res?.result?.["text/plain"]?.trim() || res?.stdout?.trim() || (res.result?.image ? "Image Below" : "No Output")
+            ) + "\n```";
+
+            if (res?.result?.image){ // @ts-ignore
+                embed.image = { url: res.result.image };
+            }
+
             return interaction.editReply({ embeds: [embed]});
         })
         .catch(() => {

src/util/SageCell.js

@@ -187,6 +187,11 @@ export default class SageCell {
                     result = msg.content?.data ?? null;
                 }
 
+                if (msgType === "display_data"){
+                    const filename = msg.content?.data?.["text/image-filename"];
+                    result = filename ? { image: `${this.baseUrl}/kernel/${kernelId}/files/${filename}` } : msg.content?.data ?? null;
+                }
+
                 if (msgType === "status" &&
                     msg.content?.execution_state === "idle"){
                     finish({ stdout, stderr, result });