Updated main.py file and added page_utils.py file to store color schema

main.py

@@ -8,7 +8,7 @@ import torch
 from PIL import Image
 
 from models import phc_models
-from utils import utils
+from utils import utils, page_utils
 
 device = torch.device('cpu')
 if torch.cuda.is_available():
@@ -193,26 +193,16 @@ def predict_bilateral(cc_file, mlo_file):
 
 def run():
     """Run Gradio App."""
-    with open('DESCRIPTION.md', encoding='utf-8') as f:
-        description = f.read()
-
-    with gr.Blocks() as demo:
+    with open('index.html', encoding='utf-8') as f:
+        html_content = f.read()
+
+    with gr.Blocks(theme=gr.themes.Default(primary_hue=page_utils.KALBE_THEME_COLOR, secondary_hue=page_utils.KALBE_THEME_COLOR).set(
+        button_primary_background_fill='*primary_600',
+        button_primary_background_fill_hover='*primary_500',
+        button_primary_text_color='white',
+    )) as demo:
         with gr.Column():
-            gr.Markdown(
-                """
-                <h1 style="text-align: center;">Bilateral View Hypercomplex Breast Classification</h1>
-                """
-            )
-            with gr.Row():
-                gr.Markdown(description)
-                gr.Markdown(
-                    """
-                    ## Model Architecture
-                    <img src="file/figures/phresnet18.png" width=auto>
-
-                    Parameterized Hypercomplex ResNets-18 Variants (PHYResNet).
-                    """
-                )
+            gr.HTML(html_content)
         with gr.Row():
             with gr.Column():
                 cc_file = gr.File(file_count='single',
@@ -220,8 +210,8 @@ def run():
                 mlo_file = gr.File(file_count='single',
                                    file_types=SUPPORTED_IMG_EXT, label='MLO View')
                 with gr.Row():
-                    process_btn = gr.Button('Process')
                     clear_btn = gr.Button('Clear')
+                    process_btn = gr.Button('Process', variant="primary")
             with gr.Column():
                 output_gallery = gr.Gallery(
                     label='Highlighted Area').style(grid=[2], height='auto')

utils/page_utils.py

@@ -0,0 +1,51 @@
+from typing import Optional
+
+
+class ColorPalette:
+    """Color Palette Container."""
+    all = []
+
+    def __init__(
+        self,
+        c50: str,
+        c100: str,
+        c200: str,
+        c300: str,
+        c400: str,
+        c500: str,
+        c600: str,
+        c700: str,
+        c800: str,
+        c900: str,
+        c950: str,
+        name: Optional[str] = None,
+    ):
+        self.c50 = c50
+        self.c100 = c100
+        self.c200 = c200
+        self.c300 = c300
+        self.c400 = c400
+        self.c500 = c500
+        self.c600 = c600
+        self.c700 = c700
+        self.c800 = c800
+        self.c900 = c900
+        self.c950 = c950
+        self.name = name
+        ColorPalette.all.append(self)
+
+
+KALBE_THEME_COLOR = ColorPalette(
+    name='kalbe',
+    c50='#f2f9e8',
+    c100='#dff3c4',
+    c200='#c2e78d',
+    c300='#9fd862',
+    c400='#7fc93f',
+    c500='#3F831C',
+    c600='#31661a',
+    c700='#244c13',
+    c800='#18340c',
+    c900='#0c1b06',
+    c950='#050a02',
+)