pass base64 images

app_dialogue.py

@@ -106,6 +106,17 @@ def pil_to_temp_file(
     return filename
 
 
+def pil_to_base64(
+    img: PIL.Image.Image,
+    resize: bool = False,
+) -> str:
+    """Save a PIL image into a temp file"""
+    if resize:
+        img = img.resize((224, 224), Image.LANCZOS)
+    base64_img = processing_utils.encode_pil_to_base64(img)
+    return base64_img
+
+
 def add_file_gallery(selected_state: gr.SelectData, gallery_list: List[str]):
     return (
         "Write a meme about this image.",
@@ -320,8 +331,8 @@ def prompt_list_to_tgi_input(prompt_list: List[str]) -> str:
                     print(f"Image is not loading")
             else:
                 elem_pil = Image.open(elem)
-            elem = pil_to_temp_file(elem_pil, resize=True)
-            result_string_input += f"![]({gradio_link(img_path=elem)})"
+            base64_img = pil_to_base64(elem_pil, resize=True)
+            result_string_input += f"![]({base64_img})"
         else:
             result_string_input += elem
     return result_string_input