Update forensics/ela_hybrid.py

forensics/ela_hybrid.py

@@ -77,4 +77,23 @@ def generate_hybrid_ela_func(img_input):
     # Generate ELA
     hybrid_array = generate_ela_hybrid(img_input, quality=75, scale_factor=100)
     visualizations = visualize_hybrid(hybrid_array)
-    return list(visualizations)  # Returns [RGB PIL Image, ELA PIL Image]
+    return list(visualizations)  # Returns [RGB PIL Image, ELA PIL Image]
+
+
+def generate_concatenated_hybrid(img_array):
+    """
+    Concatenate original RGB and ELA map into a single image.
+    Returns a single PIL Image (side-by-side).
+    """
+    # Extract RGB and ELA channels
+    original_rgb = Image.fromarray((img_array[:, :, :3] * 255).astype(np.uint8))  # 3 channels
+    ela_map = Image.fromarray((img_array[:, :, 3:] * 255).astype(np.uint8))  # 3 channels
+
+    # Resize to match height
+    ela_map = ela_map.resize(original_rgb.size)
+
+    # Concatenate and return as a single PIL Image
+    combined = Image.new("RGB", (ela_map.width + original_rgb.width, original_rgb.height))  # Single row
+    combined.paste(original_rgb, (0, 0))
+    combined.paste(ela_map, (original_rgb.width, 0))
+    return combined