DocumentComparator/Application.py at master · jaskier07/DocumentComparator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
from functools import partial
from tkinter import *
from tkinter import filedialog
from tkinter import font
from tkinter import messagebox
from tkinter.ttk import *

from SimilarityTable import SimilarityTable
from drawing.GraphDrawer import GraphDrawer
from DocumentComparator import DocumentComparator
from utils.IOUtils import IOUtils
import nltk

DEMO_MODE = False


def hide_components():
    bar['value'] = 0
    bar.grid_remove()
    bar.update()

    button_show_similarity_table.grid_remove()

    label_info_progressbar['text'] = ''


def show_similarity_table(arr, pdf_names):
    SimilarityTable().createAndShow(arr, pdf_names, window)


def compare_documents(paths_to_pdf_files, pdf_names):
    bar['value'] = 0
    bar.update()

    label_info_progressbar['text'] = 'Comparing...'
    label_info_progressbar.update()
    bar.grid(row=4, column=0)

    dc = DocumentComparator()
    arr = dc.compare_documents(paths_to_pdf_files, bar)

    label_info_progressbar['text'] = 'Comparing completed.'
    label_info_progressbar.update()

    button_show_similarity_table.configure(text='Show similarity table',
                                           command=partial(show_similarity_table, arr, pdf_names))
    button_show_similarity_table.grid(row=5, column=0)

    # TODO create thread here and make sure that previous has ended work
    drawer = GraphDrawer(DEMO_MODE)
    drawer.draw(arr, pdf_names)


def browse_files():
    folder_path = filedialog.askopenfilenames(filetypes=(("pdf files", "*.pdf"), ("all files", "*.*")))
    if folder_path != '':
        [paths_to_pdf_files, pdf_names] = IOUtils.list_pdf_files_in_dir(folder_path)

        hide_components()

        if len(pdf_names) < 2:
            messagebox.showerror('File selection', 'Select two or more PDF files.')
        else:
            listbox.delete(0, END)
            for (index, elem) in enumerate(pdf_names):
                listbox.insert(index, elem)

            label_header_info['text'] = 'PDF selection:'
            button_compare_documents = Button(text="Compare documents",
                                              command=partial(compare_documents, paths_to_pdf_files, pdf_names))
            button_compare_documents.grid(column=0, row=2, sticky="S", padx=(5, 5))


def configure_styles():
    default_font = font.nametofont("TkDefaultFont")
    default_font.configure(size=12)
    window.option_add("*Font", default_font)


if not DEMO_MODE:
    nltk.download('stopwords')
    nltk.download('wordnet')

    window = Tk()
    window.title("Document comparator")
    window.geometry('660x400')

    configure_styles()

    label_program_name = Label(window, text='Document comparator')
    label_program_name.configure(font=('calibri', 20, "bold"), foreground="#3d423a")
    label_program_name.grid(row=0, column=0, padx=(5, 5))

    label_header_info = Label(window, text='First select PDF files...')
    label_header_info.configure(font=('calibri', 15))
    label_header_info.grid(row=0, column=1, padx=(5, 5), sticky=W, columnspan=3)

    listbox = Listbox(window)
    listbox.yview()
    listbox.grid(row=1, column=1, sticky=N + W, rowspan=14, columnspan=3)
    listbox.configure(width=40, height=15)

    scroll_vertical = Scrollbar(window, orient=VERTICAL)
    scroll_vertical.config(command=listbox.yview)
    scroll_vertical.grid(row=1, column=4, sticky=N + S, rowspan=14)

    scroll_horizontal = Scrollbar(window, orient=HORIZONTAL)
    scroll_horizontal.config(command=listbox.xview)
    scroll_horizontal.grid(row=15, column=1, sticky=E + W, columnspan=3)

    listbox.config(yscrollcommand=scroll_vertical.set, xscrollcommand=scroll_horizontal.set)

    label_info_progressbar = Label(window, text='')
    label_info_progressbar.grid(row=3, column=0, sticky="S")

    bar = Progressbar(window, length=200)

    button_show_similarity_table = Button()

    button_browse = Button(text="Select files", command=browse_files)
    button_browse.grid(row=1, column=0, sticky="S", padx=(5, 5))

    window.protocol("WM_DELETE_WINDOW", IOUtils.delete_files_from_assets())
    window.mainloop()
else:
    arr = [[0, 0.53682293, 0.49240103, 0.49318128, 0.04009207, 0.03515585],
           [0.53682293, 0, 0.49677289, 0.54755336, 0.01423449, 0.01342816],
           [0.49240103, 0.49677289, 0, 0.4856281, 0.03298175, 0.0299892],
           [0.49318128, 0.54755336, 0.4856281, 0, 0.02699985, 0.02418876],
           [0.04009207, 0.01423449, 0.03298175, 0.02699985, 0, 0.97959166],
           [0.03515585, 0.01342816, 0.0299892, 0.02418876, 0.97959166, 0]]
    filenames = ['IO Analiza biznesowa i systemowa.pdf', 'IO Obszary działań IO.pdf', 'IO Projektowanie.pdf',
                 'IO Wprowadzenie.pdf', 'Kamizelka.pdf', 'Latarnik.pdf']

    GraphDrawer(demo_mode=DEMO_MODE).draw(arr, filenames)