Update app.py

app.py

@@ -5,7 +5,7 @@ from io import BytesIO
 from PIL import Image
 import pandas as pd
 
-def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
+def validate_inputs(U, A, B, C, AB, AC, BC, ABC):
     union_ABC = A + B + C - AB - AC - BC + ABC
     errors = []
     
@@ -25,7 +25,7 @@ def validate_inputs(A, B, C, AB, AC, BC, ABC, U):
         errors.append(f"A∩B∩C ({ABC}) no puede ser mayor que B∩C ({BC}).")
     
     return errors
-def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
+def suggest_intersections(U, A, B, C, AB, AC, BC, ABC):
     union_ABC = A + B + C - AB - AC - BC + ABC
     
     min_AB = min(A, B, ABC)
@@ -45,7 +45,7 @@ def suggest_intersections(A, B, C, AB, AC, BC, ABC, U):
     }
     return suggestions
 
-def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
+def calculate_probabilities(U, A, B, C, AB, AC, BC, ABC):
     total = U if U > 0 else (A + B + C - AB - AC - BC + ABC)
     if total == 0:
         return {
@@ -119,7 +119,7 @@ def calculate_probabilities(A, B, C, AB, AC, BC, ABC, U):
     df = pd.DataFrame(list(formatted_probs.items()), columns=["Descripción", "Valor"])
     return df
 
-def draw_venn(A, B, C, AB, AC, BC, ABC, U):
+def draw_venn(U, A, B, C, AB, AC, BC, ABC):
     plt.figure(figsize=(10,10))
     venn = venn3(subsets=(max(0, A - AB - AC + ABC), max(0,B - AB - BC + ABC), max(0,AB - ABC), max(0,C- AC - BC + ABC), max(AC - ABC, 0), max(BC - ABC,0), ABC), set_labels=('A', 'B', 'C'))
     img = BytesIO()
@@ -129,15 +129,15 @@ def draw_venn(A, B, C, AB, AC, BC, ABC, U):
     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)
+def main(U, A, B, C, AB, AC, BC, ABC):
+    errors = validate_inputs(U, A, B, C, AB, AC, BC, ABC)
     if errors:
         # Devolver None para la imagen y el DataFrame si hay errores
         return None, pd.DataFrame({"Errores de validación": errors}), {}
 
-    venn_image = draw_venn(A, B, C, AB, AC, BC, ABC, U)
-    probabilities_df = calculate_probabilities(A, B, C, AB, AC, BC, ABC, U)
-    suggestions = suggest_intersections(A, B, C, AB, AC, BC, ABC, U)
+    venn_image = draw_venn(U, A, B, C, AB, AC, BC, ABC)
+    probabilities_df = calculate_probabilities(U, A, B, C, AB, AC, BC, ABC)
+    suggestions = suggest_intersections(U, A, B, C, AB, AC, BC, ABC)
     
     return venn_image, probabilities_df, suggestions