import os
import gradio as gr

# Ejemplos imágenes región de Antofagasta
os.system("wget -O cordillera_806.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/cordillera_806.png")
os.system("wget -O cordillera_943.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/cordillera_943.png")
os.system("wget -O glaciar_657.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/glaciar_657.png")
os.system("wget -O glaciar_719.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/glaciar_719.png")

# Ejemplos imágenes de la zona del Huasco
os.system("wget -O demo_c_0.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_c_0.png")
os.system("wget -O demo_c_1.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_c_1.png")
os.system("wget -O demo_c_2.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_c_2.png")
os.system("wget -O demo_c_3.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_c_3.png")
os.system("wget -O demo_g_0.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_g_0.png")
os.system("wget -O demo_g_1.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_g_1.png")
os.system("wget -O demo_g_2.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_g_2.png")
os.system("wget -O demo_g_3.png https://huggingface.co/spaces/alkzar90/rock-glacier-detector/resolve/main/example_images/demo_g_3.png")


iface = gr.Interface.load(name='alkzar90/skynet', 
                          src='models', 
                          api_key='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',
                          title='Skynet: detector de glaciares de roca 🛰️',
                          description='Demo para ilustrar inferencia sobre imágenes satelitales para discriminar glaciares de roca.',
                          examples=[['cordillera_806.png'], ['cordillera_943.png'], ['glaciar_657.png'], ['glaciar_719.png'],
                                    ['demo_c_0.png'], ['demo_c_1.png'], ['demo_c_2.png'], ['demo_c_3.png'],
                                    ['demo_g_0.png'], ['demo_g_1.png'], ['demo_g_2.png'], ['demo_g_3.png']],
                          cache_examples=False)


if __name__ == "__main__":
    iface.launch()