Update app.py

app.py

@@ -222,10 +222,10 @@ for i in combined_words_list[0]:
 # highlighted_sentences = []
 
 
-def highlight_text(text, substrings, color):
+def highlight_text(text, substrings):
     highlighted_text = text
-    for substring in substrings:
-        highlighted_text = highlighted_text.replace(substring, f'<span style="background-color: {color}; color: white;">{substring}</span>')
+    for i, substring in enumerate(substrings):
+        highlighted_text = highlighted_text.replace(substring, f'<span style="background-color: {colors[i]}; color: white;">{substring}</span>')
     return highlighted_text
 
 # Assuming you have main_sentence, paraphrases, and common_substrings defined
@@ -233,20 +233,17 @@ def highlight_text(text, substrings, color):
 colors = ['blue', 'green', 'orange', 'purple', 'red']  # Different colors for each paraphrase
 
 # Highlight main_sentence
-highlighted_main_sentence = main_sentence
-for i, common_substring in enumerate(common_substrings):
-    highlighted_main_sentence = highlight_text(highlighted_main_sentence, common_substring, colors[i])
+highlighted_main_sentence = highlight_text(main_sentence, common_substrings[0])
 
 st.markdown("\nHighlighted Main Sentence:")
 st.write(highlighted_main_sentence, unsafe_allow_html=True)
 
 # Highlight paraphrases
-for i, (paraphrase, common_substring) in enumerate(zip(paraphrases, common_substrings)):
-    highlighted_paraphrase = highlight_text(paraphrase, common_substring, colors[i])
+for i, (paraphrase, common_substring) in enumerate(zip(paraphrases, common_substrings[1:])):
+    highlighted_paraphrase = highlight_text(paraphrase, common_substring)
     st.markdown(f"\nHighlighted Paraphrase {i+1}:")
     st.write(highlighted_paraphrase, unsafe_allow_html=True)
 
-
     
 
 # Assuming you have defined common_substrings and remove_overlapping functions
@@ -258,13 +255,13 @@ for i, (paraphrase, common_substring) in enumerate(zip(paraphrases, common_subst
 #     highlighted_sentence = highlighted_sentence.replace(substring, f'<span style="background-color: blue; color: white;">{substring}</span>')
 #     highlighted_text.append(substring)
 
-#st.markdown("Common substrings that occur in all five lists:")
+# st.markdown("Common substrings that occur in all five lists:")
 
 # for substring in highlighted_text:
 #     st.write(substring)
 
-st.markdown("\nHighlighted Main Sentence with LCS:")
-st.write(highlighted_sentence, unsafe_allow_html=True)
+# st.markdown("\nHighlighted Main Sentence with LCS:")
+# st.write(highlighted_sentence, unsafe_allow_html=True)