Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
| import gradio as gr | |
| from whereabouts.utils import download | |
| from whereabouts.Matcher import Matcher | |
| # download the various address databases | |
| download('au_all_sm', 'saunteringcat/whereabouts-db') | |
| download('au_all_lg', 'saunteringcat/whereabouts-db') | |
| download('au_vic_lg', 'saunteringcat/whereabouts-db') | |
| download('au_nsw_lg', 'saunteringcat/whereabouts-db') | |
| download('us_ca_sm', 'saunteringcat/whereabouts-db') | |
| download('us_ma_sm', 'saunteringcat/whereabouts-db') | |
| # create a matcher object | |
| matcher1 = Matcher('au_all_sm') | |
| matcher2 = Matcher('au_all_lg') | |
| matcher3 = Matcher('au_vic_lg') | |
| matcher4 = Matcher('au_nsw_lg') | |
| matcher5 = Matcher('us_ca_sm') | |
| matcher6 = Matcher('us_ma_sm') | |
| default_address_values = """ | |
| 3333 Channel Way, San Diego, CA | |
| 1500 Orange Avenuee, Colonado, CA | |
| 3129 Arden Wy, Sacramento, 95825 | |
| 2000 Allston Way, Berkly, 94704 | |
| """ | |
| # function to geocode results | |
| def geocode(addresses, db_name='au_all_sm', how='standard'): | |
| if db_name == 'au_all_sm': | |
| matcher = matcher1 | |
| elif db_name == 'au_all_lg': | |
| matcher = matcher2 | |
| elif db_name == 'au_vic_lg': | |
| matcher = matcher3 | |
| elif db_name == 'au_nsw_lg': | |
| matcher = matcher4 | |
| elif db_name == 'us_ca_sm': | |
| matcher = matcher5 | |
| elif db_name == 'us_ma_sm': | |
| matcher = matcher6 | |
| address_list = addresses.split('\n') | |
| return matcher.geocode(address_list, how=how) | |
| # the gradio interface | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown("# whereabouts") | |
| gr.HTML(""" | |
| <div style="display:flex;column-gap:4px;"> | |
| <a href='https://github.com/ajl2718/whereabouts'> | |
| <img src='https://img.shields.io/badge/Code-github-blue'> | |
| </a> | |
| </div> | |
| """) | |
| gr.Markdown("Fast, accurate open source geocoding in Python") | |
| gr.Markdown(""" | |
| [whereabouts](https://www.github.com/ajl2718/whereabouts) is an open source package for Python for fast and accurate geocoding within | |
| your own environment. It currently supports geocoding for all of Australia with some US states currently being built using OpenAddresses. | |
| It uses duckDB to geocode 100s - 1000s of addresses per second based on an algorithm from [this paper](https://arxiv.org/abs/1708.01402)""") | |
| gr.Markdown(""" | |
| ``` | |
| pip install whereabouts | |
| python -m whereabouts download us_ca_sm | |
| ``` | |
| This demo shows whereabouts with some example databases. | |
| """) | |
| with gr.Row(): | |
| with gr.Column(): | |
| dropdown_choice = gr.Dropdown(choices=[("Australia - Small", "au_all_sm"), | |
| ("Australia - Large", "au_all_lg"), | |
| ("Victoria, Australia - Large", "au_vic_lg"), | |
| ("New South Wales, Australia - Large", "au_nsw_lg"), | |
| ("California - Small", "us_ca_sm"), | |
| ("Massachusetts - Small", "us_ma_sm")], | |
| value="us_ca_sm", | |
| multiselect=False, | |
| label="Database", | |
| info="Select from one of the geocoding databases based on country, region and size") | |
| radio_choice = gr.Radio(choices=["standard", "trigram"], | |
| value='standard', | |
| label="Matching algorithm", | |
| info="Trigram matching is more accurate but slower. Only available for the large databases.") | |
| text_input = gr.Textbox(lines=2, label="Addresses to geocode (one row per address)", value=default_address_values) | |
| geocode_button = gr.Button(variant='primary') | |
| json_output = gr.JSON(label="Output JSON data") | |
| geocode_button.click(fn=geocode, inputs=[text_input, dropdown_choice, radio_choice], outputs=json_output, api_name="whereabouts_geocoder") | |
| demo.launch() |