class SummaryWriter(object):
  ''' A wrapper around the Tensorflow Summary writer.
  '''
  tf = __import__('tensorflow')

  def __init__(self, logdir, flush_secs=20):
    self.tf_writer = self.tf.train.SummaryWriter(logdir, flush_secs=flush_secs)

  def add_summary(self, step, **kwargs):
    summary_values = [self.tf.Summary.Value(tag=tag, simple_value=value)
                      for (tag, value) in kwargs.iteritems()]
    self.tf_writer.add_summary(self.tf.Summary(value=summary_values), step)


if __name__ == '__main__':
  from neural_srl.shared.constants import ROOT_DIR
  from os.path import join
  writer = SummaryWriter(join(ROOT_DIR, 'temp'))
  writer.add_summary(1, v1=0.5, v2=2)
  writer.add_summary(2, v1=0.6, v2=4)