Update app.py

app.py

@@ -1,2 +1,64 @@
 import json
 import os
+from typing import Optional, Union
+from PIL import Image
+import base64
+from io import BytesIO
+
+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
+
+
+file_path = "./test_message_gpt.json"
+messages = json.load(open(file_path, "r", encoding="utf-8"))
+
+for message_item in messages:
+    role = message_item['role']
+    content = message_item['content']
+
+    for content_item in content:
+        content_type = content_item['type']
+        if content_type == "text":
+            content_value = content_item['text']
+        elif conent_type == "image_url":
+            content_value = content_item['image_url']['url']
+            image = base64_to_image(content_value)
+