Update app.py

app.py

@@ -8,6 +8,7 @@ import concurrent.futures
 import threading
 import base64
 import io
+from io import BytesIO
 from itertools import chain
 from PIL import Image
 from datetime import datetime
@@ -114,14 +115,32 @@ def test_model_availability(api_key, model_name):
         )
         return False
     
-def create_base64_markdown_image(image_url):
+def create_base64_markdown_image(image_url, max_width=None, max_height=None):
     try:
         response = requests.get(image_url, stream=True)
         response.raise_for_status()
 
         image_data = response.content
-        base64_encoded = base64.b64encode(image_data).decode('utf-8')
-        mime_type = response.headers.get('Content-Type', 'image/png')
+        image = Image.open(BytesIO(image_data))
+
+        if max_width or max_height:
+            if max_width and max_height:
+                image.thumbnail((max_width, max_height), Image.LANCZOS)
+            elif max_width:
+                wpercent = (max_width / float(image.size[0]))
+                hsize = int((float(image.size[1]) * float(wpercent)))
+                image = image.resize((max_width, hsize), Image.LANCZOS)
+            elif max_height:
+                 hpercent = (max_height / float(image.size[1]))
+                 wsize = int((float(image.size[0]) * float(hpercent)))
+                 image = image.resize((wsize, max_height), Image.LANCZOS)
+
+        buffered = BytesIO()
+        image.save(buffered, format="PNG")
+        compressed_image_data = buffered.getvalue()
+
+        base64_encoded = base64.b64encode(compressed_image_data).decode('utf-8')
+        mime_type = "image/png"
         markdown_image_link = f"![](data:{mime_type};base64,{base64_encoded})"
         logging.info(f"Created base64 markdown image link.")
         return markdown_image_link