interface.py

#Bibliotecas
import tkinter as tk
from PIL import ImageTk, Image
from serial import Serial
import cv2 as cv
from requests import get, post
from datetime import datetime, timedelta
import csv
from unidecode import unidecode
import json
from threading import Thread
from time import sleep
from utilitarios.dicionario import traduzir, escrever_braile
from reconhecimento.detector import tirar_foto, Detector_aws

#Variáveis globais
global imagem, nome_arquivo_foto
global painel_foto_interno, estado_atual_palavra, etiqueta_palavra
global braile, palavra_traduzida
global meu_serial
global dicionario_palavras

#Inicializa variáveis globais
dicionario_palavras = {'palavras':[]}

#Funções auxiliares
    #Braile
        #melhora função de desenhar circulo
def _create_circle(self, x, y, r, **kwargs):
    return self.create_oval(x-r, y-r, x+r, y+r, **kwargs)
tk.Canvas.create_circle = _create_circle

    #Comunicação com o arduino
        #trata palavra traduzida e braile pro arduino ex:'bala','100100,200220,300300,400400'
def formata_palavra_braile(palavra, braile_palavra):
    formatacao = "'{}','".format(palavra)
    for lista_digito in braile_palavra:
        for lista_coluna in lista_digito:
            for bola in lista_coluna:
                formatacao += str(bola)
                
        formatacao += ","
    
    formatacao = formatacao[:-1]
    formatacao += "'"
    
    print("=palavra="+formatacao)
    
    return formatacao

def monitorar_serial():
    while True:
        if meu_serial != None:
            texto_recebido = meu_serial.readline().decode().strip()
            if texto_recebido != "":
                print("Retorno Serial:", texto_recebido)
    
        sleep(0.1)

#Inicia serial
meu_serial = Serial("COM22", baudrate = 115200, timeout = 0.1)

thread = Thread(target = monitorar_serial)
thread.daemon = True
thread.start()

#Inicia vídeo
stream = cv.VideoCapture(0)
if (stream.isOpened() == False):
  print("Câmera não encontrada")

#Chave amozon
with open('reconhecimento/ChavesAcesso.csv','r') as credenciais:
    next(credenciais)
    chave,senha = list(csv.reader(credenciais))[0]

#Abre o arquivo JSON
try:
    with open('utilitarios/palavras.json', 'r') as arquivo_json:
    #Salva JSON no dicionário
        dicionario_palavras = json.load(arquivo_json)
            
except IOError:
    print("Arquivo JSON não encontrado")

#Criação da janela
janela = tk.Tk()
janela.title("Reconhecedor Braile")
janela.geometry("908x376")

    #Icone
icone = ImageTk.PhotoImage(Image.open('Fotos/icone/Icone.png'))
janela.wm_iconphoto(False, icone)

#Janelas
    #Foto
painel_foto_externo = tk.Label(janela, background = "black", width = 57, height = 17)
painel_foto_externo.place(x = 100, y = 17)
painel_foto_interno = tk.Label(janela, width = 54, height = 14)
painel_foto_interno.place(x = 103, y = 20)

    #Estato
janela_estato_externo = tk.Canvas(janela, background = "black", width = 131, height = 160)
janela_estato_externo.place(x = 508, y = 116)
janela_estato_interno = tk.Canvas(janela, background = "grey93", width = 121, height = 150)
janela_estato_interno.place(x = 513, y = 121)

janela_estato_interno.create_text(55, 45, text = "Palavra:", font = 1)
estado_atual_palavra = tk.Label(janela, background = "grey93", text = "Fotografe!", font = 1)
estado_atual_palavra.place(x = 535, y = 123)

etiqueta_palavra = tk.Label(janela, text = "", bg = 'grey80', width = 16)
etiqueta_palavra.place(x = 516, y = 178)

    #Botões
janela_botoes_interno = tk.Canvas(janela, background = "black", width = 131, height = 95)
janela_botoes_interno.place(x = 508, y = 15)
janela_botoes_externo = tk.Canvas(janela, background = "grey93", width = 121, height = 85)
janela_botoes_externo.place(x = 513, y = 20)
        #Listbox
janela_botoes_interno = tk.Canvas(janela, background = "black", width = 163, height = 44)
janela_botoes_interno.place(x = 643, y = 232)
janela_botoes_externo = tk.Canvas(janela, background = "grey93", width = 153, height = 34)
janela_botoes_externo.place(x = 648, y = 237)

    #Braile
janela_braile_externo = tk.Canvas(janela, background = "black", width = 885, height = 76)
janela_braile_externo.place(x = 10, y = 282)
janela_braile_interno = tk.Canvas(janela, background = "grey93", width = 875, height = 66)
janela_braile_interno.place(x = 15, y = 287)

    #Listbox
lb_palavras = tk.Listbox(janela, bd = 4, width = 26, height = 13, justify = tk.CENTER)


for palavra in dicionario_palavras['palavras']:
    lb_palavras.insert(tk.END, palavra)
    
lb_palavras.place(x = 644, y = 16)

#Botões
    #Funções
        #Fotografar
