Commit fd71b761 authored by Haoming's avatar Haoming

use re instead of hardcoding

Now supports all natively provided upscaler as well
parent d18eb10e
...@@ -5,6 +5,9 @@ from modules.ui_components import ResizeHandleRow ...@@ -5,6 +5,9 @@ from modules.ui_components import ResizeHandleRow
def hook_scale_update(inputs): def hook_scale_update(inputs):
import re
pattern = r'(\d)[xX]|[xX](\d)'
resize = upscaler = None resize = upscaler = None
for script in inputs: for script in inputs:
if script.label == "Resize": if script.label == "Resize":
...@@ -15,13 +18,16 @@ def hook_scale_update(inputs): ...@@ -15,13 +18,16 @@ def hook_scale_update(inputs):
break break
def update_scale(upscaler: str, slider: float): def update_scale(upscaler: str, slider: float):
if upscaler[1] in ('x', 'X'): match = re.search(pattern, upscaler)
try:
scale = int(upscaler[0]) if match:
return gr.update(value=scale) if match.group(1):
except ValueError: return gr.update(value=int(match.group(1)))
return gr.update(value=slider)
else:
return gr.update(value=int(match.group(2)))
else:
return gr.update(value=slider) return gr.update(value=slider)
if resize and upscaler: if resize and upscaler:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment