Update app.py

app.py

@@ -145,6 +145,23 @@ def model(prompt):
 
     return [highlighted_user_prompt, highlighted_accepted_sentences, highlighted_discarded_sentences] + trees1 + trees2 + reparaphrased_sentences_list + [three_D_plot]
 
+# Logic for the new "Paraphrase and Discarded Sentence Generator" button
+def generate_paraphrase_and_discarded_sentences(prompt):
+    user_prompt = prompt
+    paraphrased_sentences = generate_paraphrase(user_prompt)
+    analyzed_paraphrased_sentences, selected_sentences, discarded_sentences = analyze_entailment(user_prompt, paraphrased_sentences, 0.7)
+    
+    # Combine discarded sentences with their entailment scores
+    discarded_sentences_with_scores = [
+        f"{sentence} (Entailment Score: {score:.2f})"
+        for sentence, score in discarded_sentences.items()
+    ]
+    
+    # Prepare paraphrased sentences for display
+    paraphrased_sentences_html = highlight_common_words_dict([], selected_sentences, "Paraphrased Sentences")
+    discarded_sentences_html = "<br>".join(discarded_sentences_with_scores)
+    
+    return paraphrased_sentences_html, discarded_sentences_html
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     gr.Markdown("# **AIISC Watermarking Model**")
@@ -156,6 +173,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         submit_button = gr.Button("Submit")
         clear_button = gr.Button("Clear")
         generate_non_melting_point_button = gr.Button("Generate Non-Melting Point")  # New button
+        paraphrase_discard_button = gr.Button("Paraphrase and Discarded Sentence Generator")
 
     with gr.Row():
         highlighted_user_prompt = gr.HTML()
@@ -226,6 +244,12 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         inputs=user_input,
         outputs=highlighted_user_prompt
     )
+
+    paraphrase_discard_button.click(
+        generate_paraphrase_and_discarded_sentences,
+        inputs=user_input,
+        outputs=[highlighted_accepted_sentences, highlighted_discarded_sentences]
+    )
     clear_button.click(lambda: "", inputs=None, outputs=user_input)
     clear_button.click(lambda: "", inputs=None, outputs=[highlighted_user_prompt, highlighted_accepted_sentences, highlighted_discarded_sentences] + tree1_tabs + tree2_tabs + reparaphrased_sentences_tabs + [three_D_plot])