Update app.py

app.py

@@ -102,11 +102,12 @@ def processar_imagem_de_entrada_wrapper(img, modelo):
             continue
         
         # Calcular forma do cluster usando a relação entre área e perímetro
-        _, contours, _ = cv2.findContours(np.uint8(cluster_mask), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-        perimeter = cv2.arcLength(contours[0], True)
-        compactness = 4 * np.pi * area / (perimeter ** 2)
-        
-        cluster_features.append({'area': area, 'compactness': compactness})
+        contours, _ = cv2.findContours(np.uint8(cluster_mask), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+        if len(contours) > 0:
+            perimeter = cv2.arcLength(contours[0], True)
+            compactness = 4 * np.pi * area / (perimeter ** 2)
+            
+            cluster_features.append({'area': area, 'compactness': compactness})
     
     # Decidir se há doença com base nas características dos clusters
     has_disease_flag = any(feature['area'] >= 200 and feature['compactness'] < 0.3 for feature in cluster_features)
@@ -128,6 +129,7 @@ def processar_imagem_de_entrada_wrapper(img, modelo):
     return spent, img_out, status_doenca, explanation
 
 
+
 # Criar a interface Gradio
 my_app = gr.Interface(
     fn=processar_imagem_de_entrada_wrapper,