Update app.py

app.py

@@ -5,6 +5,8 @@ from io import BytesIO
 from PIL import Image
 import pandas as pd
 
+# Funciones para el caso de tres conjuntos A, B y C
+
 def validate_inputs(U, A, B, C, AB, AC, BC, ABC):
     union_ABC = A + B + C - AB - AC - BC + ABC
     errors = []
@@ -116,7 +118,7 @@ def calculate_probabilities(U, A, B, C, AB, AC, BC, ABC):
         "Complemento de A U B U C": U - (A + B + C - AB - AC - BC + ABC)
     }
     
-    # Convert to DataFrame for better visualization
+    # Convertir a DataFrame para mejor visualización
     df = pd.DataFrame(list(formatted_probs.items()), columns=["Descripción", "Valor"])
     return df
 
@@ -133,7 +135,6 @@ def draw_venn(U, A, B, C, AB, AC, BC, ABC):
 def main(U, A, B, C, AB, AC, BC, ABC):
     errors = validate_inputs(U, A, B, C, AB, AC, BC, ABC)
     if errors:
-        # Devolver None para la imagen y el DataFrame si hay errores
         return None, pd.DataFrame({"Errores de validación": errors}), {}
 
     venn_image = draw_venn(U, A, B, C, AB, AC, BC, ABC)
@@ -142,6 +143,64 @@ def main(U, A, B, C, AB, AC, BC, ABC):
     
     return venn_image, probabilities_df, suggestions
 
+# Funciones para el caso de dos conjuntos A y B
+
+def calculate_probabilities_2(U, A, B, AB):
+    total = U if U > 0 else (A + B - AB)
+    if total == 0:
+        return {
+            "P(A)": 0,
+            "P(B)": 0,
+            "P(A ∩ B)": 0,
+        }
+
+    P_A = A / total
+    P_B = B / total
+    P_AB = AB / total
+
+    PA_given_B = P_AB / P_B if P_B > 0 else 0
+    PB_given_A = P_AB / P_A if P_A > 0 else 0
+
+    P_A_union_B = P_A + P_B - P_AB
+
+    formatted_probs = {
+        "P(A)": f"{P_A:.2%} ({A}/{total})",
+        "P(B)": f"{P_B:.2%} ({B}/{total})",
+        "P(A ∩ B)": f"{P_AB:.2%} ({AB}/{total})",
+        "P(A | B)": f"{PA_given_B:.2%} (P(A ∩ B) / P(B)) = ({P_AB:.4f} / {P_B:.4f}) = ((A ∩ B) / B) = ({AB} / {B})",
+        "P(B | A)": f"{PB_given_A:.2%} (P(B ∩ A) / P(A)) = ({P_AB:.4f} / {P_A:.4f}) = ((B ∩ A) / A) = ({AB} / {A})",
+        "P(A ∪ B)": f"{P_A_union_B:.2%} (P(A) + P(B) - P(A ∩ B)) = ({P_A:.4f} + {P_B:.4f} - {P_AB:.4f})  = (A + B - A ∩ B) / U = ({A} + {B} - {AB}) / {U}",
+        "U (Universal Set)": total,
+        "Complemento de A U B": U - (A + B - AB)
+    }
+
+    df = pd.DataFrame(list(formatted_probs.items()), columns=["Descripción", "Valor"])
+    return df
+
+def suggest_intersections_2(U, A, B, AB):
+    union_AB = A + B - AB
+    
+    min_AB = min(A, B)
+    max_AB = min(A, B)
+    suggestions = {
+        "Mínimo valor sugerido para U": union_AB,
+        "Valor sugerido para A ∩ B": f"{min_AB} - {max_AB}",
+    }
+    return suggestions
+
+def validate_inputs_2(U, A, B, AB):
+    union_AB = A + B - AB
+    errors = []
+    
+    if U < union_AB:
+        errors.append(f"El conjunto universal U ({U}) no puede ser menor que la unión de A y B ({union_AB}).")
+    if A < AB:
+        errors.append(f"A ({A}) no puede ser menor que A∩B ({AB}).")
+    if B < AB:
+        errors.append(f"B ({B}) no puede ser menor que A∩B ({AB}).")
+    
+    return errors
+
 def draw_venn_2(A, B, AB):
     plt.figure(figsize=(10, 10))
     venn = venn2(subsets=(A - AB, B - AB, AB), set_labels=('A', 'B'))
@@ -152,25 +211,37 @@ def draw_venn_2(A, B, AB):
     image = Image.open(img)
     return image
 
-def main_2(A, B, AB):
+def main_2(U, A, B, AB):
+    errors = validate_inputs_2(U, A, B, AB)
+    if errors:
+        return None, pd.DataFrame({"Errores de validación": errors}), {}
+
     venn_image = draw_venn_2(A, B, AB)
-    return venn_image
+    probabilities_df = calculate_probabilities_2(U, A, B, AB)
+    suggestions = suggest_intersections_2(U, A, B, AB)
+    
+    return venn_image, probabilities_df, suggestions
 
 # Interfaz para dos conjuntos A y B
 interface_2 = gr.Interface(
     fn=main_2,
     inputs=[
+        gr.Number(label="U (Universal Set)"),
         gr.Number(label="A"),
         gr.Number(label="B"),
         gr.Number(label="A ∩ B")
     ],
-    outputs=gr.Image(type="pil", label="Diagrama de Venn"),
-    title="Diagrama de Venn para dos conjuntos (A y B)",
-    description="Genera un diagrama de Venn para dos conjuntos A y B.",
+    outputs=[
+        gr.Image(type="pil", label="Diagrama de Venn"),
+        gr.Dataframe(label="Tabla de Probabilidades"),
+        gr.JSON(label="Sugerencias de Intersección")
+    ],
+    title="Calculadora de Probabilidades y Diagrama de Venn para dos conjuntos (A y B)",
+    description="Calcula las probabilidades, intersecciones sugeridas y genera un diagrama de Venn para dos conjuntos.",
     live=True
 )
 
-# Interfaz para tres conjuntos A, B y C (usando tu código original)
+# Interfaz para tres conjuntos A, B y C
 interface_3 = gr.Interface(
     fn=main,
     inputs=[
@@ -188,7 +259,7 @@ interface_3 = gr.Interface(
         gr.Dataframe(label="Tabla de Probabilidades"),
         gr.JSON(label="Sugerencias de Intersección")
     ],
-    title="Calculadora de Probabilidades y Diagrama de Venn (Tres conjuntos)",
+    title="Calculadora de Probabilidades y Diagrama de Venn para tres conjuntos (A, B y C)",
     description="Calcula las probabilidades, intersecciones sugeridas y genera un diagrama de Venn para tres conjuntos.",
     live=True
 )