Update app.py

app.py

@@ -20,6 +20,7 @@ from models import (
     LabelRobertaForTokenClassification,
 )
 
+
 DEFAULT_TEXT = "τίω δέ μιν ἐν καρὸς αἴσῃ."
 BUTTON_CSS = "float: right; --tw-border-opacity: 1; border-color: rgb(229 231 235 / var(--tw-border-opacity)); --tw-gradient-from: rgb(243 244 246 / 0.7); --tw-gradient-stops: var(--tw-gradient-from), var(--tw-gradient-to, rgb(243 244 246 / 0)); --tw-gradient-to: rgb(229 231 235 / 0.8); --tw-text-opacity: 1; color: rgb(55 65 81 / var(--tw-text-opacity));    border-width: 1px; --tw-bg-opacity: 1; background-color: rgb(255 255 255 / var(--tw-bg-opacity)); background-image: linear-gradient(to bottom right, var(--tw-gradient-stops)); display: inline-flex; flex: 1 1 0%; align-items: center; justify-content: center;    --tw-shadow: 0 1px 2px 0 rgb(0 0 0 / 0.05); --tw-shadow-colored: 0 1px 2px 0 var(--tw-shadow-color); box-shadow: var(--tw-ring-offset-shadow, 0 0 #0000), var(--tw-ring-shadow, 0 0 #0000), var(--tw-shadow); -webkit-appearance: button; border-radius: 0.5rem; padding-top: 0.5rem; padding-bottom: 0.5rem; padding-left: 1rem; padding-right: 1rem; font-size: 1rem; line-height: 1.5rem; font-weight: 600;"
 DEFAULT_COLOR = "white"
@@ -33,7 +34,6 @@ MODEL_PATHS = {
 MODEL_MAX_LENGTH = 512
 
 AUTH_TOKEN = os.environ.get("TOKEN") or True
-
 # PoS
 pos_tokenizer = AutoTokenizer.from_pretrained(
     MODEL_PATHS["POS"], model_max_length=MODEL_MAX_LENGTH, use_auth_token=AUTH_TOKEN, revision="8bd84df2bcaee089307fd604c80139a34ac71f12",
@@ -196,7 +196,7 @@ def parse(
         )["arcs"]
 
     options = {"compact": compact, "bg": bg, "color": text}
-    svg = displacy.render(doc, manual=True, style="dep", options=options)
+    svg = displacy.render(doc, manual=True,  options=options) #style="dep",
 
     if col_lemmata:
         soup = BeautifulSoup(svg, "lxml-xml")
@@ -207,26 +207,31 @@ def parse(
 
     return svg, download_link
 
+
 def setup_parser_ui():
-    theme = gr.themes.Monochrome()
-    with gr.Blocks(theme=theme) as demo:  
-        with gr.Group():
-            gr.Markdown("# Athena's Lens")
-            gr.Markdown(
-                    "### From Ἀlkaios to Ὠrigen: A Modern Lens on Timeless Texts"
-                )
-        with gr.Group():
-            gr.Markdown("## Enter some text")
+    demo = gr.Blocks(css="scrollbar.css")
+    with demo:
+        with gr.Row():
             with gr.Row():
-                with gr.Column(scale=0.5):
-                    text_input = gr.Textbox(
-                        value=DEFAULT_TEXT, interactive=True, label="Input Text"
+                with gr.Column():
+                    gr.Markdown("# Athena's Lens")
+                    gr.Markdown(
+                        "### From Ἀlkaios to Ὠrigen: A Modern Lens on Timeless Texts"
                     )
-            with gr.Row():
-                with gr.Column(scale=0.25):
-                    button = gr.Button("Update", variant="primary")
-
-        with gr.Group():
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown(" ## Enter some text")
+                with gr.Row():
+                    with gr.Column(scale=0.5):
+                        text_input = gr.Textbox(
+                            value=DEFAULT_TEXT, interactive=True, label="Input Text"
+                        )
+                with gr.Row():
+                    with gr.Column(scale=0.25):
+                        button = gr.Button("Update", variant="primary")#.style(
+                            #full_width=False
+                        #)
+        with gr.Row():
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -242,7 +247,6 @@ def setup_parser_ui():
                         bg = gr.Textbox(label="Background Color", value=DEFAULT_COLOR)
                     with gr.Column():
                         text = gr.Textbox(label="Text Color", value="black")
-            with gr.Group():
                 with gr.Row():
                     dep_output = gr.HTML(
                         value=parse(
@@ -254,16 +258,20 @@ def setup_parser_ui():
                             False,
                             DEFAULT_COLOR,
                             "black",
-                        )[0]
+                        )[0],
+                        elem_classes=["output-html"]
                     )
                 with gr.Row():
-                    with gr.Column():
-                        dep_button = gr.Button("Update Parser", variant="primary")
+                    with gr.Column(scale=0.25):
+                        dep_button = gr.Button(
+                            "Update Parser", variant="primary"
+                        )#.style(full_width=False)
                     with gr.Column():
                         dep_download_button = gr.HTML(
                             value=download_svg(dep_output.value)
                         )
-        with gr.Group():
+
+        with gr.Row():
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -277,21 +285,22 @@ def setup_parser_ui():
                         )
                         gr.Markdown(
                             """```bibtex
-    @incollection{riemenschneider-frank-2023-exploring,
-        title = "Exploring Large Language Models for Classical Philology",
-        author = "Riemenschneider, Frederick  and Frank, Anette",
-        booktitle = "Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)",
-        month = jul,
-        year = "2023",
-        address = "Toronto, Canada",
-        publisher = "Association for Computational Linguistics",
-        url = "https://aclanthology.org/2023.acl-long.846",
-        doi = "10.18653/v1/2023.acl-long.846",
-        pages = "15181--15199",
-    }
-    ```
-                        """
+@incollection{riemenschneider-frank-2023-exploring,
+    title = "Exploring Large Language Models for Classical Philology",
+    author = "Riemenschneider, Frederick  and Frank, Anette",
+    booktitle = "Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)",
+    month = jul,
+    year = "2023",
+    address = "Toronto, Canada",
+    publisher = "Association for Computational Linguistics",
+    url = "https://aclanthology.org/2023.acl-long.846",
+    doi = "10.18653/v1/2023.acl-long.846",
+    pages = "15181--15199",
+}
+```
+                """
                         )
+
         button.click(
             execute_parse,
             inputs=[
@@ -324,7 +333,6 @@ def setup_parser_ui():
     return demo
 
 
-
 def main():
     demo = setup_parser_ui()
     demo.launch()