Update app.py

app.py

@@ -94,6 +94,9 @@ if 'form_submitted' not in st.session_state:
 if 'shelter_index' not in st.session_state:
     st.session_state.shelter_index = 0
 
+if 'shelters_filtered' not in st.session_state:
+    st.session_state.shelters_filtered = False
+
 # Page config
 st.set_page_config(
     page_title="ShelterSearch",
@@ -139,78 +142,81 @@ if not st.session_state.form_submitted:
             json.dump(data, f)
 
         st.session_state.form_submitted = True
-        st.session_state.data = data
         st.rerun()
 else:
-    with open('data.json', 'r') as f:
-        data = json.load(f)
-
-    shelters = pd.read_csv("database.csv")
+    if not st.session_state.shelters_filtered:
+        with open('data.json', 'r') as f:
+            data = json.load(f)
 
-    # filter city
-    shelters = shelters[(shelters['City'] == data['City'])]
+        shelters = pd.read_csv("database.csv")
     
-    # filter sex
-    shelters = shelters[(shelters['Sex'] == data['Sex']) | (shelters['Sex'] == 'All')]
-
-    # filter lgbtq
-    if data['LGBTQ'] == 'No':
-        shelters = shelters[(shelters['LGBTQ'] == "No")]
-
-    # filter domestic violence
-    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']
+        # filter city
+        shelters = shelters[(shelters['City'] == data['City'])]
         
-        shelters_coordinates = shelters.apply(lambda row: get_coordinates(row['Zip Code'], geocoding_api_key), axis=1).tolist()
-        user_coordinates = get_coordinates(data['Zip Code'], geocoding_api_key)
-    
-        distances = []
-        for coordinates in shelters_coordinates:
-             distances.append(haversine(coordinates[0], coordinates[1], user_coordinates[0], user_coordinates[1]))
+        # filter sex
+        shelters = shelters[(shelters['Sex'] == data['Sex']) | (shelters['Sex'] == 'All')]
     
-        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')
+        # filter lgbtq
+        if data['LGBTQ'] == 'No':
+            shelters = shelters[(shelters['LGBTQ'] == "No")]
 
-    # services
-    if data['Needs'] != "":     
-        OpenAI_API_KEY = os.environ["OPENAI_API_KEY"]
+        # filter domestic violence
+        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']
+            
+            shelters_coordinates = shelters.apply(lambda row: get_coordinates(row['Zip Code'], geocoding_api_key), axis=1).tolist()
+            user_coordinates = get_coordinates(data['Zip Code'], geocoding_api_key)
         
-        services_scores = shelters.apply(lambda row: call_gpt(data['Needs'], row['Services'], OpenAI_API_KEY), axis=1).tolist()
-        services_scores = [s / 10 for s in services_scores]
+            distances = []
+            for coordinates in shelters_coordinates:
+                 distances.append(haversine(coordinates[0], coordinates[1], user_coordinates[0], user_coordinates[1]))
         
-        shelters['services_score'] = services_scores
-        scores.append('services_score')
+            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')
+    
+        # get services scores
+        if data['Needs'] != "":     
+            OpenAI_API_KEY = os.environ["OPENAI_API_KEY"]
+            
+            services_scores = shelters.apply(lambda row: call_gpt(data['Needs'], row['Services'], OpenAI_API_KEY), axis=1).tolist()
+            services_scores = [s / 10 for s in services_scores]
+            
+            shelters['services_score'] = services_scores
+            scores.append('services_score')
+    
+        # 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)
+        shelters = shelters.head(3)
 
-    # calcualte cumulative score
-    shelters['total_score'] = shelters[scores].sum(axis=1)
-    shelters['total_score'] = shelters['total_score'] / len(scores)
+        # convert pandas df into list of dicts
+        shelters = shelters.to_dict(orient='records')
 
-    shelters = shelters.sort_values(by='total_score', ascending=True)
-    shelters = shelters.head(3)
-    
-    # convert pandas df into list of dicts
-    shelters = shelters.to_dict(orient='records')
+        st.session_state.shelters_filtered = True
+        st.session_state.shelters = shelters
 
     # Display the current shelter information
-    shelter = shelters[st.session_state.shelter_index]
+    shelter = st.session_state.shelters[st.session_state.shelter_index]
     
     st.header(f"{shelter['Organization Name']}: {shelter['Program Name']}")
     st.divider()
@@ -261,4 +267,5 @@ else:
         if st.button("Reset"):
             st.session_state.shelter_index = 0
             st.session_state.form_submitted = False
+            st.session_state.shelters_filtered = False
             st.experimental_rerun()