Update app.py

app.py

@@ -17,16 +17,19 @@ def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
     return errors
 
 def suggest_intersections(A, B, C, U):
+    # Máximos valores permitidos para las intersecciones
     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)
     
+    # Mínimos valores permitidos para las intersecciones
     min_AB = max(0, A + B - U)
     min_AC = max(0, A + C - U)
     min_BC = max(0, B + C - U)
     min_ABC = max(0, A + B + C - U)
     
+    # Máximos y mínimos valores permitidos para los conjuntos
     max_A = U - (B + C - BC)
     max_B = U - (A + C - AC)
     max_C = U - (A + B - AB)
@@ -87,4 +90,45 @@ 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))
+    venn = venn3(subsets=(A, B, AB, C, AC, BC, ABC), set_labels=('A', 'B', 'C'))
+    plt.title(f"Diagrama de Venn con U = {U}")
     
+    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):
+    if U == 0:
+        U = A + B + C
+
+    errors = validate_inputs(A, B, C, AB, AC, BC, ABC, U)
+    suggestions = suggest_intersections(A, B, C, U)
+    if errors:
+        return None, {"error": "\n".join(errors), "sugerencias": suggestions}
+    
+    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, probabilities
+
+iface = gr.Interface(
+    fn=main,
+    inputs=[
+        gr.Number(label="Conjunto Universal (U)", value=0),
+        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")
+    ],
+    outputs=[
+        gr.Image(type="pil", label="Diagrama de Venn"),
+        gr.JSON(label="Resultados y Sugerencias")
+    ],
+    live=True
+)
+
+iface.launch()