Upload 5 files

__init__.py

@@ -0,0 +1,5 @@
+import os
+import sys
+
+# add current folder to the python path
+sys.path.append(os.path.dirname(__file__))

app.py

@@ -9,8 +9,7 @@ from pytorch_ie.auto import AutoPipeline
 from pytorch_ie.documents import TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions
 from pytorch_ie.models import *  # noqa: F403
 from pytorch_ie.taskmodules import *  # noqa: F403
-
-from rendering_utils import render_pretty_table, render_spacy
+from rendering_utils import render_displacy, render_pretty_table
 
 RENDER_WITH_DISPLACY = "displaCy + highlighted arguments"
 RENDER_WITH_PRETTY_TABLE = "Pretty Table"
@@ -38,7 +37,7 @@ def render(document_txt: str, render_with: str, render_kwargs_json: str) -> str:
     if render_with == RENDER_WITH_PRETTY_TABLE:
         html = render_pretty_table(document, **render_kwargs)
     elif render_with == RENDER_WITH_DISPLACY:
-        html = render_spacy(document, **render_kwargs)
+        html = render_displacy(document, **render_kwargs)
     else:
         raise ValueError(f"Unknown render_with value: {render_with}")
 
@@ -70,8 +69,7 @@ if __name__ == "__main__":
     )
 
     default_render_kwargs = {
-        "style": "ent",
-        "options": {
+        "entity_options": {
             # we need to convert the keys to uppercase because the spacy rendering function expects them in uppercase
             "colors": {
                 "own_claim".upper(): "#009933",

rendering_utils.py

@@ -4,7 +4,6 @@ from typing import Dict, List, Optional, Union
 
 from pytorch_ie.annotations import BinaryRelation
 from pytorch_ie.documents import TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions
-
 from rendering_utils_displacy import EntityRenderer
 
 
@@ -25,12 +24,11 @@ def render_pretty_table(
     return html
 
 
-def render_spacy(
+def render_displacy(
     document: TextDocumentWithLabeledSpansBinaryRelationsAndLabeledPartitions,
-    style="ent",
     inject_relations=True,
     colors_hover=None,
-    options={},
+    entity_options={},
     **render_kwargs,
 ):
 
@@ -43,7 +41,7 @@ def render_spacy(
         "title": None,
     }
 
-    renderer = EntityRenderer(options=options)
+    renderer = EntityRenderer(options=entity_options)
     html = renderer.render([spacy_doc], page=True, minify=True).strip()
 
     html = "<div style='max-width:100%; max-height:360px; overflow:auto'>" + html + "</div>"

requirements.txt

@@ -2,5 +2,4 @@ transformers==4.36.0
 gradio
 prettytable==3.10.0
 pie-modules>=0.10.9,<0.11.0
-spacy==2.3.9
 beautifulsoup4==4.12.3