Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import cv2
 import numpy as np
+from PIL import Image
 
-def exposure_fusion(images):
+def exposure_fusion(image_paths):
     try:
-        # Convert PIL images (RGB) to OpenCV format (BGR)
-        images_cv = [cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) for img in images]
+        # Open images from filepaths and convert to OpenCV format (BGR)
+        images_cv = [cv2.cvtColor(np.array(Image.open(path)), cv2.COLOR_RGB2BGR) for path in image_paths]
         
         # Align images using AlignMTB
         align_mtb = cv2.createAlignMTB()
@@ -23,10 +24,10 @@ def exposure_fusion(images):
     except Exception as e:
         return f"Error: {e}"
 
-def stabilize_crop_and_exposure_fusion(images):
+def stabilize_crop_and_exposure_fusion(image_paths):
     try:
-        # Convert images from PIL (RGB) to OpenCV format (BGR)
-        images_cv = [cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) for img in images]
+        # Open images from filepaths and convert to OpenCV format (BGR)
+        images_cv = [cv2.cvtColor(np.array(Image.open(path)), cv2.COLOR_RGB2BGR) for path in image_paths]
 
         # Align images using AlignMTB
         align_mtb = cv2.createAlignMTB()
@@ -37,7 +38,6 @@ def stabilize_crop_and_exposure_fusion(images):
         bounding_rects = []
         for img in aligned_images:
             gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-            # Pixels above a small threshold are considered valid
             _, mask = cv2.threshold(gray, 10, 255, cv2.THRESH_BINARY)
             coords = cv2.findNonZero(mask)
             if coords is not None:
@@ -72,14 +72,13 @@ def stabilize_crop_and_exposure_fusion(images):
     except Exception as e:
         return f"Error: {e}"
 
-def process_images(images, advanced):
-    if not images:
+def process_images(image_paths, advanced):
+    if not image_paths:
         return None
-    # If advanced option is selected, use stabilization & cropping before fusion.
     if advanced:
-        return stabilize_crop_and_exposure_fusion(images)
+        return stabilize_crop_and_exposure_fusion(image_paths)
     else:
-        return exposure_fusion(images)
+        return exposure_fusion(image_paths)
 
 # Gradio Interface: Upload multiple images and choose the processing method.
 inputs = [