Update app.py

app.py

@@ -5,6 +5,42 @@ import requests
 import os
 import math
 
+def get_urgency_score(user, shelter):
+    if user == "Today": 
+        if shelter == "Immdiaite": return 0
+        if shelter == "High": return 0.75
+        if shelter == "Moderate": return 1
+    elif user == "In the next few days":
+        if shelter == "Immdiaite": return 0.25
+        if shelter == "High": return 0
+        if shelter == "Moderate": return 0.75
+    elif user == "In a week or so":
+        if shelter == "Immdiaite": return 0.75
+        if shelter == "High": return 0.25
+        if shelter == "Moderate": return 0
+
+def get_duration_score(user, shelter):
+    if user == "Overnight":
+        if shelter == "Overnight": return 0
+        if shelter == "Temporary": return 0.5
+        if shelter == "Transitional": return 0.75
+        if shelter == "Long-Term": return 1
+    elif user == "A month or less":
+        if shelter == "Overnight": return 0.5
+        if shelter == "Temporary": return 0
+        if shelter == "Transitional": return 0.25
+        if shelter == "Long-Term": return 0.75
+    elif user == "A couple of months":
+        if shelter == "Overnight": return 0.75
+        if shelter == "Temporary": return 0.25
+        if shelter == "Transitional": return 0
+        if shelter == "Long-Term": return 0.5
+    elif user == "A year or more":
+        if shelter == "Overnight": return 1
+        if shelter == "Temporary": return 0.75
+        if shelter == "Transitional": return 0.5
+        if shelter == "Long-Term": return 0
+
 def get_zip_codes(city, state):
     url = f'http://api.zippopotam.us/us/{state}/{city}'
     response = requests.get(url)
@@ -53,7 +89,7 @@ def form_dialog():
     domestic_violence = st.radio("Have you experienced domestic violence (some shelters serve these individuals specifically", ["No", "Yes"])
     
     urgency = st.radio("How quickly do you need help?", ("Today", "In the next few days", "In a week or more"))
-    duration = st.radio("How long do you need a place to stay?", ("Overnight", "A month of less", "A couple of months", "A year or more"))
+    duration = st.radio("How long do you need a place to stay?", ("Overnight", "A month or less", "A couple of months", "A year or more"))
     needs = st.text_area("Optional - Needs (tell us what you need and how we can help)", value="")
 
     if st.button("Submit"):
@@ -125,53 +161,24 @@ else:
          distances.append(haversine(coordinates[0], coordinates[1], user_coordinates[0], user_coordinates[1]))
 
     max = max(distances) if (max(distances) != 0) else 1
-    print(max)
     shelters['zipcode_score'] = [d / max for d in distances]
 
-    # urgency
-
-    # Immidiate
-    # --> immidiate: 0
-    # --> high: 0.75
-    # --> moderate: 1
+    # get urgency scores 
+    urgency_scores = []
+    for shelter in shelters:
+        urgency_scores.append(get_urgency_score(data['Urgency'], shelter['Urgency']))
 
-    # High
-    # --> immidiate: 0.25
-    # --> high: 0
-    # --> moderate = 0.75
-
-    # Moderate
-    # --> immidiate: 0.75
-    # --> high: 0.25
-    # --> moderate: 0
+    shelters['urgency_score'] = urgency_scores
 
     # duration
+    duration_scores = []
+    for shelter in shelters:
+        duration_scores.append(get_duration_score(data['Duration'], shelter['Duration']))
 
-    # Overnight
-    # --> overnight: 0
-    # --> temp: 0.5
-    # --> trans: 0.75
-    # --> long: 1
-
-    # Temp
-    # --> overnight: 0.25
-    # --> temp: 0
-    # --> trans: 0.25
-    # --> long: 0.75
-
-    # Trans
-    # --> overnight: 0.75
-    # --> temp: 0.25
-    # --> trans: 0
-    # --> long: 0.25
-
-    # Long
-    # --> overnight: 1
-    # --> temp: 0.75
-    # --> trans: 0.5
-    # --> long: 0
+    shelters['duration_score'] = urgency_scores
 
     # services
+    
         
     st.table(shelters)