Update app.py

app.py

@@ -46,11 +46,9 @@ def model(prompt):
     analyzed_paraphrased_sentences, selected_sentences, discarded_sentences = analyze_entailment(sentence, paraphrased_sentences, 0.7)
     common_grams = find_common_subsequences(sentence, selected_sentences)
     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_dict(common_grams, analyzed_paraphrased_sentences, "Paraphrased Sentences")  # Fix parameter order
-    highlighted_selected_sentences = highlight_common_words_dict(common_grams, selected_sentences, "Selected Sentences from the Paraphrased Sentences", "#C1E1C1")
-    highlighted_discarded_sentences = highlight_common_words_dict(common_grams, discarded_sentences, "Discarded Sentences from the Paraphrased Sentences", "#FDE0E0")
-    tree = generate_plot(sentence)
-    return highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree
+    highlighted_paraphrased_sentences = highlight_common_words_dict(common_grams, selected_sentences, discarded_sentences, "Sentences Generated by the Paraphraser")
+    tree = generate_plot(sentence, list(selected_sentences.keys()))
+    return highlighted_user_prompt, highlighted_paraphrased_sentences, tree
 
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
@@ -67,17 +65,14 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         highlighted_user_prompt = gr.HTML()
 
     with gr.Row():
-        highlighted_selected_sentences = gr.HTML()
-
-    with gr.Row():
-        highlighted_discarded_sentences = gr.HTML()
+        highlighted_paraphrased_sentences = gr.HTML()
 
     with gr.Row():
         tree = gr.Plot()
 
-    submit_button.click(model, inputs=user_input, outputs=[highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree])
+    submit_button.click(model, inputs=user_input, outputs=[highlighted_user_prompt, highlighted_paraphrased_sentences, tree])
     clear_button.click(lambda: "", inputs=None, outputs=user_input)
-    clear_button.click(lambda: "", inputs=None, outputs=[highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree])
+    clear_button.click(lambda: "", inputs=None, outputs=[highlighted_user_prompt, highlighted_paraphrased_sentences, tree])
 
 # Launch the demo
 demo.launch(share=True)