Update app.py

app.py

@@ -115,43 +115,31 @@ def test_model_availability(api_key, model_name):
         )
         return False
     
-def create_base64_markdown_image(image_url, max_width=None, max_height=None):
+def create_base64_markdown_image(image_url):
     try:
         response = requests.get(image_url, stream=True)
         response.raise_for_status()
 
         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:
-                # 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:
-                # 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:
-                # 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)
+        
+        new_width = original_width // 2
+        new_height = original_height // 2
+        
+        resized_image = image.resize((new_width, new_height), Image.LANCZOS)
 
         buffered = BytesIO()
-        image.save(buffered, format="PNG")
-        compressed_image_data = buffered.getvalue()
+        resized_image.save(buffered, format="PNG")
+        
+        resized_image_data = buffered.getvalue()
+
+        base64_encoded = base64.b64encode(resized_image_data).decode('utf-8')
+        
+        mime_type = 'image/png' #response.headers.get('Content-Type', 'image/png')
 
-        base64_encoded = base64.b64encode(compressed_image_data).decode('utf-8')
-        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