def fotografar():
    global imagem, nome_arquivo_foto, etiqueta_palavra
    
            #tira foto
    imagem = Image.fromarray(video_tratado) 
    nome_arquivo_foto = datetime.now().strftime('Foto_teste_%d_%m_%y__%H_%M_%S.jpg')    
    imagem.save("Fotos/"+nome_arquivo_foto)
    
    print("Tirei foto!")

            #ativa botão de upload
    botao_upload['state'] = tk.NORMAL
    
            #altera palavra estado atual
    estado_atual_palavra.config(text = "Reconheça!")
    estado_atual_palavra.place(x = 532, y = 123)
    
    etiqueta_palavra.config(text = "")
    
    desenha_braile()

        #Upload
def envia_foto():
    global meu_serial, nome_arquivo_foto, braile, palavra_traduzida, estado_atual_palavra, etiqueta_palavra
    print("Enviando para a IA")
    
            #enviar foto para o identificador da amazon
    detector_amazon = Detector_aws(senha = senha, chave = chave)
    with open("Fotos/"+nome_arquivo_foto,'rb') as imagem_fonte:
        bytes_fonte = imagem_fonte.read()
    
            #recebe itens detectados
    itens_detectados = detector_amazon.receber_dados(bytes_imagem = bytes_fonte, etiquetas = 8).get('Labels')

            #separa palavra com maior confiança
    palavra_maior_confianca = itens_detectados[0].get('Name')
    
            #traduz palavra
                #en -> pt-br
    palavra_traduzida = traduzir(texto = palavra_maior_confianca)
    
                #pt-br -> braile
    braile = escrever_braile(unidecode(palavra_traduzida))
    
            #exibe palavra traduzida
    etiqueta_palavra = tk.Label(janela, text = palavra_traduzida, bg = 'grey80', width = 16, justify = tk.CENTER)
    etiqueta_palavra.place(x = 515, y = 178)
            
            #envia para o arduino
    palavra_arduino = "foto,\n" + formata_palavra_braile(unidecode(palavra_traduzida), braile)
    meu_serial.write(palavra_arduino.encode("UTF-8"))
    
            #exibe brile
    preeche_braile()
    
            #altera estado atual
    estado_atual_palavra.config(text = "Fotografe!")
    estado_atual_palavra.place(x = 535, y = 123)
    
            #desativa botão de upload
    botao_upload['state'] = tk.DISABLED

        #Adiciona palavra na lista
def adiciona_palavra_lista():
    global dicionario_palavras

            #pega palavara
    palavra_adicionada = entrada_nova_palavra.get()
    
            #adiciona na interface
    lb_palavras.insert(tk.END, palavra_adicionada)
            
            #limpa janela de entrada
    entrada_nova_palavra.delete(0,tk.END)
    
            #adiciona na lista
    dicionario_palavras['palavras'].append(unidecode(str(palavra_adicionada)))
    print(dicionario_palavras)
            
            #adiciona no JSON
    with open('utilitarios/palavras.json', 'w') as arquivo_json:
        json.dump(dicionario_palavras, arquivo_json)
    
        #Traduzir palavra da lista
def traduzir_palavra_lista():
    global braile, palavra_traduzida
    
            #pega palavra selecionada
    palavra_selecionada = lb_palavras.get(tk.ACTIVE)
    
            #traduz pra braile
    braile = escrever_braile(unidecode(str(palavra_selecionada)))
    
            #desenha braile
    palavra_traduzida = str(palavra_selecionada)
    preeche_braile()
    
        #Deletar palavra da lista
def deletar_palavra_lista():
    global dicionario_palavras, arquivo_json
            
            #pega palavara
    palavra_selecionada = lb_palavras.get(tk.ACTIVE)
    
            #deleta palavra selecionada
    lb_palavras.delete(tk.ANCHOR)
    
            #retira da lista
    indice = dicionario_palavras['palavras'].index(palavra_selecionada)
    dicionario_palavras['palavras'].pop(indice)
    print("==removido=="+str(dicionario_palavras))
    
            #retira do JSON
    with open('utilitarios/palavras.json', 'w') as arquivo_json:
        json.dump(dicionario_palavras, arquivo_json)

        #Envia lista de palavras para salvar no arduino
def envia_lista_arduino():
    global meu_serial, dicionario_palavras
            #cria identificador de lista
    lista_arduino = "lista," + str(len(dicionario_palavras['palavras'])) + "\n"
    meu_serial.write(lista_arduino.encode("UTF-8"))
            #envia para o arduino
    for palavra in dicionario_palavras['palavras']:
        lista_arduino = formata_palavra_braile(palavra, escrever_braile(unidecode(palavra))) + "\n"
        meu_serial.write(lista_arduino.encode("UTF-8"))
        print(lista_arduino)
        sleep(2.0)

        #Envia sinal para zerar posição dos servos
def reset_config_servo():
#     calibrar = "calibrate\n"
    meu_serial.write("calibrate\n".encode("UTF-8"))
    
        #Envia sinal para zerar a eprom
def reset_eprom():
#     limpar = "limpa\n"
    meu_serial.write("limpa\n".encode("UTF-8"))
    
    #Posicionar botões
        #Fotografar
