from ipymarkup.span import Span
from ipymarkup.dep import Dep
from ipymarkup.demo import show_table

from ipymarkup import (
    show_span_box_markup,
    show_span_ascii_markup,
    show_dep_markup
)
from ipymarkup.palette import palette, BLUE, ORANGE, PURPLE

show_table()

text = '<i>abc</i>'
spans = [Span(0, 3)]
show_span_box_markup(text, spans)

# to hide u' in python2
words = ['<i>abc</i>', 'cde']
spans = [Dep(0, 1, '<b>1</b>')]
show_dep_markup(words, spans)

text = r'''\begin{aligned}
  \nabla \cdot \vec{\mathbf{B}} = 0
\end{aligned}'''
spans = []
show_span_box_markup(text, spans)

from ipymarkup.span import Span
from ipymarkup.dep import Dep
from ipymarkup.demo import show_table

from ipymarkup import (
    show_span_box_markup,
    show_span_ascii_markup,
    show_dep_markup
)
from ipymarkup.palette import palette, BLUE, ORANGE, PURPLE

text = 'a d a b a a a b c c c f d'
spans = [
    Span(0, 13, 'a'),
    Span(2, 25, 'd'),
    Span(6, 15, 'b'),
    Span(16, 21, 'c'),
    Span(22, 23, 'f'),
]

show_span_ascii_markup(text, spans)

show_span_ascii_markup(text, spans, width=8)

text = '01234567890123456789'
spans = [
    Span(6, 15, 'abcdef'),
    Span(0, 3, 'abcdef'),
    Span(1, 9, 'abcdef')
]
show_span_ascii_markup(text, spans, width=10)

text = '123'
spans = []
show_span_ascii_markup(text, spans)

text = '''\t1
34'''
spans = [Span(1, 5)]
show_span_ascii_markup(text, spans)

text = '''0123456'''
spans = [
    Span(0, 5),
    Span(0, 3),
    Span(0, 3, 'a'),
    Span(0, 3, 'c')
]
show_span_ascii_markup(text, spans)

class C(object):
    def __init__(self, start, stop, type):
        self.start = start
        self.stop = stop
        self.type = type

        
spans = [
    (0, 5),
    [0, 3],
    (0, 3, 'a'),
    C(0, 3, 'c'),
]
show_span_ascii_markup(text, spans)

text = '1234567890'
spans = [(0, 2), (3, 6, 'PER'), (7, 9, 'LOC-a')]
show_span_box_markup(text, spans, palette=palette(BLUE, {None: BLUE, 'LOC-a': ORANGE}, PER=BLUE))

show_span_box_markup(text, spans, palette=palette('blue', PURPLE, {None: 'blue', 'PER': 'purple'}))