Update app.py

app.py

@@ -162,15 +162,10 @@ def read_excel_sheets(file):
         return None
 
 def z_testes(n1, n2, p1, p2):
-    """Performs Z-test for proportions and returns p-value."""
-    try:
-        pooled_p = (n1 * p1 + n2 * p2) / (n1 + n2)
-        se = np.sqrt(pooled_p * (1 - pooled_p) * (1 / n1 + 1 / n2))
-        z = (p1 - p2) / se
-        p_value = 2 * (1 - norm.cdf(abs(z)))
-        return p_value
-    except ZeroDivisionError:
-        return np.nan
+    p_hat = (p1 + p2) / (n1 + n2)
+    z = (p1 / n1 - p2 / n2) / ((p_hat * (1 - p_hat) * (1 / n1 + 1 / n2)) ** 0.5)
+    p_value = 2 * (1 - norm.cdf(abs(z)))
+    return p_value
 
 def z_test_data(df):
     styles = pd.DataFrame('', index=df.index, columns=df.columns)