my-sd/modules_forge/gradio_compile.py
2024-01-25 22:31:43 -08:00

25 lines
905 B
Python

def gradio_compile(items, prefix):
for k, v in items["required"].items():
t = v[0]
d = v[1] if len(v) > 1 else None
name = (prefix + '_' + k).replace(' ', '_').lower()
title = name.replace('_', ' ').capitalize()
if t == 'INT':
default = int(d['default'])
min = int(d['min'])
max = int(d['max'])
step = int(d.get('step', 1))
print(f'{name} = gr.Slider(label=\'{title}\', minimum={min}, maximum={max}, step={step}, value={default})')
elif t == 'FLOAT':
default = float(d['default'])
min = float(d['min'])
max = float(d['max'])
step = float(d.get('step', 0.001))
print(f'{name} = gr.Slider(label=\'{title}\', minimum={min}, maximum={max}, step={step}, value={default})')
else:
print('error ' + str(t))
return