Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageChops, ImageFilter
 from transformers import CLIPProcessor, CLIPModel, BlipProcessor, BlipForConditionalGeneration
 import torch
 import matplotlib.pyplot as plt
+import numpy as np  # 修复未导入问题
 
 # 初始化模型
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
@@ -138,8 +139,8 @@ with gr.Blocks() as demo:
 
     api_key_input = gr.Textbox(label="API Key", placeholder="输入您的 API Key", type="password")
     api_type_input = gr.Radio(label="API 类型", choices=["GPT", "DeepSeek"], value="GPT")
-    images_a_input = gr.File(label="上传文件夹A图片", file_types=[".png", ".jpg", ".jpeg"], file_count="multiple")
-    images_b_input = gr.File(label="上传文件夹B图片", file_types=[".png", ".jpg", ".jpeg"], file_count="multiple")
+    images_a_input = gr.File(label="上传文件夹A图片", file_types=[".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".gif", ".webp"], file_count="multiple")
+    images_b_input = gr.File(label="上传文件夹B图片", file_types=[".png", ".jpg", ".jpeg", ".bmp", ".tiff", ".gif", ".webp"], file_count="multiple")
     analyze_button = gr.Button("开始批量分析")
 
     with gr.Row():