# based on https://huggingface.co/spaces/NimaBoscarino/climategan/blob/main/app.py # noqa: E501
# thank you @NimaBoscarino

import os
import gradio as gr
import googlemaps
from skimage import io
from urllib import parse
from inferences import ClimateGAN


def predict(api_key):
    def _predict(*args):
        print("args: ", args)
        image = place = None
        if len(args) == 1:
            image = args[0]
        else:
            assert len(args) == 2, "Unknown number of inputs {}".format(len(args))
            image, place = args

        if api_key and place:
            geocode_result = gmaps.geocode(place)

            address = geocode_result[0]["formatted_address"]
            static_map_url = f"https://maps.googleapis.com/maps/api/streetview?size=640x640&location={parse.quote(address)}&source=outdoor&key={api_key}"
            img_np = io.imread(static_map_url)
        else:
            img_np = image
        flood, wildfire, smog = model.inference(img_np)
        return img_np, flood, wildfire, smog

    return _predict


if __name__ == "__main__":

    api_key = os.environ.get("GMAPS_API_KEY")
    gmaps = None
    if api_key is not None:
        gmaps = googlemaps.Client(key=api_key)

    model = ClimateGAN(model_path="config/model/masker")

    inputs = inputs = [gr.inputs.Image(label="Input Image")]
    if api_key:
        inputs += [gr.inputs.Textbox(label="Address or place name")]

    gr.Interface(
        predict(api_key),
        inputs=[
            gr.inputs.Textbox(label="Address or place name"),
            gr.inputs.Image(label="Input Image"),
        ],
        outputs=[
            gr.outputs.Image(type="numpy", label="Original image"),
            gr.outputs.Image(type="numpy", label="Flooding"),
            gr.outputs.Image(type="numpy", label="Wildfire"),
            gr.outputs.Image(type="numpy", label="Smog"),
        ],
        title="ClimateGAN: Visualize Climate Change",
        description='Climate change does not impact everyone equally. This Space shows the effects of the climate emergency, "one address at a time". Visit the original experience at <a href="https://thisclimatedoesnotexist.com/">ThisClimateDoesNotExist.com</a>.<br>Enter an address or place name, and ClimateGAN will generate images showing how the location could be impacted by flooding, wildfires, or smog.',  # noqa: E501
        article="<p style='text-align: center'>This project is an unofficial clone of <a href='https://thisclimatedoesnotexist.com/'>ThisClimateDoesNotExist</a> | <a href='https://github.com/cc-ai/climategan'>ClimateGAN GitHub Repo</a></p>",  # noqa: E501
        # examples=[
        #     "Vancouver Art Gallery",
        #     "Chicago Bean",
        #     "Duomo Siracusa",
        # ],
        css=".footer{display:none !important}",
    ).launch()