Update app.py

app.py

@@ -33,7 +33,8 @@ import gradio as gr
 from tree import generate_plot
 from paraphraser import generate_paraphrase
 from lcs import find_common_subsequences
-from highlighter import highlight_common_words
+from highlighter import highlight_common_words, highlight_common_words_dict
+from entailment import analyze_entailment
 
 nltk.download('stopwords')
 
@@ -42,12 +43,14 @@ nltk.download('stopwords')
 def model(prompt):
     sentence = prompt
     paraphrased_sentences = generate_paraphrase(sentence)
-    common_grams = find_common_subsequences(sentence, paraphrased_sentences)
+    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(common_grams, paraphrased_sentences, "Paraphrased Sentences")  # Fix parameter order
-    discarded_sentences = []
+    # 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_paraphrased_sentences, discarded_sentences, tree
+    return highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree
 
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
@@ -61,20 +64,20 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         clear_button = gr.Button("Clear")
 
     with gr.Row():
-        selected_sentence = gr.HTML()
+        highlighted_user_prompt = gr.HTML()
 
     with gr.Row():
-        html_output = gr.HTML()
+        highlighted_selected_sentences = gr.HTML()
 
     with gr.Row():
-        discarded_sentences = gr.Textbox(label="Discarded Sentences")
+        highlighted_discarded_sentences = gr.HTML()
 
     with gr.Row():
         tree = gr.Plot()
 
-    submit_button.click(model, inputs=user_input, outputs=[selected_sentence, html_output, discarded_sentences, tree])
+    submit_button.click(model, inputs=user_input, outputs=[highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree])
     clear_button.click(lambda: "", inputs=None, outputs=user_input)
-    clear_button.click(lambda: "", inputs=None, outputs=[selected_sentence, html_output, discarded_sentences, tree])
+    clear_button.click(lambda: "", inputs=None, outputs=[highlighted_user_prompt, highlighted_selected_sentences, highlighted_discarded_sentences, tree])
 
 # Launch the demo
 demo.launch(share=True)