gradio/demo/diff_texts.py
2021-08-25 21:42:45 +00:00

22 lines
642 B
Python

# Demo: (Textbox, Textbox) -> (HighlightedText)
import gradio as gr
from difflib import Differ
def diff_texts(text1, text2):
d = Differ()
return [
(token[2:], token[0] if token[0] != " " else None) for token in d.compare(text1, text2)
]
iface = gr.Interface(
diff_texts,
[
gr.inputs.Textbox(
lines=3, default="The quick brown fox jumped over the lazy dogs."),
gr.inputs.Textbox(
lines=3, default="The fast brown fox jumps over lazy dogs."),
],
gr.outputs.HighlightedText(color_map={"+": "lightgreen", "-": "pink", }))
if __name__ == "__main__":
iface.launch()