Update appbackup.py

appbackup.py

@@ -91,7 +91,90 @@ def display_results(predictions, site):
                 xaxis=dict(tickmode='array', tickvals=predictions['Year Index'], ticktext=predictions['Year'])
             )))
 
+# Generate summary of medications reaching the target T-score
+def generate_goal_summary(predictions, target_tscore=-2.4):
+    def year_to_int(year):
+        # Convert "1st", "2nd", "3rd", etc., to numeric values
+        try:
+            return int(year.rstrip("stndrdth"))  # Remove suffixes like "st", "nd", "rd", "th"
+        except ValueError:
+            return 0  # Default to 0 if year cannot be converted
+
+    goal_reached = []
+    
+    for result in predictions:
+        drug = result['Drug']
+        predictions_data = result['Predictions']
+        
+        for year, tscore in zip(predictions_data['Year'], predictions_data['Predicted T-score']):
+            if tscore >= target_tscore:
+                # Convert year to an integer using helper function
+                numeric_year = year_to_int(year)
+                goal_reached.append({'Medication': drug, 'Year': numeric_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 (T-score ≥ -2.4)")
+    
+    if not goal_summary:
+        st.info("No medications reach the target T-score.")
+    else:
+        summary_table = pd.DataFrame(goal_summary)
+        st.table(summary_table)
+
+# Medication Selection with Collapsible Categories
+def select_medications():
+    st.subheader("Select Medications to Display")
+    show_all = st.checkbox("Show All Medications", key="show_all")
+    
+    selected_medications = []
+    if not show_all:
+        # Define categories and medications
+        categories = {
+            "Bisphosphonates": [
+                "Alendronate", "Risedronate", "Ibandronate oral", 
+                "Zoledronate", "Ibandronate IV (3mg)"
+            ],
+            "RANK Ligand Inhibitors": [
+                "Denosumab", "Denosumab + Teriparatide"
+            ],
+            "Anabolic Agents": [
+                "Teriparatide", "Teriparatide + Denosumab"
+            ],
+            "Sclerostin Inhibitors": [
+                "Romosozumab", "Romosozumab + Denosumab", 
+                "Romosozumab + Alendronate", "Romosozumab + Ibandronate", 
+                "Romosozumab + Zoledronate"
+            ]
+        }
+        
+        # Create collapsible sections
+        for category, medications in categories.items():
+            with st.expander(category):
+                for med in medications:
+                    # Use a unique key for each checkbox
+                    if st.checkbox(med, key=f"{category}_{med}"):
+                        selected_medications.append(med)
+    else:
+        # Include all medications if "Show All" is selected
+        selected_medications = [
+            "Alendronate", "Risedronate", "Ibandronate oral", 
+            "Zoledronate", "Ibandronate IV (3mg)", "Denosumab", 
+            "Denosumab + Teriparatide", "Teriparatide", 
+            "Teriparatide + Denosumab", "Romosozumab", 
+            "Romosozumab + Denosumab", "Romosozumab + Alendronate", 
+            "Romosozumab + Ibandronate", "Romosozumab + Zoledronate"
+        ]
+    
+    return selected_medications
+
 # Streamlit UI
+# Main function
 def main():
     st.title("BMD and T-score Prediction Tool")
     
@@ -119,33 +202,8 @@ def main():
         format="%.3f"
     )
     
-    # Medication Selection with Checkboxes
-    st.subheader("Select Medications to Display")
-    
-    # Add "Show All" Option
-    show_all = st.checkbox("Show All Medications")
-    
-    # Define medications by rows
-    medication_rows = [
-        ["Alendronate", "Risedronate", "Ibandronate oral"],
-        ["Zoledronate", "Ibandronate IV (3mg)"],
-        ["Denosumab", "Denosumab + Teriparatide"],
-        ["Teriparatide", "Teriparatide + Denosumab"],
-        ["Romosozumab", "Romosozumab + Denosumab", "Romosozumab + Alendronate"],
-        ["Romosozumab + Ibandronate", "Romosozumab + Zoledronate"]
-    ]
-    
-    # Create checkboxes for each row
-    selected_medications = []
-    if not show_all:
-        for row in medication_rows:
-            cols = st.columns(len(row))
-            for col, med in zip(cols, row):
-                if col.checkbox(med):
-                    selected_medications.append(med)
-    else:
-        # If "Show All" is checked, include all medications
-        selected_medications = [med for row in medication_rows for med in row]
+    # Medication Selection
+    selected_medications = select_medications()  # Ensure this is only called once
     
     # Load constants and medication data
     medication_data = load_medication_data()
@@ -159,7 +217,13 @@ 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_tscore=-2.4)
+            display_goal_summary(goal_summary)
+            
+            # Display individual medication results
             display_results(filtered_predictions, selected_site)
 
+
 if __name__ == "__main__":
     main()