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
|
#!/usr/bin/env python3
# python3 -m pip install python-Levenshtein
from Levenshtein import distance as levenshtein_distance
import typing
DEBUG = False
# Find the window where the distance between these two transcriptions is
# minimized and use it to stitch them together.
def matchStringList(old_words: typing.List[str],
new_words: typing.List[str], window_size = 6) -> str:
if old_words == new_words:
return " ".join(old_words)
elif len(old_words) >= window_size and len(new_words) >= window_size:
# Find the window where the cumulative string distance
# between the words in that window in the old/new transcription
# is minimized.
old_slice = old_words[len(old_words) - window_size:]
best_match_i = None
best_match_d = window_size * 1000
for i in range(0, 1 + len(new_words) - window_size):
new_slice = new_words[i:i + window_size]
cur_d = 0
for j in range(0, window_size):
cur_d += levenshtein_distance(old_slice[j], new_slice[j])
if cur_d < best_match_d:
best_match_i = i
best_match_d = cur_d
old_prefix = old_words[0:len(old_words) - window_size]
overlap = new_words[best_match_i:best_match_i + window_size]
new_suffix = new_words[best_match_i + window_size:]
#print("Best match i: {}".format(best_match_i))
#print("Window size: {}".format(window_size))
#print("Old prefix: {}".format(old_prefix))
#print("Overlap: {}".format(overlap))
#print("New suffix: {}".format(new_suffix))
return " ".join(old_prefix + new_words[best_match_i:])
else:
return " ".join(new_words)
def matchSpaceDelimitedStrings(old_text: str, new_text: str, window_size = 4) -> str:
old_words = old_text.split()
new_words = new_text.split()
return matchStringList(old_words, new_words, window_size)
def matchStrings(old_text: str, new_text: str, window_size = 3) -> str:
if old_text == new_text:
return old_text
elif len(old_text) >= window_size and len(new_text) >= window_size:
# Find the window where the cumulative string distance
# between the text in that window in the old/new transcription
# is minimized.
best_match_i = None
best_match_j = None
best_match_d = window_size * 1000
for i in range(0, 1 + len(old_text) - window_size):
old_slice = old_text[i:i + window_size]
for j in range(0, 1 + len(new_text) - window_size):
new_slice = new_text[j:j + window_size]
cur_d = 0
for k in range(0, window_size):
cur_d += levenshtein_distance(old_slice[k], new_slice[k])
if cur_d <= best_match_d:
best_match_i = i
best_match_j = j
best_match_d = cur_d
if DEBUG:
print("optimum at old '{}'/{} new '{}'/{} d={}".format(
old_slice, i, new_slice, j, cur_d))
old_prefix = old_text[0:best_match_i]
overlap = new_text[best_match_j:best_match_j + window_size]
new_suffix = new_text[best_match_j + window_size:]
if DEBUG:
print("Best match i: {}".format(best_match_i))
print("Best match j: {}".format(best_match_j))
print("Window size: {}".format(window_size))
print("Old prefix: {}".format(old_prefix))
print("Overlap: {}".format(overlap))
print("New suffix: {}".format(new_suffix))
print("Input 1: {}".format(old_text))
print("Input 2: {}".format(new_text))
print("Output: {}".format(old_prefix +
new_text[best_match_j:]))
return old_prefix + new_text[best_match_j:]
else:
return new_text
if __name__ == "__main__":
# Identical transcriptions should not be changed.
assert(matchSpaceDelimitedStrings("This is a test case.", "This is a test case.", window_size = 3) == "This is a test case.")
# A suffix should be detected and ignored.
assert(matchSpaceDelimitedStrings("This is a test case.", "is a test case.", window_size = 3) == "This is a test case.")
# A lengthening suffix should be correctly appended.
assert(matchSpaceDelimitedStrings("This is a test", "is a test case.", window_size = 3) == "This is a test case.")
# A strictly longer transcription should override the old prefix.
assert(matchSpaceDelimitedStrings("This is a test", "This is a test case.", window_size = 3) == "This is a test case.")
# Paranoia: repetitive text broke the older implementation, so I included
# some test cases without fully understanding what the old problem was.
assert(matchSpaceDelimitedStrings("test test test", "test test test test test test", window_size
= 3) == "test test test test test test")
assert(matchSpaceDelimitedStrings("test test test test test test", "test test test", window_size
= 3) == "test test test test test test")
print(matchStrings("foo bar", "bar baz"))
print(matchStrings("alpha beta", "beta gamma"))
in1 = "Okay, what about now? Looks like it sort of works. Key word being sort of."
in2 = "okay what about now looks like it sort of works key word being sort of looks"
bad_out = "Okay, what about now? Looks like it sort of works. Key word being sort of works key word being sort of looks"
good_out = "Okay, what about now? Looks like it sort of works. Key word being sort of looks"
print(matchStrings(in1, in2))
in1 = "This repository can take"
in2 = "This repository contains the code for"
bad_out = "This repository can tode for"
good_out = "This repository contains the code for"
print(matchStrings(in1, in2))
print("Tests passed.")
|
