Update app.py

app.py

@@ -39,6 +39,15 @@ def compute_difference_images(img_a, img_b):
         "diff_overlay": diff_overlay
     }
 
+# 保存图像到文件
+def save_images(images):
+    paths = []
+    for key, img in images.items():
+        path = f"{key}.png"
+        img.save(path)
+        paths.append((path, key.replace("_", " ").capitalize()))
+    return paths
+
 # BLIP生成更详尽描述
 def generate_detailed_caption(image):
     inputs = blip_processor(image, return_tensors="pt")
@@ -79,6 +88,7 @@ def analyze_images(image_a, image_b, api_key):
     img_a = image_a.convert("RGB")
     img_b = image_b.convert("RGB")
     images_diff = compute_difference_images(img_a, img_b)
+    saved_images = save_images(images_diff)
 
     # BLIP生成描述
     caption_a = generate_detailed_caption(img_a)
@@ -112,7 +122,7 @@ def analyze_images(image_a, image_b, api_key):
         "caption_a": caption_a,
         "caption_b": caption_b,
         "text_analysis": text_analysis,
-        "images_diff": images_diff,
+        "saved_images": saved_images,
         "bar_chart": bar_chart_path,
         "pie_chart": pie_chart_path
     }
@@ -147,17 +157,8 @@ with gr.Blocks() as demo:
     # 分析逻辑
     def process_analysis(img_a, img_b, api_key):
         results = analyze_images(img_a, img_b, api_key)
-        diff_images = [
-            ("Original A", results["images_diff"]["original_a"]),
-            ("Original B", results["images_diff"]["original_b"]),
-            ("Sketch A", results["images_diff"]["sketch_a"]),
-            ("Sketch B", results["images_diff"]["sketch_b"]),
-            ("Normal A", results["images_diff"]["normal_a"]),
-            ("Normal B", results["images_diff"]["normal_b"]),
-            ("Difference Overlay", results["images_diff"]["diff_overlay"]),
-        ]
         return (
-            diff_images,
+            results["saved_images"],
             results["caption_a"],
             results["caption_b"],
             results["text_analysis"],