Update app.py

app.py

@@ -13,6 +13,47 @@ from typing import Optional, Union
 import gradio as gr
 import markdown
 
+def base64_to_image(
+    base64_str: str, 
+    remove_prefix: bool = True, 
+    convert_mode: Optional[str] = "RGB"
+) -> Union[Image.Image, None]:
+    """
+    将Base64编码的图片字符串转换为PIL Image对象
+    
+    Args:
+        base64_str: Base64编码的图片字符串（可带data:前缀）
+        remove_prefix: 是否自动去除"data:image/..."前缀（默认True）
+        convert_mode: 转换为指定模式（如"RGB"/"RGBA"，None表示不转换）
+    
+    Returns:
+        PIL.Image.Image 对象，解码失败时返回None
+        
+    Examples:
+        >>> img = base64_to_image("data:image/png;base64,iVBORw0KGg...")
+        >>> img = base64_to_image("iVBORw0KGg...", remove_prefix=False)
+    """
+    try:
+        # 1. 处理Base64前缀
+        if remove_prefix and "," in base64_str:
+            base64_str = base64_str.split(",")[1]
+
+        # 2. 解码Base64
+        image_data = base64.b64decode(base64_str)
+        
+        # 3. 转换为PIL Image
+        image = Image.open(BytesIO(image_data))
+        
+        # 4. 可选模式转换
+        if convert_mode:
+            image = image.convert(convert_mode)
+            
+        return image
+    
+    except (base64.binascii.Error, OSError, Exception) as e:
+        print(f"Base64解码失败: {str(e)}")
+        return None
+
 def encode_image(image):
     """
     将PIL.Image对象或图像文件路径转换为base64编码字符串，并获取分辨率信息