Update app.py

app.py

@@ -83,3 +83,48 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     # Convert to DataFrame for better visualization
     df = pd.DataFrame(list(formatted_probs.items()), columns=["Descripción", "Valor"])
     return df
+
+def draw_venn(A, B, C, AB, AC, BC, ABC):
+    plt.figure(figsize=(10,10))
+    venn = venn3(subsets=(A, B, AB, C, AC, BC, ABC), set_labels=('A', 'B', 'C'))
+    img = BytesIO()
+    plt.savefig(img, format='png')
+    img.seek(0)
+    image = Image.open(img)
+    return image
+
+def main(A, B, C, AB, AC, BC, ABC, U):
+    errors = validate_inputs(A, B, C, AB, AC, BC, ABC, U)
+    if errors:
+        return f"Errores de validación: {', '.join(errors)}", None, None
+    
+    suggestions = suggest_intersections(A, B, C, U)
+    probabilities_df = calculate_probabilities(A, B, C, AB, AC, BC, ABC, U)
+    venn_image = draw_venn(A, B, C, AB, AC, BC, ABC)
+    
+    return suggestions, probabilities_df, venn_image
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=main,
+    inputs=[
+        gr.inputs.Number(label="A"),
+        gr.inputs.Number(label="B"),
+        gr.inputs.Number(label="C"),
+        gr.inputs.Number(label="A ∩ B"),
+        gr.inputs.Number(label="A ∩ C"),
+        gr.inputs.Number(label="B ∩ C"),
+        gr.inputs.Number(label="A ∩ B ∩ C"),
+        gr.inputs.Number(label="U (Universal Set)")
+    ],
+    outputs=[
+        gr.outputs.JSON(label="Sugerencias de Intersección"),
+        gr.outputs.Dataframe(label="Tabla de Probabilidades"),
+        gr.outputs.Image(type="pil", label="Diagrama de Venn")
+    ],
+    title="Calculadora de Probabilidades y Diagrama de Venn",
+    description="Calcula las probabilidades, intersecciones sugeridas y genera un diagrama de Venn."
+)
+
+if __name__ == "__main__":
+    interface.launch()