biomed-llama_multimodal

app.py

@@ -383,22 +383,20 @@ def process_image(image_path, user_prompt, modality=None):
         for idx, prompt in enumerate(relevant_prompts):
             try:
                 mask_list = interactive_infer_image(model, pil_image, [prompt])
-                if not mask_list or len(mask_list) == 0:
+                if mask_list is None or len(mask_list) == 0:
                     analysis_results.append(f"No mask generated for '{prompt}'")
                     continue
                 
                 pred_mask = mask_list[0]
-                # if pred_mask is None or not pred_mask.any():
-                #     analysis_results.append(f"Empty mask generated for '{finding}'")
-                #     continue
-                
-                # p_value = check_mask_stats(image, pred_mask.astype(np.uint8) * 255, detected_modality, finding)
-                # analysis_results.append(f"P-value for '{finding}' ({detected_modality}): {p_value:.4f}")
+                # Check if mask is valid using numpy's any() function
+                if pred_mask is None or not np.any(pred_mask):
+                    analysis_results.append(f"Empty mask generated for '{prompt}'")
+                    continue
                 
                 overlay_image = image.copy()
                 color = colors[idx % len(colors)]
                 mask_indices = pred_mask > 0.5
-                if mask_indices.any():
+                if np.any(mask_indices):  # Use np.any() for boolean array check
                     overlay_image[mask_indices] = color
                     results.append(overlay_image)
             except Exception as e: