import stan
import seaborn as sns
import matplotlib.pyplot as plt

model = """
parameters {
  real <lower=0, upper=1> p;
}

model {
  p ~ beta(2.0,2.0);
  1 ~ bernoulli(p);
}
"""

posterior = stan.build(model)

fit = posterior.sample(num_chains=4, num_samples=5000)
df = fit.to_frame()
print(df.describe().T)
sns.histplot(data=df, x="p", bins=50)
plt.savefig('plot.pdf')