Update app.py

app.py

@@ -122,25 +122,36 @@ def create_base64_markdown_image(image_url, max_width=None, max_height=None):
 
         image_data = response.content
         image = Image.open(BytesIO(image_data))
+        original_width, original_height = image.size
 
         if max_width or max_height:
+            # Calculate the new dimensions while maintaining aspect ratio
             if max_width and max_height:
-                image.thumbnail((max_width, max_height), Image.LANCZOS)
+                # If both max_width and max_height are provided, scale to fit within the box
+                ratio = min(max_width / original_width, max_height / original_height)
+                new_width = int(original_width * ratio)
+                new_height = int(original_height * ratio)
+                image = image.resize((new_width, new_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)
+                # Scale based on max_width
+                ratio = max_width / original_width
+                new_width = max_width
+                new_height = int(original_height * ratio)
+                image = image.resize((new_width, new_height), 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)
+                # Scale based on max_height
+                ratio = max_height / original_height
+                new_height = max_height
+                new_width = int(original_width * ratio)
+                image = image.resize((new_width, new_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"
+        mime_type = "image/png"  # Always use image/png since we are saving as png
         markdown_image_link = f"![](data:{mime_type};base64,{base64_encoded})"
         logging.info(f"Created base64 markdown image link.")
         return markdown_image_link