Update app.py

app.py

@@ -91,7 +91,35 @@ 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):
+    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:
+                goal_reached.append({'Medication': drug, 'Year': int(year)})
+                break  # Stop checking further years for this drug
+
+    # Sort by year to prioritize earlier achievement
+    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)")
+    
+    if not goal_summary:
+        st.info("No medications reach the target BMD.")
+    else:
+        summary_table = pd.DataFrame(goal_summary)
+        st.table(summary_table)
+
 # Streamlit UI
+# Main function
 def main():
     st.title("BMD and T-score Prediction Tool")
     
@@ -159,6 +187,11 @@ def main():
         if not filtered_predictions:
             st.warning("No medications selected. Please select at least one medication or use the 'Show All' option.")
         else:
+            # Generate and display goal treatment summary
+            goal_summary = generate_goal_summary(filtered_predictions, target_bmd=2.4)
+            display_goal_summary(goal_summary)
+            
+            # Display individual medication results
             display_results(filtered_predictions, selected_site)
 
 if __name__ == "__main__":