resize images sent to tgi

app_dialogue.py

@@ -9,7 +9,9 @@ from pathlib import Path
 from typing import List, Optional, Tuple
 from urllib.parse import urlparse
 from PIL import Image, ImageDraw, ImageFont
+from io import BytesIO
 
+import requests
 import concurrent.futures
 import random
 import gradio as gr
@@ -86,9 +88,14 @@ def hash_bytes(bytes: bytes):
 
 
 def pil_to_temp_file(
-    img: PIL.Image.Image, dir: str = DEFAULT_TEMP_DIR, format: str = "png"
+    img: PIL.Image.Image,
+    dir: str = DEFAULT_TEMP_DIR,
+    format: str = "png",
+    resize: bool = False,
 ) -> str:
     """Save a PIL image into a temp file"""
+    if resize:
+        img = img.resize((224, 224), Image.LANCZOS)
     bytes_data = processing_utils.encode_pil_to_bytes(img, format)
     temp_dir = Path(dir) / hash_bytes(bytes_data)
     temp_dir.mkdir(exist_ok=True, parents=True)
@@ -304,9 +311,16 @@ def prompt_list_to_tgi_input(prompt_list: List[str]) -> str:
     for elem in prompt_list:
         if is_image(elem):
             if is_url(elem):
-                result_string_input += f"![]({elem})"
+                try:
+                    response = requests.get(elem)
+                    if response.status_code == 200:
+                        elem_pil = Image.open(BytesIO(response.content))
+                except Exception:
+                    print(f"Image is not loading")
             else:
-                result_string_input += f"![]({gradio_link(img_path=elem)})"
+                elem_pil = Image.open(elem)
+            elem = pil_to_temp_file(elem_pil, resize=True)
+            result_string_input += f"![]({gradio_link(img_path=elem)})"
         else:
             result_string_input += elem
     return result_string_input