Update app.py

app.py

@@ -93,6 +93,13 @@ def display_results(predictions, site):
 
 # 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:
@@ -100,8 +107,10 @@ def generate_goal_summary(predictions, target_tscore=-2.4):
         predictions_data = result['Predictions']
         
         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)})
+            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