Display generated images inline.

EtiennePerot · Oct 12, 2024 · 17ae258 · 17ae258
1 parent fae5797
commit 17ae258
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/open-webui/functions/run_code.py b/open-webui/functions/run_code.py
@@ -502,6 +502,8 @@ def _inline_markdown(self):
             """Render the file as inline text or markdown if small enough; otherwise return None."""
             if self._size_bytes > self.MAX_INLINE_TEXT_BYTES:
                 return None
+            if self._mime_type.startswith("image/"):
+                return f"\U0001f5bc [{self.name}]({self.url}):  \n![{self.name}]({self.url})"
             if not self._mime_type.startswith("text/"):
                 return None
             with open(self._file_path, "rb") as f:

diff --git a/src/openwebui/functions/run_code.py b/src/openwebui/functions/run_code.py
@@ -461,6 +461,8 @@ def _inline_markdown(self):
             """Render the file as inline text or markdown if small enough; otherwise return None."""
             if self._size_bytes > self.MAX_INLINE_TEXT_BYTES:
                 return None
+            if self._mime_type.startswith("image/"):
+                return f"\U0001f5bc [{self.name}]({self.url}):  \n![{self.name}]({self.url})"
             if not self._mime_type.startswith("text/"):
                 return None
             with open(self._file_path, "rb") as f: