Merge pull request #10 from intv0id/users/maoltea/supportPNGOutPut

Support PNG images as output
intv0id · Oct 14, 2021 · d9c7abb · d9c7abb
2 parents c204d77 + 70f6436
commit d9c7abb
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/src/Jupyscope/Extensions/CellExtensions/OutputCellExtension.cs b/src/Jupyscope/Extensions/CellExtensions/OutputCellExtension.cs
@@ -33,11 +33,28 @@ private static string convertDataToHtml(DataOutput dataCell)
             var result = "";
             foreach (var data in dataCell.Data)
             {
-                if (data.Key.StartsWith("image") && data.Value is System.Text.Json.JsonArray dataArr)
+                if (data.Key.StartsWith("image"))
                 {
+                    var imgSrc = "";
+                    if (data.Value is System.Text.Json.JsonArray dataArr)
+                    {
+                        imgSrc = string.Join("", dataArr);
+                    }
+                    else if (data.Value is string dataStr)
+                    {
+                        imgSrc = dataStr;
+                    }
+                    else
+                    {
+                        continue;
+                    }
                     if (data.Key.Contains("svg"))
                     {
-                        result += string.Join("", dataArr);
+                        result += imgSrc;
+                    }
+                    else if (data.Key.Contains("png"))
+                    {
+                        result += $"<img src=\" data:{data.Key};base64,{imgSrc}\" alt=\"Output image\"/>";
                     }
                 }
             }