botao_fotografar = tk.Button(janela, text = "Fotografar",font = ('arial',8), command = fotografar)
botao_fotografar.place(x = 546, y = 24)

        #Reconhecer
botao_upload = tk.Button(janela, text = "Reconhecer", command = envia_foto, state = 'disabled')
botao_upload.place(x = 539, y = 51)

        #Adicionar
botao_adicionar = tk.Button(janela, text = "Adicionar", command = adiciona_palavra_lista)
botao_adicionar.place(x = 545, y = 240)

            #Entrada de texto
entrada_nova_palavra = tk.Entry(janela, width = 20)
entrada_nova_palavra.place(x = 514, y = 215)

        #Traduzir
botao_traduzir = tk.Button(janela, text = "Traduzir", command = traduzir_palavra_lista)
botao_traduzir.place(x = 650, y = 243)

        #Deletar
botao_traduzir = tk.Button(janela, text = "Deletar", command = deletar_palavra_lista)
botao_traduzir.place(x = 708, y = 243)

        #Salvar
botao_salvar = tk.Button(janela, text = "Salvar", command = envia_lista_arduino)
botao_salvar.place(x = 761, y = 243)

        #Recalibrar
botao_recalibrar = tk.Button(janela, text = "Recalibrar", command = reset_config_servo)
botao_recalibrar.place(x = 515, y = 80)

        #Zerar eprom
botao_zerar = tk.Button(janela, text = "0EPROM", command = reset_eprom)
botao_zerar.place(x = 580, y = 80)

#Braile
    #Desenho da estrutura do braile
    #//fill = white -> 0/false, fill = black -> 1/true
def desenha_braile():
        #variáveis de desenho
    x_coluna1 = 14
    x_coluna2 = x_coluna1 + 15
    y1 = 14
    y2 = y1 + 15
    y3 = y2 + 15
    distancia_digito = 44
    
        #variável do loop 
    digito = 0
    
    #"apaga" digitos anteriores
    janela_braile_interno.create_rectangle(18, 54, 18+845, 54+12, fill = "grey93", outline = "grey93")
    
    while digito < 20:
            #digito n
                #coluna 1
        janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y1, 4, fill = "grey93", outline = "grey", width = 1)
        janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y2, 4, fill = "grey93", outline = "grey", width = 1)
        janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y3, 4, fill = "grey93", outline = "grey", width = 1)

                #coluna 2
        janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y1, 4, fill = "grey93", outline = "grey", width = 1)
        janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y2, 4, fill = "grey93", outline = "grey", width = 1)
        janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y3, 4, fill = "grey93", outline = "grey", width = 1)
                
        digito += 1

desenha_braile()

    #Exibe braile
def preeche_braile():
    global braile, palavra_traduzida
    
        #variáveis de desenho
    x_coluna1 = 14
    x_coluna2 = x_coluna1 + 15
    y1 = 14
    y2 = y1 + 15
    y3 = y2 + 15
    distancia_digito = 44
    
        #variáveis do loop 
    digito = 0
    coluna = 0
    bola = 0
    
        #"apaga" digitos anteriores
    desenha_braile()
    
    while digito < len(braile):
        #ignora espaços em branco
        if braile[digito] == None:
            digito += 1
                        
        while coluna < 2:
            if coluna == 0: #desenha na primeira coluna
                while bola < 3:
                    if braile[digito][coluna][bola] == 1:
                        if bola == 0: #desenha bola do topo
                            janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y1, 4, fill = "black", outline = "grey", width = 1)

                        elif bola == 1: #desenha bola do meio
                            janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y2, 4, fill = "black", outline = "grey", width = 1)

                        else: #desenha bola de baixo
                            janela_braile_interno.create_circle(x_coluna1+distancia_digito*digito, y3, 4, fill = "black", outline = "grey", width = 1)

                    bola += 1
                        
            else: #desenha na segunda coluna         
                while bola < 3:
                    if braile[digito][coluna][bola] == 1:
                        if bola == 0: #desenha bola do topo
                            janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y1, 4, fill = "black", outline = "grey", width = 1)
                                
                        elif bola == 1: #desenha bola do meio
                            janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y2, 4, fill = "black", outline = "grey", width = 1)

                        else: #desenha bola de baixo
                            janela_braile_interno.create_circle(x_coluna2+distancia_digito*digito, y3, 4, fill = "black", outline = "grey", width = 1)

                    bola += 1
                    
            coluna += 1
            bola = 0
        
        janela_braile_interno.create_text(x_coluna1+8+distancia_digito*digito, y3+15, text = palavra_traduzida[digito])
        digito += 1
        coluna = 0
        bola = 0

#Lop para exibir a stream
while True:
    video = stream.read()[1]
    video = cv.flip(video, 1)
    video_tratado = cv.cvtColor(video, cv.COLOR_BGR2RGB)
    video = ImageTk.PhotoImage(Image.fromarray(video_tratado))
    painel_foto_interno.config(image = video, width = 395, height = 251)
    painel_foto_interno.image = video
    janela.update()
    
stream.release()
cv.destroyAllWindows()

#janela.tk.mainloop()