Update app.py

app.py

@@ -20,7 +20,6 @@ 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"
@@ -34,6 +33,7 @@ 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",
@@ -207,31 +207,26 @@ def parse(
 
     return svg, download_link
 
-
 def setup_parser_ui():
-    demo = gr.Blocks(css="scrollbar.css")
-    with demo:
-        with gr.Box():
+    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")
             with gr.Row():
-                with gr.Column():
-                    gr.Markdown("# Athena's Lens")
-                    gr.Markdown(
-                        "### From Ἀlkaios to Ὠrigen: A Modern Lens on Timeless Texts"
+                with gr.Column(scale=0.5):
+                    text_input = gr.Textbox(
+                        value=DEFAULT_TEXT, interactive=True, label="Input Text"
                     )
-        with gr.Box():
-            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.Box():
+            with gr.Row():
+                with gr.Column(scale=0.25):
+                    button = gr.Button("Update", variant="primary")
+
+        with gr.Group():
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -247,6 +242,7 @@ 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(
@@ -261,29 +257,26 @@ def setup_parser_ui():
                         )[0]
                     )
                 with gr.Row():
-                    with gr.Column(scale=0.25):
-                        dep_button = gr.Button(
-                            "Update Parser", variant="primary"
-                        ).style(full_width=False)
+                    with gr.Column():
+                        dep_button = gr.Button("Update Parser", variant="primary")
                     with gr.Column():
                         dep_download_button = gr.HTML(
                             value=download_svg(dep_output.value)
                         )
-
-            with gr.Box():
-                with gr.Column():
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("## Contact")
-                            gr.Markdown(
-                                "If you have any questions, suggestions, comments, or problems, feel free to [reach out](mailto:[email protected])."
-                            )
-                            gr.Markdown("## Citation")
-                            gr.Markdown(
-                                "This space uses models from [this](https://aclanthology.org/2023.acl-long.846.pdf) paper."
-                            )
-                            gr.Markdown(
-                                """```bibtex
+        with gr.Group():
+            with gr.Column():
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("## Contact")
+                        gr.Markdown(
+                            "If you have any questions, suggestions, comments, or problems, feel free to [reach out](mailto:riemenschneider@cl.uni-heidelberg.de)."
+                        )
+                        gr.Markdown("## Citation")
+                        gr.Markdown(
+                            "This space uses models from [this](https://aclanthology.org/2023.acl-long.846.pdf) paper."
+                        )
+                        gr.Markdown(
+                            """```bibtex
     @incollection{riemenschneider-frank-2023-exploring,
         title = "Exploring Large Language Models for Classical Philology",
         author = "Riemenschneider, Frederick  and Frank, Anette",
@@ -297,9 +290,8 @@ def setup_parser_ui():
         pages = "15181--15199",
     }
     ```
-                    """
-                            )
-
+                        """
+                        )
         button.click(
             execute_parse,
             inputs=[
@@ -329,8 +321,8 @@ def setup_parser_ui():
             ],
             outputs=[dep_output, dep_download_button],
         )
+    return demo
 
-    demo.launch()
 
 
 def main():