#!/usr/bin/env python
# coding: utf-8

# # <img style='float: left' src="http://lightning-viz.github.io/images/logo.png"> <br> <br> &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;Image polygon plots in <a href='http://lightning-viz.github.io/'><font color='#9175f0'>Lightning</font></a>

# ## <hr> Setup

# In[1]:


from lightning import Lightning

from sklearn import datasets


# ## Connect to server

# In[2]:


lgn = Lightning(ipython=True, host='http://public.lightning-viz.org')


# ## <hr> Region drawing

# The `image-poly` visualization let's you draw polygonal regions on images and then query them in the same notebook!
# <br>
# Try drawing a region on the image. Hold command to pan, and option to edit the region. 
# <br>
# Note that we assign the visualization to an output variable, so that we can query it later, but we must print that output to get the image to show.

# In[3]:


imgs = datasets.load_sample_images().images
viz = lgn.imagepoly(imgs[0])
viz


# Draw some regions on the image above. Then check the value of `viz.coords`.

# In[4]:


p = viz.polygons()


# In[5]:


lgn.imagepoly(imgs[0], polygons=p)


# In this case we drew four regions, so we get four arrays of coordinates (in original image space).
# <br>
# The region overlays themselves do not persist on the image after a refresh, so you'll have to draw your own to see them.