from pyzipcode import ZipCodeDatabase

import folium

import pandas as pd

clin_addresses = pd.read_excel("/Users/shania/PycharmProjects/ClinicalAttritionRateMap/project_testCleanedUSZip.xlsx")

zcdb = ZipCodeDatabase()

center_map = folium.Map(location=[39.8283, -98.5795], zoom_start=4)

for index in clin_addresses.index:

    zip_code = str(clin_addresses.loc[index, 'zip'])

    # Remove the extended part from the ZIP code if it exists

    if '-' in zip_code:

        zip_code = zip_code.split('-')[0]

    location_name = clin_addresses.loc[index, 'location_name']

    nct_id = clin_addresses.loc[index, 'nct_id']

    dropout_rate = clin_addresses.loc[index, 'dropout_percentage_all']

    phase = clin_addresses.loc[index, 'phase']

    study_type = clin_addresses.loc[index, 'study_type']

    try:

        location = zcdb[zip_code]

        lon, lat = location.longitude, location.latitude

        coordinates = [lat, lon]

        folium.Marker(

            location=coordinates,

            popup=f"{location_name}\nNCT ID: {nct_id}\nDropout Rate: {dropout_rate}\nPhase: {phase}\nStudy Type: {study_type}"

        ).add_to(center_map)

    except KeyError:

        print(f"Couldn't find location for ZIP code: {zip_code}")

# Save the map to an HTML file

center_map.save('map-clinical_trials.html')