From 87fc830981db9650299b2128f3c7f342501d0aa2 Mon Sep 17 00:00:00 2001
From: Manuel Schmid <dev@mash1t.de>
Date: Wed, 22 Nov 2023 22:17:24 +0100
Subject: [PATCH] add steps info to performance selection radio buttons

---
 javascript/viewer.js | 2 +-
 modules/flags.py     | 8 +++++++-
 webui.py             | 3 ++-
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/javascript/viewer.js b/javascript/viewer.js
index 3df32ccf..84d38b8e 100644
--- a/javascript/viewer.js
+++ b/javascript/viewer.js
@@ -66,7 +66,7 @@ function on_style_selection_blur() {
 }
 
 onUiLoaded(async () => {
-    let spans = document.querySelectorAll('.aspect_ratios span');
+    let spans = document.querySelectorAll('.aspect_ratios span, .performance_selections span');
 
     spans.forEach(function (span) {
         span.innerHTML = span.innerHTML.replace(/&lt;/g, '<').replace(/&gt;/g, '>');
diff --git a/modules/flags.py b/modules/flags.py
index 2e1c0992..9f53cf72 100644
--- a/modules/flags.py
+++ b/modules/flags.py
@@ -33,7 +33,13 @@ default_parameters = {
 }  # stop, weight
 
 inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6']
-performance_selections = ['Speed', 'Quality', 'Extreme Speed']
+
+
+performance_selections = [
+    (f'Speed <span style="color: grey;"> \U00002223 30 steps</span>', 'Speed'),
+    (f'Quality <span style="color: grey;"> \U00002223 60 steps</span>', 'Quality'),
+    (f'Extreme Speed <span style="color: grey;"> \U00002223 8 steps, LCM</span>', 'Extreme Speed')
+]
 
 inpaint_option_default = 'Inpaint or Outpaint (default)'
 inpaint_option_detail = 'Improve Detail (face, hand, eyes, etc.)'
diff --git a/webui.py b/webui.py
index 0c356fa2..c796e70e 100644
--- a/webui.py
+++ b/webui.py
@@ -214,7 +214,8 @@ with shared.gradio_root:
             with gr.Tab(label='Setting'):
                 performance_selection = gr.Radio(label='Performance',
                                                  choices=modules.flags.performance_selections,
-                                                 value=modules.config.default_performance)
+                                                 value=modules.config.default_performance,
+                                                 elem_classes='performance_selections')
                 aspect_ratios_selection = gr.Radio(label='Aspect Ratios', choices=modules.config.available_aspect_ratios,
                                                    value=modules.config.default_aspect_ratio, info='width × height',
                                                    elem_classes='aspect_ratios')