Update app.py

app.py

@@ -68,7 +68,7 @@ def paraphrase(
 
 def remove_punctuations(text):
     # Remove punctuations while preserving hyphenated words
-    return re.sub(r'(?<!\w)-|-(?!\w)', ' ', re.sub(r'[^\w\s-]', '', text))
+    return re.sub(r'(?<!\w)-|-(?!\w)', ' ', re.sub(r'[^\w\s,-]', '', text))
 
 def tokenize(sentence):
     # Remove punctuations using the updated function and tokenize the sentence into words
@@ -264,20 +264,20 @@ st.markdown("\nHighlighted Main Sentence with LCS:")
 st.write(highlighted_sentence, unsafe_allow_html=True)
 
 
-highlighted_sentence = paraphrases[0]
-highlighted_text = []
-
-for substring in combined_words_list[0]:
-    highlighted_sentence = highlighted_sentence.replace(substring, f'<span style="background-color: blue; color: white;">{substring}</span>')
-    highlighted_text.append(substring)
+highlighted_sentence_list = []
 
-st.markdown("Common substrings between main_sentence and Paraphrase1:")
+for i in range(0, 5):
+    highlighted_sentence = paraphrases[i]
+    highlighted_text = []
 
-# for substring in highlighted_text:
-#     st.write(substring)
+    for substring in combined_words_list[i]:
+        highlighted_sentence = highlighted_sentence.replace(substring, f'<span style="background-color: blue; color: white;">{substring}</span>')
+        highlighted_text.append(substring)
+    
+    highlighted_sentence_list.append(highlighted_sentence)
 
-st.markdown("\nHighlighted Main Sentence with LCS:")
-st.write(highlighted_sentence, unsafe_allow_html=True)
+st.markdown("\nHighlighted Paraphrase Sentences with LCS:")
+st.write(highlighted_sentence_list, unsafe_allow_html=True)