Update app.py

try image background filter

app.py

@@ -4,11 +4,27 @@ import numpy as np
 import gradio as gr
 
 import os
-from PIL import Image
+from PIL import Image, ImageDraw
 
 from codeinterpreterapi import CodeInterpreterSession
 
 
+def imGreyAlpha(im):
+    grey = im.convert('L') # 转成灰度
+    px = grey.load() # 获取灰度数组
+
+    im2 = Image.new(mode="RGBA", size=(w, h), color=(255,255,255,0)) # 新建图片
+    draw = ImageDraw.Draw(im2) # 获取绘制句柄
+
+    w,h = im.size # 获取图片大小
+
+    # 遍历像素点
+    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 # 返回图像
+
 
 def codeinterpreter(openai_key,prompt, files):
 
@@ -35,21 +51,29 @@ def codeinterpreter(openai_key,prompt, files):
         
 
 with gr.Blocks() as app:
-    with gr.Row():
-        with gr.Column():
-            inp1=gr.Textbox(label="openai_key")
-            inp2=gr.Textbox(label="prompt",info="input the prompt")
-            inp3=gr.Files()
-            btn = gr.Button(value="Submit")
-        with gr.Column():
-            out1=gr.Textbox(label="result")
-            out2=gr.Gallery()
-        
-    gr.Examples([["Plot the nvidea stock vs microsoft stock over the last 6 months."],
+    with gr.Tab(label="codeinterpreter"):
+        with gr.Row():
+            with gr.Column():
+                inp1=gr.Textbox(label="openai_key")
+                inp2=gr.Textbox(label="prompt",info="input the prompt")
+                inp3=gr.Files()
+                btn = gr.Button(value="Submit")
+            with gr.Column():
+                out1=gr.Textbox(label="result")
+                out2=gr.Gallery()
+        gr.Examples([["Plot the nvidea stock vs microsoft stock over the last 6 months."],
                ["Plot a sin wave and show it to me."],
                ["贵州茅台最近半年走势"],
                ["Plot the bitcoin chart of 2023 YTD"]],
                 [inp2])
+    with gr.Tab(label="image background filter"):
+        with gr.Row():
+            with gr.Column():
+                inp10=gr.Image(label="original image")
+                btn10 = gr.Button(value="Submit")
+            with gr.Column():
+                out10=gr.Image(label="result")
     btn.click(codeinterpreter, inputs=[inp1,inp2,inp3], outputs=[out1,out2],api_name="getresult")
+    btn10.click(imGreyAlpha,inputs[inp10],outputs=[out10],api_name="imageFilter")
 
 app.launch()