Update app.py

app.py

@@ -17,54 +17,8 @@ def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
         errors.append("El conjunto universal U es menor que la suma total de los conjuntos y sus intersecciones.")
     return errors
 
-def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
-    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))
-    max_ABC = min(max_AB, max_AC, max_BC)
-    
-    min_AB = max(0, A + B - U + C)
-    min_AC = max(0, A + C - U + B)
-    min_BC = max(0, B + C - U + A)
-    min_ABC = max(0, A + B + C - U)
-    
-    max_A = U - (B + C - BC)
-    max_B = U - (A + C - AC)
-    max_C = U - (A + B - AB)
-    min_A = max(AB + AC - ABC, 0)
-    min_B = max(AB + BC - ABC, 0)
-    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,
-    }
-    return suggestions
-
 def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     total = U
-    if total == 0:
-        return {
-            "P(A)": 0,
-            "P(B)": 0,
-            "P(C)": 0,
-            "P(A ∩ B)": 0,
-            "P(A ∩ C)": 0,
-            "P(B ∩ C)": 0,
-            "P(A ∩ B ∩ C)": 0,
-        }
 
     P_A = A / total
     P_B = B / total
@@ -74,7 +28,6 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     P_BC = BC / total
     P_ABC = ABC / total
 
-    # Probabilidades condicionales
     P_A_given_B = P_AB / P_B if P_B > 0 else 0
     P_B_given_A = P_AB / P_A if P_A > 0 else 0
     P_A_given_C = P_AC / P_C if P_C > 0 else 0
@@ -82,7 +35,6 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     P_B_given_C = P_BC / P_C if P_C > 0 else 0
     P_C_given_B = P_BC / P_B if P_B > 0 else 0
 
-    # Teorema de Bayes
     P_B_given_A_bayes = (P_A_given_B * P_B) / P_A if P_A > 0 else 0
 
     formatted_probs = {
@@ -107,12 +59,11 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
 def plot_venn(A, B, C, AB, AC, BC, ABC, U):
     plt.figure(figsize=(8, 8))
     
-    # Dibujar el conjunto universal
+    # 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))
-    ax.set_xlim(-0.1, 1.1)
-    ax.set_ylim(-0.1, 1.1)
     
+    # Dibujar el diagrama de Venn
     subsets = {
         '100': A - AB - AC + ABC,
         '010': B - AB - BC + ABC,
@@ -123,6 +74,13 @@ def plot_venn(A, B, C, AB, AC, BC, ABC, U):
         '111': ABC
     }
     venn = venn3(subsets=subsets, set_labels=('A', 'B', 'C'))
+    
+    # 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()
@@ -131,18 +89,50 @@ def plot_venn(A, B, C, AB, AC, BC, ABC, U):
     img = Image.open(buf)
     return img
 
-def main(A, B, C, AB, AC, BC, ABC, U):
-    if U == 0:
-        U = A + B + C
+def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
+    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))
+    max_ABC = min(max_AB, max_AC, max_BC)
+    
+    min_AB = max(0, A + B - U + C)
+    min_AC = max(0, A + C - U + B)
+    min_BC = max(0, B + C - U + A)
+    min_ABC = max(0, A + B + C - U)
+    
+    max_A = U - (B + C - BC)
+    max_B = U - (A + C - AC)
+    max_C = U - (A + B - AB)
+    min_A = max(AB + AC - ABC, 0)
+    min_B = max(AB + BC - ABC, 0)
+    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,
+    }
+    return suggestions
+
+def main(A, B, C, AB, AC, BC, ABC, U):
     errors = validate_inputs(A, B, C, AB, AC, BC, ABC, U)
-    suggestions = suggest_intersections(A, B, C, AB, AC, BC, ABC, U)
     if errors:
-        return None, {"error": "\n".join(errors), "sugerencias": suggestions}
+        return None, {"error": "\n".join(errors), "sugerencias": suggest_intersections(A, B, C, AB, AC, BC, ABC, U)}
     
     venn_diagram = plot_venn(A, B, C, AB, AC, BC, ABC, U)
     probabilities = calculate_probabilities(A, B, C, AB, AC, BC, ABC, U)
-    return venn_diagram, {"probabilidades": probabilities, "sugerencias": suggestions}
+    return venn_diagram, {"probabilidades": probabilities, "sugerencias": suggest_intersections(A, B, C, AB, AC, BC, ABC, U)}
 
 iface = gr.Interface(
     fn=main,