Update app.py

app.py

@@ -34,22 +34,21 @@ def generate_predictions(medication_data, site, bmd, mu, sigma):
         drug = row['Medication']
         predictions = {'Year': ['0'], 'Predicted BMD': [round(bmd, 3)], 'Predicted T-score': [round(calculate_tscore(bmd, mu, sigma), 1)]}
         
-        baseline_bmd = bmd
         for year in row.index[1:-1]:  # Skip 'Medication' and 'Site' columns
             if not pd.isna(row[year]):
                 percentage_increase = row[year]
-                predicted_bmd = calculate_bmd(baseline_bmd, percentage_increase)
+                # BMD calculated using initial BMD (from user input)
+                predicted_bmd = bmd * (1 + percentage_increase / 100)
                 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))
-                
-                baseline_bmd = predicted_bmd
         
         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}")