gradio/demo/diff_texts.py
2020-11-11 06:21:05 -08:00

30 lines
654 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]) 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",
" ": "none",
}))
iface.test_launch()
if __name__ == "__main__":
iface.launch()