mirror of
https://github.com/sd-webui/stable-diffusion-webui.git
synced 2024-12-13 18:02:31 +03:00
104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
# This file is part of sygil-webui (https://github.com/Sygil-Dev/sandbox-webui/).
|
|
|
|
# Copyright 2022 Sygil-Dev team.
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
# base webui import and utils.
|
|
# from sd_utils import *
|
|
from sd_utils import st
|
|
|
|
# streamlit imports
|
|
|
|
# streamlit components section
|
|
|
|
# other imports
|
|
from barfi import st_barfi, barfi_schemas, Block
|
|
|
|
# Temp imports
|
|
|
|
# end of imports
|
|
# ---------------------------------------------------------------------------------------------------------------
|
|
|
|
|
|
def layout():
|
|
# st.info("Under Construction. :construction_worker:")
|
|
|
|
# from barfi import st_barfi, Block
|
|
|
|
# add = Block(name='Addition')
|
|
# sub = Block(name='Subtraction')
|
|
# mul = Block(name='Multiplication')
|
|
# div = Block(name='Division')
|
|
|
|
# barfi_result = st_barfi(base_blocks= [add, sub, mul, div])
|
|
# or if you want to use a category to organise them in the frontend sub-menu
|
|
# barfi_result = st_barfi(base_blocks= {'Op 1': [add, sub], 'Op 2': [mul, div]})
|
|
|
|
col1, col2, col3 = st.columns([1, 8, 1])
|
|
|
|
with col2:
|
|
feed = Block(name="Feed")
|
|
feed.add_output()
|
|
|
|
def feed_func(self):
|
|
self.set_interface(name="Output 1", value=4)
|
|
|
|
feed.add_compute(feed_func)
|
|
|
|
splitter = Block(name="Splitter")
|
|
splitter.add_input()
|
|
splitter.add_output()
|
|
splitter.add_output()
|
|
|
|
def splitter_func(self):
|
|
in_1 = self.get_interface(name="Input 1")
|
|
value = in_1 / 2
|
|
self.set_interface(name="Output 1", value=value)
|
|
self.set_interface(name="Output 2", value=value)
|
|
|
|
splitter.add_compute(splitter_func)
|
|
|
|
mixer = Block(name="Mixer")
|
|
mixer.add_input()
|
|
mixer.add_input()
|
|
mixer.add_output()
|
|
|
|
def mixer_func(self):
|
|
in_1 = self.get_interface(name="Input 1")
|
|
in_2 = self.get_interface(name="Input 2")
|
|
value = in_1 + in_2
|
|
self.set_interface(name="Output 1", value=value)
|
|
|
|
mixer.add_compute(mixer_func)
|
|
|
|
result = Block(name="Result")
|
|
result.add_input()
|
|
|
|
def result_func(self):
|
|
self.get_interface(name="Input 1")
|
|
|
|
result.add_compute(result_func)
|
|
|
|
load_schema = st.selectbox("Select a saved schema:", barfi_schemas())
|
|
|
|
compute_engine = st.checkbox("Activate barfi compute engine", value=False)
|
|
|
|
barfi_result = st_barfi(
|
|
base_blocks=[feed, result, mixer, splitter],
|
|
compute_engine=compute_engine,
|
|
load_schema=load_schema,
|
|
)
|
|
|
|
if barfi_result:
|
|
st.write(barfi_result)
|