Update app.py

app.py

@@ -1,4 +1,6 @@
+from PIL import Image
 import base64
+from io import BytesIO
 import os
 from mistralai import Mistral
 import gradio as gr
@@ -84,12 +86,19 @@ def infer(prompt, quality_select, styles_Radio, FooocusExpansion_select, seed=42
 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")
+        # 打开图片文件
+        image = Image.open(image_path).convert("RGB")
+
+        # 将图片转换为字节流
+        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:  # 添加通用异常处理
         print(f"Error: {e}")
         return None