Update html

app.py

@@ -5,6 +5,7 @@ from typing import List, Dict, Optional, Union
 import os
 import yaml
 
+from modules.html_constants import (HEADER, DEFAULT_THEME, CSS)
 from modules.sam_inference import SamInference
 from modules.model_downloader import DEFAULT_MODEL_TYPE
 from modules.paths import (OUTPUT_DIR, OUTPUT_PSD_DIR, SAM2_CONFIGS_DIR, TEMP_DIR, OUTPUT_FILTER_DIR, MODELS_DIR)
@@ -17,8 +18,11 @@ from modules.video_utils import get_frames_from_dir
 class App:
     def __init__(self,
                  args: argparse.Namespace):
-        self.demo = gr.Blocks()
         self.args = args
+        self.demo = gr.Blocks(
+            theme=self.args.theme,
+            css=CSS
+        )
         self.sam_inf = SamInference(
             model_dir=self.args.model_dir,
             output_dir=self.args.output_dir
@@ -96,6 +100,8 @@ class App:
         _mask_hparams = self.default_hparams["mask_hparams"]
 
         with self.demo:
+            md_header = gr.Markdown(HEADER, elem_id="md_header")
+
             with gr.Tabs():
                 with gr.TabItem("Layer Divider"):
                     with gr.Row():
@@ -210,6 +216,7 @@ if __name__ == "__main__":
                         help='Whether to automatically start Gradio app or not')
     parser.add_argument('--share', type=bool, default=False, nargs='?', const=True,
                         help='Whether to create a public link for the app or not')
+    parser.add_argument('--theme', type=str, default=DEFAULT_THEME, help='Gradio Blocks theme')
     args = parser.parse_args()
 
     demo = App(args=args)

modules/html_constants.py

@@ -1,5 +1,25 @@
-DEFAULT_THEME = 'finlaymacklon/sooth_slatem'
+DEFAULT_THEME = None
+
+HEADER = """
+## [sam2-playground](https://github.com/jhj0517/sam2-playground)
+
+<div>
+    <a href="">
+        <img src="https://img.shields.io/badge/GitHub-Wiki-blue?logo=github" alt="Github Wiki" style="display:inline-block;">
+    </a>
+    <a href="">
+        <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Colab" style="display:inline-block;">
+    </a>
+</div>
 
-GITHUB_LINK = """
-## [sam2-playground](https://github.com/jhj0517/Whsiper-WebUI)
 """
+
+CSS = """
+#md_header a {
+  color: black;
+  text-decoration: none;
+}
+#md_header a:hover {
+  text-decoration: underline;
+}
+"""