labeler_PETA.py

# img_viewer.py

from layout import *
import os.path
import logging

from utils import *

# Level of warnings
logging.basicConfig(format='[%(levelname)s]: %(message)s', level=logging.CRITICAL)

# Dictionary of labels
labels = {}

# idx to an element in the folder
idx = 0
# Run the Event Loop

def update_img(filename, window):
    ''' 
    Update the frame where the image is shown
    input: name of the updated file
            window where the update is done
    output: image window updated 
    '''
    window["-TOUT-"].update(filename)
    window["-IMAGE-"].update(data=get_img_data(filename))

while True:
    event, values = window.read()
    
    if event == "Exit" or event == sg.WIN_CLOSED:
        break
    # Folder name was filled in, make a list of files in the folder
    if event == "-FOLDER-":
        folder = values["-FOLDER-"]
        try:
            # Get list of files in folder
            file_list = os.listdir(folder)
        except:
            file_list = []

        fnames = [
            f
            for f in file_list
            if os.path.isfile(os.path.join(folder, f))
            and f.lower().endswith((".png", ".jpg"))
        ]
        filename = os.path.join(
                values["-FOLDER-"], fnames[idx]
            )

        #attributes={fileName:[] for fileName in fnames}

        logging.info('Filename at START: {}'.format(filename))
        update_img(filename, window)

    elif event == 'Next':
        try:
            logging.info('Filename before NEXT: {}'.format(filename))
            logging.info('Values of values_array: {}'.format(values))
            attributes=pre_process_dict(values, fnames[idx])
            logging.info("Attributes after process in Next: {}".format(attributes))
            append_row(attributes, idx)
            logging.info("Dataframe df attributes: \n {}".format(df_attributes))
            
            idx += 1

            #prueba
            #print('Estos son los values: {}'.format(values))
            
            
            #print('Estos son los values despues: {}'.format(values))
            filename = os.path.join(values["-FOLDER-"], fnames[idx])
            update_img(filename, window)
            logging.info("Values before put them by default: {}".format(values))

            try:
                row_dict = extract_row(idx)
                for key in row_dict:
                    window[key].update(row_dict[key])
            except:
                for key in values:
                    if key!= '-FOLDER-' and key!= 'Browse':
                        window[key].update(False)
            logging.info("Values after put them by befault: {}".format(values))

                
        except:
            pass
    elif event == 'Back':
        # try:
        logging.info('Filename before BACK: {}'.format(filename))
        logging.info('Values of values_array: {}'.format(values))
        attributes=pre_process_dict(values, fnames[idx])
        logging.info("Attributes after process in Back: {}".format(attributes))
        append_row(attributes, idx)
        logging.info("Dataframe df attributes: \n {}".format(df_attributes))

        idx -= 1

        filename = os.path.join(values["-FOLDER-"], fnames[idx])
        update_img(filename, window)

        #TODO Extract the data from the dataframe and update values_array
        row_dict = extract_row(idx)
        for key in row_dict:
            window[key].update(row_dict[key])
        
        # except:
        #         pass
    elif event =='Save':
        attributes=pre_process_dict(values, fnames[idx])
        append_row(attributes, idx)
        df_attributes.to_csv('test.txt', index=False, header=False, sep=' ')
    
    try:
        logging.critical('Index value: {}'.format(idx))
        if (idx+1) > (len(fnames)-1):
            window['Next'].update(disabled=True)
        elif (idx-1) < 0:
            window['Back'].update(disabled=True)
        else:
            window['Save'].update(disabled=False)
            window['Next'].update(disabled=False)
            window['Back'].update(disabled=False)
    except:
        pass


    
window.close()