File size: 3,866 Bytes
fe0e27c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import streamlit as st
import pandas as pd
import plotly.graph_objs as go
# Constants from linear regression
REGRESSION_CONSTANTS = {
'FN': {'mu': 0.916852, 'sigma': 0.120754},
'TH': {'mu': 0.955439, 'sigma': 0.125406},
'LS': {'mu': 1.131649, 'sigma': 0.139618},
}
# Load medication data
@st.cache_data
def load_medication_data():
file_path = "cleaned_bmd_medication_data.xlsx"
return pd.read_excel(file_path)
# Calculate predicted BMD after medication
def calculate_bmd(bmd, percentage_increase):
return bmd * (1 + percentage_increase)
# Convert BMD to T-score
def calculate_tscore(bmd, mu, sigma):
return (bmd - mu) / sigma
# Generate prediction table for all drugs
def generate_predictions(medication_data, site, bmd, mu, sigma):
site_data = medication_data[medication_data['Site'] == site]
all_results = []
for _, row in site_data.iterrows():
drug = row['Medication']
predictions = {'Year': ['0'], 'Predicted BMD': [round(bmd, 3)], 'Predicted T-score': [round(calculate_tscore(bmd, mu, sigma), 1)]}
for year in row.index[1:-1]: # Skip 'Medication' and 'Site' columns
if not pd.isna(row[year]):
percentage_increase = row[year]
# BMD calculated using initial BMD (from user input)
predicted_bmd = bmd * (1 + percentage_increase)
predicted_tscore = calculate_tscore(predicted_bmd, mu, sigma)
predictions['Year'].append(year.replace(" Year", "")) # Simplify year label
predictions['Predicted BMD'].append(round(predicted_bmd, 3))
predictions['Predicted T-score'].append(round(predicted_tscore, 1))
all_results.append({'Drug': drug, 'Predictions': predictions})
return all_results
# Display results as table and plots
def display_results(predictions, site):
st.subheader(f"Predictions for {site}")
for result in predictions:
drug = result['Drug']
predictions = result['Predictions']
# Display table
st.write(f"### {drug}")
st.dataframe(pd.DataFrame(predictions))
# Plot BMD and T-score
bmd_plot = go.Scatter(
x=predictions['Year'], y=predictions['Predicted BMD'], mode='lines+markers',
name='Predicted BMD', line=dict(color='blue')
)
tscore_plot = go.Scatter(
x=predictions['Year'], y=predictions['Predicted T-score'], mode='lines+markers',
name='Predicted T-score', line=dict(color='green')
)
# Combine plots in a single row
col1, col2 = st.columns(2)
with col1:
st.plotly_chart(go.Figure(data=[bmd_plot], layout=go.Layout(
title=f"{drug} - Predicted BMD", xaxis_title="Year", yaxis_title="BMD (g/cm²)"
)))
with col2:
st.plotly_chart(go.Figure(data=[tscore_plot], layout=go.Layout(
title=f"{drug} - Predicted T-score", xaxis_title="Year", yaxis_title="T-score"
)))
# Streamlit UI
def main():
st.title("BMD and T-score Prediction Tool")
# Input patient data
bmd_patient = st.number_input(
"Initial BMD",
min_value=0.000, max_value=2.000,
value=0.800, step=0.001,
format="%.3f"
)
site_options = ['FN', 'TH', 'LS']
site = st.selectbox("Select Region (Site)", site_options)
# Load constants and medication data
constants = REGRESSION_CONSTANTS[site]
medication_data = load_medication_data()
# Generate and display predictions
if st.button("Predict"):
predictions = generate_predictions(medication_data, site, bmd_patient, constants['mu'], constants['sigma'])
display_results(predictions, site)
if __name__ == "__main__":
main()
|