Update app.py

app.py

@@ -3,15 +3,20 @@ from stitching import Stitcher
 import cv2
 import numpy as np
 
-def stitch_images(image_mode, image1, image2, images, detector, matcher, estimator, blender_type, crop, wave_correct):
+def stitch_images(image_mode, image1, image2, images, detector, matcher, estimator, blender_type, crop, wave_correct, progress=gr.Progress(track_tqdm=True)):
     if image_mode == "Two Images":
         if image1 is None or image2 is None:
             return None, "Please upload both images."
         cv_images = [cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) for img in [image1, image2]]
     else:  # Multiple Images
-        if len(images) < 2:
+        if not images or len(images) < 2:
             return None, "Please upload at least 2 images."
-        cv_images = [cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) for img in images]
+        cv_images = []
+        for img_path in images:
+            img = cv2.imread(img_path.name)
+            if img is None:
+                return None, f"Failed to read image: {img_path.name}"
+            cv_images.append(img)
     
     try:
         stitcher = Stitcher(
@@ -64,7 +69,6 @@ with gr.Blocks() as iface:
     
     with gr.Row():
         output_image = gr.Image(type="numpy", label="Stitched Panorama")
-        status = gr.Textbox(label="Status")
     
     stitch_button.click(
         stitch_images,