Update forensics/ela_hybrid.py

forensics/ela_hybrid.py

@@ -61,9 +61,20 @@ def visualize_hybrid(hybrid_array: np.ndarray):
     ela_image = Image.fromarray((hybrid_array[:, :, 3:] * 255).astype(np.uint8))
 
     return [rgb_image, ela_image]
+    
+def generate_hybrid_ela_func(img_np_og, description, quality=95, scale_factor=150):
+    """
+    Returns a list of two PIL Images: one for RGB, one for ELA map.
+    This matches the structure of other forensic tools.
+    """
+    from PIL import Image
+
+    # Ensure the image is in RGB numpy format
+    if img_np_og.ndim == 2:
+        img_np_og = cv2.cvtColor(img_np_og, cv2.COLOR_GRAY2RGB)
 
-# Function to generate hybrid ELA as list of PIL Images for run_forensic_task
-def generate_hybrid_ela_func(img):
-    # Convert to numpy array if needed (e.g., from OpenCV BGR to PIL RGB)
-    img_array = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    return visualize_hybrid(generate_ela_hybrid(img_array))
+    hybrid_array = generate_ela_hybrid(img_np_og, quality=quality, scale_factor=scale_factor)
+    visualizations = visualize_hybrid(hybrid_array)
+    
+    # Return list of PIL Images for Gradio compatibility
+    return visualizations