Update app.py

app.py

@@ -1,7 +1,60 @@
-import torch
+from __future__ import annotations
+from typing import Iterable
+
 import gradio as gr
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
+
+import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification, pipeline
 import os
+import time
+
+# Helper function to convert hex to RGB tuple
+def hex_to_rgb(hex_color: str) -> tuple[int, int, int]:
+    hex_color = hex_color.lstrip('#')
+    return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
+
+class MPGPoster(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.Color(*hex_to_rgb("#f47317")),
+        background_hue: colors.Color | str = colors.Color(*hex_to_rgb("#f6f6f6ff")),
+        secondary_dark_hue: colors.Color | str = colors.Color(*hex_to_rgb("#006c66")),
+        secondary_light_hue: colors.Color | str = colors.Color(*hex_to_rgb("#6ad5bc")),
+        tertiary_highlight_hue: colors.Color | str = colors.Color(*hex_to_rgb("#fbf22c")),
+        spacing_size: sizes.Size | str = sizes.spacing_md,
+        radius_size: sizes.Size | str = sizes.radius_md,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Quicksand"),
+            "ui-sans-serif",
+            "sans-serif",
+        ),
+        font_mono: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"),
+            "ui-monospace",
+            "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_dark_hue,  # Assuming secondary dark hue as main secondary
+            neutral_hue=background_hue,  # Assuming background hue as neutral
+            spacing_size=spacing_size,
+            radius_size=radius_size,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+
+# Using the theme in a Gradio interface
+mpg_poster = MPGPoster()
 
 auth_token = os.environ['HF_TOKEN']
 
@@ -89,7 +142,7 @@ iface = gr.Interface(
     title="Scoring Demo",
     description="Autobiographical Memory Analysis: This demo combines two text - and two sequence classification models to showcase our automated Autobiographical Interview scoring method. Submit a narrative to see the results.",
     examples =examples,
-    theme="soft" #monochrome
+    theme= mpg_poster #"soft" #monochrome
 )
 
 # Launch the combined interface