Update app.py

app.py

@@ -140,7 +140,19 @@ def find_longest_match(string1, string2):
     
     return longest_match
 
-
+def remove_spaces_before_punctuation(text):
+    import string
+    punctuation = string.punctuation
+    result = ""
+    for i, char in enumerate(text):
+        if i == 0:
+            result += char
+        else:
+            if char in punctuation and text[i-1] == " ":
+                result = result[:-1] + char
+            else:
+                result += char
+    return result
 
 
 prompt_list=["The official position of the United States on the Russia-Ukraine war has been consistent in supporting Ukraine's sovereignty, territorial integrity, and the peaceful resolution of the conflict."    
@@ -259,7 +271,7 @@ for i in combined_words_list[0]:
 
 # Assuming you have defined common_substrings and remove_overlapping functions
 
-highlighted_sentence = main_sentence
+highlighted_sentence = remove_spaces_before_punctuation(" ".join(tokenize(main_sentence)))
 highlighted_text = []
 
 for substring in remove_overlapping(common_substrings):
@@ -281,7 +293,7 @@ highlighted_sentence_list = []
 colors = ['blue', 'green', 'orange', 'purple', 'red']
 
 for i in range(0, 5):
-    highlighted_sentence = paraphrases[i]
+    highlighted_sentence = remove_spaces_before_punctuation(" ".join(tokenize(paraphrases[i])))
     highlighted_text = []
 
     # Assign a unique color to each paraphrase