Update app.py

app.py

@@ -201,11 +201,11 @@ def validate_inputs_2(U, A, B, AB):
     
     return errors
 
-def draw_venn_2(A, B, AB):
+def draw_venn_2(U, 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.title(f"Diagrama de Venn U = {U}"
     plt.savefig(img, format='png')
     img.seek(0)
     image = Image.open(img)
@@ -216,7 +216,7 @@ def main_2(U, A, B, AB):
     if errors:
         return None, pd.DataFrame({"Errores de validación": errors}), {}
 
-    venn_image = draw_venn_2(A, B, AB)
+    venn_image = draw_venn_2(U, A, B, AB)
     probabilities_df = calculate_probabilities_2(U, A, B, AB)
     suggestions = suggest_intersections_2(U, A, B, AB)