Update app.py

app.py

@@ -274,14 +274,23 @@ else:
 
         # get time-based scores
         time_scores = shelters.apply(lambda row: get_time_score(datetime.now(timezone), row), axis=1).tolist()
-        shelters['time_score'] = time_scores
-        scores.append('time_score')
-    
+        
+        if data['Urgency'] == "Today": 
+            for i in range(4):
+                shelters[f'time_score_{i}'] = time_scores
+                scores.append(f'time_score_{i}')
+        elif data['Urgency'] == "In the next few days":
+            shelters['time_score'] = time_scores
+            scores.append('time_score')
+        elif data['Urgency'] == "In a week or more":
+            pass
+            
         # calcualte cumulative score
         shelters['total_score'] = shelters[scores].sum(axis=1)
         shelters['total_score'] = shelters['total_score'] / len(scores)
     
         shelters = shelters.sort_values(by='total_score', ascending=True)
+        st.write(shelters)
         shelters = shelters.head(3)
 
         # convert pandas df into list of dicts