Update app.py

app.py

@@ -3,6 +3,7 @@ from parrot import Parrot
 from gtts import gTTS
 from IPython.display import Audio
 import os
+import difflib
 
 # Create a Parrot instance
 parrot = Parrot(model_tag="turing", use_gpu=False)
@@ -32,6 +33,15 @@ def speak_paraphrased_text(text, language):
     tts.save("paraphrased_text.mp3")
     return Audio("paraphrased_text.mp3", autoplay=True)
 
+def highlight_diffs(original_text, paraphrased_text, level):
+    # Use diff-match-patch library to highlight the differences between original and paraphrased text
+    dmp = difflib.Differ()
+    diffs = list(dmp.compare(original_text.split(), paraphrased_text.split()))
+    if level == "only_changes":
+        return " ".join([diff[2:] for diff in diffs if diff.startswith("+ ")])
+    else:
+        return " ".join([diff[2:] if diff.startswith("+ ") else diff[2:].replace(diff[0], "<span style='background-color:yellow'>"+diff[0]+"</span>") for diff in diffs])
+
 # Create a Gradio interface
 interface = gr.Interface(
     fn=paraphrase_text,
@@ -40,20 +50,30 @@ interface = gr.Interface(
         gr.inputs.Dropdown(list(languages.keys()), label="Select language for paraphrasing:"),
         gr.inputs.Dropdown(list(models.keys()), label="Select model for paraphrasing:"),
         gr.inputs.Slider(minimum=1, maximum=10, default=5, label="Level of paraphrasing:"),
-        gr.inputs.Checkbox(label="Generate audio output?")
-    ],
-    outputs=gr.outputs.Textbox(label="Paraphrased text:"),
-    title="Parrot - Text Paraphrasing",
-    description="Enter a text to paraphrase using Parrot.",
-    theme="default"
+        gr.inputs.Checkbox(label="Generate audio output?"),
+        gr.inputs.File(label="Upload file containing text to paraphrase (Optional):")
+],
+outputs=[
+gr.outputs.Textbox(label="Paraphrased text:"),
+gr.outputs.Download(label="Download paraphrased text as text file:"),
+gr.outputs.HTML(label="Highlighted differences:")
+],
+title="Parrot Paraphrasing Tool",
+description="Enter text to paraphrase, select language and model, and choose level of paraphrasing."
 )
 
-# Launch the interface
-interface.launch()
+def app(text, language, model, level, audio_output, file_upload):
+    if file_upload:
+        with open(file_upload.name, "r", encoding="utf-8") as f:
+            text = f.read()
+            paraphrased_text = paraphrase_text(text, languages[language], models[model], level)
+    if audio_output:
+        speak_paraphrased_text(paraphrased_text, languages[language])
+        download_output = gr.outputs.Download(paraphrased_text, label="Download paraphrased text as text file:", mime_type="text/plain")
+        highlighted_diffs = highlight_diffs(text, paraphrased_text, level)
+    return paraphrased_text, download_output, highlighted_diffs
+
+interface.test_launch(app)
 
-# If the user has selected the "Generate audio output?" checkbox, call the speak_paraphrased_text function to generate and play the audio
-if interface.inputs[4].value:
-    speak_paraphrased_text(interface.outputs.value, languages[interface.inputs[1].value])
-    
-# Delete the generated audio file
-os.remove("paraphrased_text.mp3")
+if name == "main":
+    interface.launch()