Update app.py

app.py

@@ -161,6 +161,9 @@ else:
     if data['Domestic Violence'] == "No":
         shelters = shelters[(shelters['Domestic Violence'] == "No")]
 
+    # keep track of which scores are calculated
+    scores = []
+    
     # calculate distances between zipcodes
     if data['Zip Code'] != "Unsure":
         geocoding_api_key = os.environ['OpenWeather_API_KEY']
@@ -174,17 +177,18 @@ else:
     
         max = max(distances) if (max(distances) != 0) else 1
         shelters['zipcode_score'] = [d / max for d in distances]
+        scores.append('zipcode_score')
 
     # get urgency scores 
     urgency_scores = shelters.apply(lambda row: get_urgency_score(data['Urgency'], row['Urgency']), axis=1).tolist()
     shelters['urgency_score'] = urgency_scores
+    scores.append('urgency_score')
 
     # get duration scores
     duration_scores = shelters.apply(lambda row: get_duration_score(data['Duration'], row['Duration']), axis=1).tolist()
     shelters['duration_score'] = duration_scores
+    scores.append('duration_score')
 
-    
-    
     # services
     if data['Needs'] != "":     
         OpenAI_API_KEY = os.environ["OPENAI_API_KEY"]
@@ -193,8 +197,16 @@ else:
         services_scores = [s / 10 for s in services_scores]
         
         shelters['services_score'] = services_scores
+        scores.append('services_score')
 
-    st.table(shelters)
+    # calcualte cumulative score
+    total_scores = pd.Series()
+    for s in scores:
+        total_scores += s
+    total_scores = [s/4 for s in total_scores]
+    shelters['total_score'] = total_scores
+
+    
         
     shelters = [
         {"title": "Shelter 1", "description": "This is the 1st shelter",},