From 7dbd195e45be61a8a0259955a687decb2a495539 Mon Sep 17 00:00:00 2001
From: 6vision <vision_wangpc@sina.com>
Date: Thu, 25 Jul 2024 01:12:53 +0800
Subject: [PATCH] Support images in webp format.

---
 channel/wechat/wechat_channel.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/channel/wechat/wechat_channel.py b/channel/wechat/wechat_channel.py
index 870b487..de264ff 100644
--- a/channel/wechat/wechat_channel.py
+++ b/channel/wechat/wechat_channel.py
@@ -9,7 +9,7 @@ import json
 import os
 import threading
 import time
-
+import uuid
 import requests
 
 from bridge.context import *
@@ -229,6 +229,12 @@ class WechatChannel(ChatChannel):
                 image_storage.write(block)
             logger.info(f"[WX] download image success, size={size}, img_url={img_url}")
             image_storage.seek(0)
+            if img_url.endswith(".webp"):
+                try:
+                    image_storage = _convert_webp_to_png(image_storage)
+                except Exception as e:
+                    logger.error(f"Failed to convert image: {e}")
+                    return
             itchat.send_image(image_storage, toUserName=receiver)
             logger.info("[WX] sendImage url={}, receiver={}".format(img_url, receiver))
         elif reply.type == ReplyType.IMAGE:  # 从文件读取图片
@@ -266,6 +272,7 @@ def _send_login_success():
     except Exception as e:
         pass
 
+
 def _send_logout():
     try:
         from common.linkai_client import chat_client
@@ -274,6 +281,7 @@ def _send_logout():
     except Exception as e:
         pass
 
+
 def _send_qr_code(qrcode_list: list):
     try:
         from common.linkai_client import chat_client
@@ -281,3 +289,19 @@ def _send_qr_code(qrcode_list: list):
             chat_client.send_qrcode(qrcode_list)
     except Exception as e:
         pass
+
+
+def _convert_webp_to_png(webp_image):
+        from PIL import Image
+        try:
+            webp_image.seek(0)
+            img = Image.open(webp_image).convert("RGBA")
+            png_image = io.BytesIO()
+            unique_filename = f"{uuid.uuid4()}.png"
+            img.save(png_image, format="PNG")
+            png_image.name = unique_filename
+            png_image.seek(0)
+            return png_image
+        except Exception as e:
+            logger.error(f"Failed to convert WEBP to PNG: {e}")
+            raise