Update app.py

app.py

@@ -1,6 +1,51 @@
 import streamlit as st
 import json
 import pandas as pd
+import requests
+
+def get_coordinates(address: str, api_key: str) -> list:
+    """
+    Get the coordinates (latitude and longitude) of an address using the OpenCage Data Geocoding API.
+
+    Parameters:
+    address (str): The address to geocode.
+    api_key (str): Your OpenCage Data API key.
+
+    Returns:
+    list: A list containing the latitude and longitude of the address.
+    """
+    # Encode the address for the URL
+    encoded_address = requests.utils.quote(address)
+    
+    # Construct the API URL
+    url = f"https://api.opencagedata.com/geocode/v1/json?q={encoded_address}&key={api_key}"
+    
+    # Make the API request
+    response = requests.get(url)
+    
+    if response.status_code == 200:
+        # Parse the JSON response
+        data = response.json()
+        
+        # Check if there are any results
+        if data['results']:
+            # Extract the latitude and longitude from the first result
+            latitude = data['results'][0]['geometry']['lat']
+            longitude = data['results'][0]['geometry']['lng']
+            return [latitude, longitude]
+        else:
+            raise ValueError("No results found for the given address.")
+    else:
+        raise Exception(f"API request failed with status code {response.status_code}")
+
+def haversine(lat1, lon1, lat2, lon2):
+    R = 6371  # Earth radius in kilometers. Use 3956 for miles.
+    dlat = math.radians(lat2 - lat1)
+    dlon = math.radians(lon2 - lon1)
+    a = math.sin(dlat / 2) ** 2 + math.cos(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.sin(dlon / 2) ** 2
+    c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))
+    distance = R * c
+    return distance
 
 def validate_zipcode(zipcode):
     try:
@@ -90,8 +135,26 @@ else:
     # filter domestic violence
     if data['Domestic Violence'] == 'No':
         shelters = shelters[(shelters['Domestic Violence'] == "No")]
+
+    # calculate distances between zipcodes
+    api_key = os.environ['OpenCage_API_KEY']
+    
+    shelters_coordinates = shelters.apply(lambda row: get_coordinates(row['Zip Code'], api_key), axis=1).tolist()
+    user_coordinates = get_coordinates(data['Zip Code'])
+
+    distances = []
+    for coordinates in shelters_coordinates:
+        distances.append(haversine(coordinates[0], coordinates[1], user_coordinates[0], user_coordinates[1]))
+
+    shelters['zipcode_score'] = distances
+
+    # urgency
+
+    # duration
+
+    # services
+        
     st.table(shelters)
-    print(type(shelters))
 
     # Display the current shelter information
     shelter = shelters_def[st.session_state.shelter_index]
@@ -111,4 +174,4 @@ else:
         if st.button("Next"):
             if st.session_state.shelter_index < len(shelters_def) - 1:
                 st.session_state.shelter_index += 1
-                st.experimental_rerun()
+                st.experimental_rerun()