Update app.py

app.py

@@ -1,19 +1,21 @@
 import gradio as gr
 import matplotlib.pyplot as plt
-from matplotlib_venn import venn3, venn3_circles
+from matplotlib.patches import Rectangle
+from matplotlib_venn import venn3
 from io import BytesIO
 from PIL import Image
 
 def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
+    total_needed = A + B + C - AB - AC - BC + ABC
     errors = []
+    if total_needed > U:
+        errors.append("El conjunto universal U debe ser mayor o igual a la suma de los subconjuntos y sus intersecciones.")
     if A < AB + AC - ABC:
         errors.append("A no puede ser menor que la suma de AB y AC menos ABC.")
     if B < AB + BC - ABC:
         errors.append("B no puede ser menor que la suma de AB y BC menos ABC.")
     if C < AC + BC - ABC:
         errors.append("C no puede ser menor que la suma de AC y BC menos ABC.")
-    if U < A + B + C - AB - AC - BC + ABC:
-        errors.append("El conjunto universal U es menor que la suma total de los conjuntos y sus intersecciones.")
     return errors
 
 def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
@@ -56,8 +58,13 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     return formatted_probs
 
 def plot_venn(A, B, C, AB, AC, BC, ABC, U):
-    plt.figure(figsize=(10, 10))
+    plt.figure(figsize=(8, 8))
     
+    # Dibujar el conjunto universal como un rectángulo
+    ax = plt.gca()
+    ax.add_patch(Rectangle((0, 0), 1, 1, fill=None, edgecolor='black', linewidth=2))
+    
+    # Dibujar el diagrama de Venn
     subsets = {
         '100': A - AB - AC + ABC,
         '010': B - AB - BC + ABC,
@@ -68,8 +75,13 @@ def plot_venn(A, B, C, AB, AC, BC, ABC, U):
         '111': ABC
     }
     venn = venn3(subsets=subsets, set_labels=('A', 'B', 'C'))
-    venn_circles = venn3_circles(subsets=subsets, linewidth=1.0)
     
+    # Mostrar la cantidad fuera de A, B y C si no coincide con U
+    suma_subconjuntos = A + B + C - AB - AC - BC + ABC
+    if U > suma_subconjuntos:
+        faltante = U - suma_subconjuntos
+        plt.text(0.5, 0.9, f"Fuera de A, B y C: {faltante}", horizontalalignment='center', fontsize=12, bbox=dict(facecolor='white', alpha=0.5))
+
     plt.title(f"Diagrama de Venn con U = {U}")
     
     buf = BytesIO()
@@ -79,6 +91,7 @@ def plot_venn(A, B, C, AB, AC, BC, ABC, U):
     return img
 
 def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
+    # Ajuste de los valores sugeridos para que sean realistas y no negativos
     max_AB = min(A, B, U - (A + B + C - AB - AC - BC + ABC))
     max_AC = min(A, C, U - (A + B + C - AB - AC - BC + ABC))
     max_BC = min(B, C, U - (A + B + C - AB - AC - BC + ABC))
@@ -97,20 +110,20 @@ def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
     min_C = max(AC + BC - ABC, 0)
 
     suggestions = {
-        "Máximo valor sugerido para A": max_A,
-        "Mínimo valor sugerido para A": min_A,
-        "Máximo valor sugerido para B": max_B,
-        "Mínimo valor sugerido para B": min_B,
-        "Máximo valor sugerido para C": max_C,
-        "Mínimo valor sugerido para C": min_C,
-        "Máximo valor sugerido para A ∩ B": max_AB,
-        "Mínimo valor sugerido para A ∩ B": min_AB,
-        "Máximo valor sugerido para A ∩ C": max_AC,
-        "Mínimo valor sugerido para A ∩ C": min_AC,
-        "Máximo valor sugerido para B ∩ C": max_BC,
-        "Mínimo valor sugerido para B ∩ C": min_BC,
-        "Máximo valor sugerido para A ∩ B ∩ C": max_ABC,
-        "Mínimo valor sugerido para A ∩ B ∩ C": min_ABC,
+        "Máximo valor sugerido para A": max(0, max_A),
+        "Mínimo valor sugerido para A": max(0, min_A),
+        "Máximo valor sugerido para B": max(0, max_B),
+        "Mínimo valor sugerido para B": max(0, min_B),
+        "Máximo valor sugerido para C": max(0, max_C),
+        "Mínimo valor sugerido para C": max(0, min_C),
+        "Máximo valor sugerido para A ∩ B": max(0, max_AB),
+        "Mínimo valor sugerido para A ∩ B": max(0, min_AB),
+        "Máximo valor sugerido para A ∩ C": max(0, max_AC),
+        "Mínimo valor sugerido para A ∩ C": max(0, min_AC),
+        "Máximo valor sugerido para B ∩ C": max(0, max_BC),
+        "Mínimo valor sugerido para B ∩ C": max(0, min_BC),
+        "Máximo valor sugerido para A ∩ B ∩ C": max(0, max_ABC),
+        "Mínimo valor sugerido para A ∩ B ∩ C": max(0, min_ABC),
     }
     return suggestions