Update app.py

app.py

@@ -10,24 +10,35 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Auto
 import os
 import colorsys
 import time
+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 rgb_to_hex(rgb_color: tuple[int, int, int]) -> str:
+    return "#{:02x}{:02x}{:02x}".format(*rgb_color)
+
+def adjust_brightness(rgb_color: tuple[int, int, int], factor: float) -> tuple[int, int, int]:
+    # Convert RGB to HSV, adjust V (value), convert back to RGB
+    hsv_color = colorsys.rgb_to_hsv(*[v / 255.0 for v in rgb_color])
+    new_v = max(0, min(hsv_color[2] * factor, 1))  # Ensure the value is within [0, 1]
+    new_rgb = colorsys.hsv_to_rgb(hsv_color[0], hsv_color[1], new_v)
+    return tuple(int(v * 255) for v in new_rgb)
+
 class MPGPoster(Base):
     @staticmethod
     def create_color_shades(hex_color: str) -> colors.Color:
-        # Manually define shades for each color
-        # This is an example, adjust the shades according to your needs
-        shades = {
-            "c50": hex_color,
-            "c100": hex_color,
-            "c200": hex_color,  # Slightly lighter or darker
-            "c300": hex_color,
-            "c400": hex_color,
-            "c500": hex_color,  # Base color
-            "c600": hex_color,
-            "c700": hex_color,
-            "c800": hex_color,
-            "c900": hex_color,
-            "c950": hex_color,  # Darkest shade
+        base_rgb = hex_to_rgb(hex_color)
+        shades = {}
+        # Define brightness factors for shades
+        factors = {
+            "c50": 1.2, "c100": 1.1, "c200": 1.05,
+            "c300": 1.025, "c400": 1.0125, "c500": 1.0,
+            "c600": 0.9, "c700": 0.8, "c800": 0.7,
+            "c900": 0.6, "c950": 0.5
         }
+        for shade, factor in factors.items():
+            new_rgb = adjust_brightness(base_rgb, factor)
+            shades[shade] = rgb_to_hex(new_rgb)
         return colors.Color(**shades)
 
     def __init__(