Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import gradio as gr
 
 import os
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageOps
 
 from codeinterpreterapi import CodeInterpreterSession
 
@@ -12,20 +12,22 @@ from codeinterpreterapi import CodeInterpreterSession
 def imGreyAlpha(im):
     if isinstance(im,np.ndarray):
         im = Image.fromarray(im)
+
+    im2 = ImageOps.invert(im)
     
-    grey = im.convert('L') # 转成灰度
-    px = grey.load() # 获取灰度数组
+    # grey = im.convert('L') # 转成灰度
+    # px = grey.load() # 获取灰度数组
 
-    w,h = im.size # 获取图片大小
-    im2 = Image.new(mode="RGBA", size=(w, h), color=(255,255,255,0)) # 新建图片
-    draw = ImageDraw.Draw(im2) # 获取绘制句柄
+    # w,h = im.size # 获取图片大小
+    # im2 = Image.new(mode="RGBA", size=(w, h), color=(255,255,255,0)) # 新建图片
+    # draw = ImageDraw.Draw(im2) # 获取绘制句柄
     
 
-    # 遍历像素点
-    for i in range(w):
-        for j in range(h):
-            alpha = px[i, j] # 获取灰度作为alpha值
-            draw.point((i, j), fill=(0, 0, 0, 255 - alpha)) # 填充像素点
+    # # 遍历像素点
+    # for i in range(w):
+    #     for j in range(h):
+    #         alpha = px[i, j] # 获取灰度作为alpha值
+    #         draw.point((i, j), fill=(0, 0, 0, 255 - alpha)) # 填充像素点
     return im2 # 返回图像