Update app.py

app.py

@@ -14,7 +14,7 @@ nlp.add_pipe("span_marker",
 
 def get_html(html: str):
     """Convert HTML so it can be rendered."""
-    WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
+    WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem"; direction: rtl; >{}</div>"""
     # Newlines seem to mess with the rendering
     html = html.replace("\n", " ")
     style = "<style>mark.entity { display: inline-block }</style>"
@@ -26,7 +26,7 @@ def page_init():
     st.header("Named Entity Recognition Demo")
 
 
-@st.cache
+@st.cache_data
 def get_html_from_server(text):
     base_url = "https://ne-api.iahlt.org/api/hebrew/ner/?text={}"
 
@@ -50,17 +50,18 @@ def get_html_from_server(text):
 
     def render_entities(text):
         entities = get_entities(text)
-        html = displacy.render(entities, style="ent", manual=True)
-        return html
+        html = displacy.render(entities,
+                               style="ent",
+                               options={"direction": "rtl"},
+                               manual=True)
+        return html.replace("ltr", "rtl")
 
-    return render_entities(text)
+    return get_html(render_entities(text))
 
 
 if __name__ == '__main__':
     page_init()
 
-    displacy_options = {}
-
     sample_text = "יו\"ר ועדת הנוער נתן סלובטיק אמר שהשחקנים של אנחנו לא משתלבים באירופה."
 
     text = st.text_area("Text", sample_text, height=200, max_chars=1000)
@@ -68,8 +69,9 @@ if __name__ == '__main__':
     style = """
     <link rel="stylesheet" href="https://fonts.googleapis.com/css?family=David+Libre">
     <style>
-    textarea {
-        font-size: 16px;
+    .stTextArea textarea {
+        font-size: 20px;
+        font-color: black;
         font-family: 'David+Libre';
         direction: rtl;
     }
@@ -78,6 +80,8 @@ if __name__ == '__main__':
         font-family: 'David+Libre';
         direction: rtl;
     }
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
    </style>
     """
     st.write(style, unsafe_allow_html=True)
@@ -87,7 +91,7 @@ if __name__ == '__main__':
         html = displacy.render(
             doc,
             style="ent",
-            options=displacy_options,
+            options={"direction": "rtl"},
             manual=False,
         )
 
@@ -96,11 +100,13 @@ if __name__ == '__main__':
 
         html = f"""
         <div style="display: flex; flex-direction: row; justify-content: space-between; direction: rtl">
-            <div style="width: 50%">
+            <div>
             <h3>Nemo model results</h3>
                 {nemo_html}
             </div>
-            <div style="width: 50%">
+        </div>
+        <div style="display: flex; flex-direction: row; justify-content: space-between; direction: rtl">
+            <div>
             <h3>IAHLT results</h3>
                 {iahlt_html}
             </div>