Update app.py

app.py

@@ -26,7 +26,7 @@ def imGreyAlpha(im):
     return im2 # 返回图像
 
 
-def codeinterpreter(openai_key,prompt, files):
+def codeinterpreter(openai_key,prompt,filterBG, files):
 
     fileList = []
     if files != None:
@@ -45,7 +45,10 @@ def codeinterpreter(openai_key,prompt, files):
         images = []
 
         for _file in response.files:
-            images.append(_file.get_image())
+            if filterBG:
+                images.append(imGreyAlpha(_file.get_image()))
+            else:
+                images.append(_file.get_image())
 
         return [response.content,images]
         
@@ -56,7 +59,8 @@ with gr.Blocks() as app:
             with gr.Column():
                 inp1=gr.Textbox(label="openai_key")
                 inp2=gr.Textbox(label="prompt",info="input the prompt")
-                inp3=gr.Files()
+                inp3=gr.Checkbox(label="do_filter_bg?", info="do background filter?")
+                inp4=gr.Files()
                 btn = gr.Button(value="Submit")
             with gr.Column():
                 out1=gr.Textbox(label="result")
@@ -73,7 +77,7 @@ with gr.Blocks() as app:
                 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")
+    btn.click(codeinterpreter, inputs=[inp1,inp2,inp3,inp4], outputs=[out1,out2],api_name="getresult")
     btn10.click(imGreyAlpha,inputs=[inp10],outputs=[out10],api_name="imageFilter")
 
 app.launch()