Update app.py

app.py

@@ -91,16 +91,16 @@ def display_results(predictions, site):
                 xaxis=dict(tickmode='array', tickvals=predictions['Year Index'], ticktext=predictions['Year'])
             )))
 
-# Generate summary of medications reaching the target BMD
-def generate_goal_summary(predictions, target_bmd=2.4):
+# Generate summary of medications reaching the target T-score
+def generate_goal_summary(predictions, target_tscore=2.4):
     goal_reached = []
     
     for result in predictions:
         drug = result['Drug']
         predictions_data = result['Predictions']
         
-        for year, bmd in zip(predictions_data['Year'], predictions_data['Predicted BMD']):
-            if bmd >= target_bmd:
+        for year, tscore in zip(predictions_data['Year'], predictions_data['Predicted T-score']):
+            if tscore >= target_tscore:  # Check against Predicted T-score
                 goal_reached.append({'Medication': drug, 'Year': int(year)})
                 break  # Stop checking further years for this drug
 
@@ -108,6 +108,7 @@ def generate_goal_summary(predictions, target_bmd=2.4):
     goal_reached_sorted = sorted(goal_reached, key=lambda x: x['Year'])
     return goal_reached_sorted
 
+
 # Display summary of goal-reaching medications
 def display_goal_summary(goal_summary):
     st.subheader("Goal Treatment Summary (BMD ≥ 2.4)")