FreshBench / diff_color.py
tREeFrOGorigami's picture
structure
5e1514b
raw
history blame
905 Bytes
from difflib import Differ
import gradio as gr
def diff_texts(text1, text2):
d = Differ()
rtn =[
(token[2:], token[0] if token[0] != " " else None)
for token in d.compare(text1, text2)
]
print(rtn)
return rtn
demo = gr.Interface(
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.",
),
],
gr.HighlightedText(
label="Diff",
combine_adjacent=True,
show_legend=True,
color_map={"+": "red", "-": "green"}),
theme=gr.themes.Base()# the return is here
)
if __name__ == "__main__":
demo.launch()