use rendering utils

app.py

@@ -11,61 +11,11 @@ from pytorch_ie.documents import TextDocumentWithLabeledSpansBinaryRelationsAndL
 from pytorch_ie.models import *  # noqa: F403
 from pytorch_ie.taskmodules import *  # noqa: F403
 
-RENDER_WITH_DISPLACY = "displaCy + highlighted arguments"
-RENDER_WITH_PRETTY_TABLE = "Pretty Table"
-
-
-def render_pretty_table(
-    document: TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions, **render_kwargs
-):
-    from prettytable import PrettyTable
-
-    t = PrettyTable()
-    t.field_names = ["head", "tail", "relation"]
-    t.align = "l"
-    for relation in list(document.binary_relations) + list(document.binary_relations.predictions):
-        t.add_row([str(relation.head), str(relation.tail), relation.label])
-
-    html = t.get_html_string(format=True)
-    html = "<div style='max-width:100%; max-height:360px; overflow:auto'>" + html + "</div>"
-
-    return html
+from .rendering_utils import render_pretty_table, render_spacy
 
 
-def render_spacy(
-    document: TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions,
-    style="ent",
-    inject_relations=True,
-    colors_hover=None,
-    **render_kwargs,
-):
-    from spacy import displacy
-
-    spans = list(document.labeled_spans) + list(document.labeled_spans.predictions)
-    spacy_doc = {
-        "text": document.text,
-        "ents": [
-            {"start": entity.start, "end": entity.end, "label": entity.label} for entity in spans
-        ],
-        "title": None,
-    }
-
-    html = displacy.render(
-        spacy_doc, page=True, manual=True, minify=True, style=style, **render_kwargs
-    )
-    html = "<div style='max-width:100%; max-height:360px; overflow:auto'>" + html + "</div>"
-    if inject_relations:
-        binary_relations = list(document.binary_relations) + list(
-            document.binary_relations.predictions
-        )
-        sorted_entities = sorted(spans, key=lambda x: (x.start, x.end))
-        html = inject_relation_data(
-            html,
-            sorted_entities=sorted_entities,
-            binary_relations=binary_relations,
-            additional_colors=colors_hover,
-        )
-    return html
+RENDER_WITH_DISPLACY = "displaCy + highlighted arguments"
+RENDER_WITH_PRETTY_TABLE = "Pretty Table"
 
 
 def inject_relation_data(