File size: 3,417 Bytes
ce05002
 
2baf94a
ce04ad4
2273894
ce05002
e0ff385
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e5d5105
6fd93c8
 
 
 
 
 
 
 
 
 
 
 
e5d5105
 
 
 
 
 
 
 
e0ff385
 
 
 
 
 
 
 
6fd93c8
e0ff385
 
e5d5105
9c6afa0
2baf94a
6fd93c8
 
 
e5d5105
 
 
 
 
 
451e5be
e0ff385
 
 
 
 
9c6afa0
f04f86a
6fd93c8
f04f86a
 
 
 
 
 
 
 
 
 
6fd93c8
 
f04f86a
 
 
e0ff385
f04f86a
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import gradio as gr
import matplotlib.pyplot as plt
from matplotlib_venn import venn3
from io import BytesIO
from PIL import Image

def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
    errors = []
    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 suggest_intersections(A, B, C, U):
    max_AB = min(A, B, U - (A + B + C - A - B))
    max_AC = min(A, C, U - (A + B + C - A - C))
    max_BC = min(B, C, U - (A + B + C - B - C))
    max_ABC = min(max_AB, max_AC, max_BC)
    
    suggestions = {
        "Máximo valor sugerido para A ∩ B": max_AB,
        "Máximo valor sugerido para A ∩ C": max_AC,
        "Máximo valor sugerido para B ∩ C": max_BC,
        "Máximo valor sugerido para A ∩ B ∩ C": max_ABC,
    }
    return suggestions

def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
    total = U if U > 0 else (A + B + C - AB - AC - BC + ABC)
    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
    P_C = C / total
    P_AB = AB / total
    P_AC = AC / total
    P_BC = BC / total
    P_ABC = ABC / total
    
    formatted_probs = {
        "P(A)": f"{P_A:.2%} ({A}/{total})",
        "P(B)": f"{P_B:.2%} ({B}/{total})",
        "P(C)": f"{P_C:.2%} ({C}/{total})",
        "P(A ∩ B)": f"{P_AB:.2%} ({AB}/{total})",
        "P(A ∩ C)": f"{P_AC:.2%} ({AC}/{total})",
        "P(B ∩ C)": f"{P_BC:.2%} ({BC}/{total})",
        "P(A ∩ B ∩ C)": f"{P_ABC:.2%} ({ABC}/{total})",
    }
    
    return formatted_probs

def plot_venn(A, B, C, AB, AC, BC, ABC):
    plt.figure(figsize=(8, 8))
    venn = venn3(subsets=(A, B, AB, C, AC, BC, ABC), set_labels=('A', 'B', 'C'))
    plt.title("Venn Diagram")
    
    buf = BytesIO()
    plt.savefig(buf, format='png')
    buf.seek(0)
    img = Image.open(buf)
    return img

def main(A, B, C, AB, AC, BC, ABC, U):
    errors = validate_inputs(A, B, C, AB, AC, BC, ABC, U)
    if errors:
        suggestions = suggest_intersections(A, B, C, U)
        return None, {"error": "\n".join(errors), "sugerencias": suggestions}
    
    venn_diagram = plot_venn(A, B, C, AB, AC, BC, ABC)
    probabilities = calculate_probabilities(A, B, C, AB, AC, BC, ABC, U)
    return venn_diagram, probabilities

iface = gr.Interface(
    fn=main,
    inputs=[
        gr.Number(label="Cantidad en A"),
        gr.Number(label="Cantidad en B"),
        gr.Number(label="Cantidad en C"),
        gr.Number(label="Cantidad en A ∩ B"),
        gr.Number(label="Cantidad en A ∩ C"),
        gr.Number(label="Cantidad en B ∩ C"),
        gr.Number(label="Cantidad en A ∩ B ∩ C"),
        gr.Number(label="Conjunto Universal (U)")
    ],
    outputs=[
        gr.Image(type="pil", label="Diagrama de Venn"),
        gr.JSON(label="Resultados")
    ],
    live=True
)

iface.launch()