Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import numpy as np 
+import matplotlib.pyplot as plt
 st.markdown("""
     <style>
     /* Set a soft background color */
@@ -80,3 +81,41 @@ st.markdown("""
 <li>Analysis</li>
 <li>Structuring</li>
 </ul>""",unsafe_allow_html=True)
+st.title("Drug Trial Analysis")
+
+# Input for number of participants
+total_participants = st.number_input("Total Participants:", min_value=100, max_value=10000, value=1000)
+
+# Input for number of participants in drug and placebo groups
+drug_group_size = st.number_input("Size of Drug Group:", min_value=1, max_value=total_participants - 1, value=500)
+placebo_group_size = total_participants - drug_group_size
+
+# Input for side effects
+side_effects_drug = st.number_input("Number of Side Effects in Drug Group:", min_value=0, max_value=drug_group_size, value=100)
+side_effects_placebo = st.number_input("Number of Side Effects in Placebo Group:", min_value=0, max_value=placebo_group_size, value=20)
+
+# Calculate probabilities
+p_side_effects_drug = side_effects_drug / drug_group_size
+p_side_effects_placebo = side_effects_placebo / placebo_group_size
+
+# Calculate Risk Difference and Relative Risk
+risk_difference = p_side_effects_drug - p_side_effects_placebo
+relative_risk = p_side_effects_drug / p_side_effects_placebo if p_side_effects_placebo > 0 else np.nan
+
+# Display results
+st.subheader("Results:")
+st.write(f"Probability of Side Effects in Drug Group: {p_side_effects_drug:.2%}")
+st.write(f"Probability of Side Effects in Placebo Group: {p_side_effects_placebo:.2%}")
+st.write(f"Risk Difference: {risk_difference:.2%}")
+st.write(f"Relative Risk: {relative_risk:.2f}" if not np.isnan(relative_risk) else "Relative Risk: Undefined (Placebo group has no side effects)")
+
+# Visualization
+st.subheader("Visualization")
+labels = ['Drug Group', 'Placebo Group']
+sizes = [p_side_effects_drug, p_side_effects_placebo]
+colors = ['lightblue', 'lightgreen']
+
+fig, ax = plt.subplots()
+ax.pie(sizes, labels=labels, colors=colors, autopct='%1.1f%%', startangle=90)
+ax.axis('equal')  # Equal aspect ratio ensures that pie chart is circular.
+st.pyplot(fig)