Merge pull request #8945 from space-nuko/gradio-theme-support

Support Gradio's theme API

2 files changed, 36 insertions, 1 deletions

diff --git a/modules/shared.py b/modules/shared.py
index 36f4ed5e..8b5f752e 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -4,6 +4,7 @@ import json
 import os

 import sys

 import time

+import requests

 

 from PIL import Image

 import gradio as gr

@@ -55,6 +56,21 @@ ui_reorder_categories = [
     "scripts",

 ]

 

+# https://huggingface.co/datasets/freddyaboulton/gradio-theme-subdomains/resolve/main/subdomains.json

+gradio_hf_hub_themes = [

+    "gradio/glass",

+    "gradio/monochrome",

+    "gradio/seafoam",

+    "gradio/soft",

+    "freddyaboulton/dracula_revamped",

+    "gradio/dracula_test",

+    "abidlabs/dracula_test",

+    "abidlabs/pakistan",

+    "dawood/microsoft_windows",

+    "ysharma/steampunk"

+]

+

+

 cmd_opts.disable_extension_access = (cmd_opts.share or cmd_opts.listen or cmd_opts.server_name) and not cmd_opts.enable_insecure_extension_access

 

 devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_esrgan, devices.device_codeformer = \

@@ -394,6 +410,7 @@ options_templates.update(options_section(('ui', "User interface"), {
     "ui_reorder": OptionInfo(", ".join(ui_reorder_categories), "txt2img/img2img UI item order"),

     "ui_extra_networks_tab_reorder": OptionInfo("", "Extra networks tab order"),

     "localization": OptionInfo("None", "Localization (requires restart)", gr.Dropdown, lambda: {"choices": ["None"] + list(localization.localizations.keys())}, refresh=lambda: localization.list_localizations(cmd_opts.localizations_dir)),

+    "gradio_theme": OptionInfo("Default", "Gradio theme (requires restart)", gr.Dropdown, lambda: {"choices": ["Default"] + gradio_hf_hub_themes})

 }))

 

 options_templates.update(options_section(('ui', "Live previews"), {

@@ -607,6 +624,24 @@ clip_model = None
 

 progress_print_out = sys.stdout

 

+gradio_theme = gr.themes.Base()

+

+

+def reload_gradio_theme(theme_name=None):

+    global gradio_theme

+    if not theme_name:

+        theme_name = opts.gradio_theme

+

+    if theme_name == "Default":

+        gradio_theme = gr.themes.Default()

+    else:

+        try:

+            gradio_theme = gr.themes.ThemeClass.from_hub(theme_name)

+        except requests.exceptions.ConnectionError:

+            print("Can't access HuggingFace Hub, falling back to default Gradio theme")

+            gradio_theme = gr.themes.Default()

+

+

 

 class TotalTQDM:

     def __init__(self):

diff --git a/modules/ui.py b/modules/ui.py
index 7370d595..dab94b93 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1565,7 +1565,7 @@ def create_ui():
     for _interface, label, _ifid in interfaces:

         shared.tab_names.append(label)

 

-    with gr.Blocks(analytics_enabled=False, title="Stable Diffusion") as demo:

+    with gr.Blocks(theme=shared.gradio_theme, analytics_enabled=False, title="Stable Diffusion") as demo:

         with gr.Row(elem_id="quicksettings", variant="compact"):

             for i, k, item in sorted(quicksettings_list, key=lambda x: quicksettings_names.get(x[1], x[0])):

                 component = create_setting_component(k, is_quicksettings=True)