Update app.py

app.py

@@ -79,20 +79,30 @@ def processar_imagem_de_entrada_wrapper(img, modelo):
     model = models[modelo]
     spent, img_out = processar_imagem_de_entrada(img, modelo, model)
 
-    # Define the function `has_disease`
-    def has_disease(img_out):
-        """
-        Checks if the angiogram has disease based on the segmentation.
-        Args:
-            img_out: The segmented angiogram.
-        Returns:
-            True if the angiogram has disease, False otherwise.
-        """
-        percentage_of_vessels = np.sum(img_out) / (img_out.shape[0] * img_out.shape[1])
-        if percentage_of_vessels > 0.5:
-            return "Sim"
-        else:
-            return "Não"
+    def has_disease(img, model, min_area=500):
+      """Checks if the angiogram has disease.
+    
+      Args:
+        img: The input angiogram.
+        model: The segmentation model.
+        min_area: The minimum area for an anomaly to be considered a disease.
+    
+      Returns:
+        True if the angiogram has disease, False otherwise.
+      """
+    
+      # Segment the angiogram.
+      mask = model.predict(img)
+    
+      # Detect anomalies in the mask.
+      anomalies = detect_anomalies(img, mask)
+    
+      # Check if any of the anomalies are large enough to be considered a disease.
+      for anomaly in anomalies:
+        if anomaly[1] >= min_area:
+          return True
+    
+      return False
 
     has_disease = has_disease(img_out)