Update app.py

app.py

@@ -2,6 +2,8 @@
 
 import gradio as gr
 from PIL import Image
+import numpy as np
+
 
 from pipeline import SmartCBC
 
@@ -25,36 +27,30 @@ def analyze_images(images, age, gender, output_mode):
     # Gradio Gallery sometimes returns list of tuples: (PIL.Image, metadata)
     def to_pil_list(img_list):
         pil_list = []
+
         for item in img_list:
+            # Gallery may return (media, meta) or just media
             if isinstance(item, tuple):
-                # assume first element is the PIL image
-                pil = item[0]
+                media = item[0]
             else:
-                pil = item
+                media = item
+    
+            # filepath string
+            if isinstance(media, str):
+                pil = Image.open(media).convert("RGB")
+            # already a PIL Image
+            elif isinstance(media, Image.Image):
+                pil = media.convert("RGB")
+            # numpy array
+            elif isinstance(media, np.ndarray):
+                pil = Image.fromarray(media).convert("RGB")
+            else:
+                raise ValueError(f"Unsupported gallery item type: {type(media)}")
+    
             pil_list.append(pil)
+    
         return pil_list
 
-    pil_images = to_pil_list(images)
-
-    # If 1 image → single-FOV path
-    if len(pil_images) == 1:
-        result = cbc.analyze(
-            image=pil_images[0],
-            age=age,
-            gender=gender
-        )
-    else:
-        # Multiple images → multi-FOV aggregation
-        result = cbc.analyze(
-            image=pil_images,   # list → analyze_batch under the hood
-            age=age,
-            gender=gender
-        )
-
-    if output_mode == "Text Report":
-        return result["report_text"], None
-    else:
-        return None, result
 
 
 # -------------------------------------------------