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

#Autor: Erickson Silva 
#Email: <erickson.silva@lavid.ufpb.br> <ericksonsilva@live.com>

#LAViD - Laboratório de Aplicações de Vídeo Digital

import os
import csv
import sys
from nltk.tree import Tree
from LerDicionarios import *

class AplicaSinonimos(object):
	"""Classe responsável por aplicar os sinonimos após aplicação de regras morfológicas/sintáticas"""

	def __init__(self):
		self.dicionarios = LerDicionarios()

	# Itera sobre os tokens obtendo os sinonimos
	def aplicar_sinonimos(self, lista_anotada):
		lista_corrigida = []
		for tupla in lista_anotada:
			sinonimo = self.verificar_sinonimo(tupla[0])
			lista_corrigida.append(sinonimo)
		try:
			return " ".join(lista_corrigida)
		except:
			return " ".join([str(x[0]) for x in lista_anotada])

	# Verifica se há sinonimo do token  
	def verificar_sinonimo(self, token):
		if self.dicionarios.has_sinonimo(token):  
			return self.dicionarios.get_sinonimo(token)
		return token