Update feifeilib/feifeiflorence.py

feifeilib/feifeiflorence.py

@@ -1,3 +1,5 @@
+from PIL import Image
+from io import BytesIO
 import base64
 import requests
 import os
@@ -10,12 +12,25 @@ Mistralclient = Mistral(api_key=api_key)
 def encode_image(image_path):
     """Encode the image to base64."""
     try:
-        with open(image_path, "rb") as image_file:
-            return base64.b64encode(image_file.read()).decode('utf-8')
+        # Open the image file
+        image = Image.open(image_path).convert("RGB")
+
+        # Resize the image to a height of 512 while maintaining the aspect ratio
+        base_height = 512
+        h_percent = (base_height / float(image.size[1]))
+        w_size = int((float(image.size[0]) * float(h_percent)))
+        image = image.resize((w_size, base_height), Image.LANCZOS)
+
+        # Convert the image to a byte stream
+        buffered = BytesIO()
+        image.save(buffered, format="JPEG")
+        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+        return img_str
     except FileNotFoundError:
         print(f"Error: The file {image_path} was not found.")
         return None
-    except Exception as e:  # Added general exception handling
+    except Exception as e:  # Add generic exception handling
         print(f"Error: {e}")
         return None