gradio/demo/diff_texts/run.py

37 lines
806 B
Python
Raw Normal View History

2020-08-20 05:27:22 +08:00
from difflib import Differ
import gradio as gr
2020-08-20 05:27:22 +08:00
def diff_texts(text1, text2):
d = Differ()
return [
(token[2:], token[0] if token[0] != " " else None)
for token in d.compare(text1, text2)
2020-08-20 05:27:22 +08:00
]
2020-08-28 23:56:03 +08:00
2022-03-28 23:11:14 +08:00
demo = gr.Interface(
2020-08-20 05:27:22 +08:00
diff_texts,
[
gr.Textbox(
label="Text 1",
info="Initial text",
lines=3,
value="The quick brown fox jumped over the lazy dogs.",
),
gr.Textbox(
label="Text 2",
info="Text to compare",
lines=3,
value="The fast brown fox jumps over lazy dogs.",
),
2020-08-20 05:27:22 +08:00
],
gr.HighlightedText(
label="Diff",
combine_adjacent=True,
).style(color_map={"+": "red", "-": "green"}),
)
if __name__ == "__main__":
2022-03-28 23:11:14 +08:00
demo.launch()