Update app.py

app.py

@@ -8,22 +8,35 @@ from gradio.themes.utils import colors, fonts, sizes
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification, pipeline
 import os
+import colorsys
 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))
 
+def generate_shades(base_color: str) -> colors.Color:
+    base_rgb = hex_to_rgb(base_color)
+    hsv = colorsys.rgb_to_hsv(*[v / 255.0 for v in base_rgb])
+
+    shades = []
+    for i in range(100, 1000, 100):
+        factor = i / 1000
+        new_v = hsv[2] * factor
+        rgb_shade = colorsys.hsv_to_rgb(hsv[0], hsv[1], new_v)
+        shades.append(tuple(int(v * 255) for v in rgb_shade))
+
+    return colors.Color(*shades)
+
 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")),
+        primary_hue: colors.Color | str = generate_shades("#f47317"),
+        background_hue: colors.Color | str = generate_shades("#f6f6f6ff"),
+        secondary_dark_hue: colors.Color | str = generate_shades("#006c66"),
+        secondary_light_hue: colors.Color | str = generate_shades("#6ad5bc"),
+        tertiary_highlight_hue: colors.Color | str = generate_shades("#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,
@@ -40,8 +53,8 @@ class MPGPoster(Base):
     ):
         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
+            secondary_hue=secondary_dark_hue,
+            neutral_hue=background_hue,
             spacing_size=spacing_size,
             radius_size=radius_size,
             text_size=text_size,
@@ -49,7 +62,6 @@ class MPGPoster(Base):
             font_mono=font_mono,
         )
 
-# Using the theme in a Gradio interface
 mpg_poster = MPGPoster()
 
 auth_token = os.environ['HF_TOKEN']