Update app.py

app.py

@@ -43,8 +43,8 @@ def model(prompt):
     sentence = prompt
     paraphrased_sentences = generate_paraphrase(sentence)
     common_grams = find_common_subsequences(sentence, paraphrased_sentences)
-    highlighted_user_prompt = highlight_common_words(common_grams, [sentence])  # Pass the sentence as a list
-    highlighted_paraphrased_sentences = highlight_common_words(common_grams, paraphrased_sentences)  # Fix parameter order
+    highlighted_user_prompt = highlight_common_words(common_grams, [sentence], "User Prompt (Highlighted and Numbered)")  # Pass the sentence as a list
+    highlighted_paraphrased_sentences = highlight_common_words(common_grams, paraphrased_sentences, "Paraphrased Sentences")  # Fix parameter order
     discarded_sentences = []
     tree = generate_plot(sentence)
     return highlighted_user_prompt, highlighted_paraphrased_sentences, discarded_sentences, tree