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()