Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import matplotlib.pyplot as plt
-from matplotlib_venn import venn3
+from matplotlib_venn import venn2, venn3
 from io import BytesIO
 from PIL import Image
 import pandas as pd
@@ -25,6 +25,7 @@ def validate_inputs(U, A, B, C, AB, AC, BC, ABC):
         errors.append(f"A∩B∩C ({ABC}) no puede ser mayor que B∩C ({BC}).")
     
     return errors
+
 def suggest_intersections(U, A, B, C, AB, AC, BC, ABC):
     union_ABC = A + B + C - AB - AC - BC + ABC
     
@@ -141,8 +142,36 @@ def main(U, A, B, C, AB, AC, BC, ABC):
     
     return venn_image, probabilities_df, suggestions
 
-# Gradio Interface
-interface = gr.Interface(
+def draw_venn_2(A, B, AB):
+    plt.figure(figsize=(10, 10))
+    venn = venn2(subsets=(A - AB, B - AB, AB), set_labels=('A', 'B'))
+    img = BytesIO()
+    plt.title(f"Diagrama de Venn para A y B")
+    plt.savefig(img, format='png')
+    img.seek(0)
+    image = Image.open(img)
+    return image
+
+def main_2(A, B, AB):
+    venn_image = draw_venn_2(A, B, AB)
+    return venn_image
+
+# Interfaz para dos conjuntos A y B
+interface_2 = gr.Interface(
+    fn=main_2,
+    inputs=[
+        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.",
+    live=True
+)
+
+# Interfaz para tres conjuntos A, B y C (usando tu código original)
+interface_3 = gr.Interface(
     fn=main,
     inputs=[
         gr.Number(label="U (Universal Set)"),
@@ -158,12 +187,14 @@ interface = gr.Interface(
         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",
-    description="Calcula las probabilidades, intersecciones sugeridas y genera un diagrama de Venn.",
+    title="Calculadora de Probabilidades y Diagrama de Venn (Tres conjuntos)",
+    description="Calcula las probabilidades, intersecciones sugeridas y genera un diagrama de Venn para tres conjuntos.",
     live=True
 )
 
+# Combinar ambas interfaces en una tabbed interface
+tabbed_interface = gr.TabbedInterface([interface_2, interface_3], ["Dos conjuntos (A y B)", "Tres conjuntos (A, B y C)"])
+
 if __name__ == "__main__":
-    interface.launch()
+    tabbed_interface.launch()