Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import matplotlib.pyplot as plt
+from matplotlib.patches import Rectangle
 from matplotlib_venn import venn3
 from io import BytesIO
 from PIL import Image
@@ -104,6 +105,14 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     return formatted_probs
 
 def plot_venn(A, B, C, AB, AC, BC, ABC, U):
+    plt.figure(figsize=(8, 8))
+    
+    # Dibujar el conjunto universal
+    ax = plt.gca()
+    ax.add_patch(Rectangle((0, 0), 1, 1, fill=None, edgecolor='black', linewidth=2))
+    ax.set_xlim(-0.1, 1.1)
+    ax.set_ylim(-0.1, 1.1)
+    
     subsets = {
         '100': A - AB - AC + ABC,
         '010': B - AB - BC + ABC,
@@ -113,7 +122,6 @@ def plot_venn(A, B, C, AB, AC, BC, ABC, U):
         '011': BC - ABC,
         '111': ABC
     }
-    plt.figure(figsize=(8, 8))
     venn = venn3(subsets=subsets, set_labels=('A', 'B', 'C'))
     plt.title(f"Diagrama de Venn con U = {U}")
     
@@ -134,7 +142,7 @@ def main(A, B, C, AB, AC, BC, ABC, U):
     
     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
+    return venn_diagram, {"probabilidades": probabilities, "sugerencias": suggestions}
 
 iface = gr.Interface(
     fn=main,