Update multi.py

multi.py

@@ -1,12 +1,12 @@
-from pathlib import Path
 from IPython.display import display, HTML, clear_output
 from IPython import get_ipython
 from ipywidgets import widgets
+from pathlib import Path
 import os
 
 home = Path.home()
 src = home / '.gutris1'
-css = src / 'multi.css'
+css_setup = src / 'setup.css'
 mark = src / 'marking.py'
 img = src / 'loading.png'
 
@@ -14,60 +14,57 @@ A1111 = src / 'A1111.py'
 Forge = src / 'Forge.py'
 ComfyUI = src / 'ComfyUI.py'
 
-os.chdir(home)
-src.mkdir(parents=True, exist_ok=True)
+def load_css():
+    with open(css_setup, "r") as file:
+        data = file.read()
 
-x = [
-    f"curl -sLo {css} https://github.com/gutris1/segsmaker/raw/main/script/multi/multi.css",
-    f"curl -sLo {img} https://github.com/gutris1/segsmaker/raw/main/script/loading.png",
-    f"curl -sLo {mark} https://github.com/gutris1/segsmaker/raw/main/script/multi/marking.py",
-    f"curl -sLo {A1111} https://github.com/gutris1/segsmaker/raw/main/script/multi/A1111.py",
-    f"curl -sLo {Forge} https://huggingface.co/Carlos2312/asd/resolve/main/Forge.py",
-    f"curl -sLo {ComfyUI} https://github.com/gutris1/segsmaker/raw/main/script/multi/ComfyUI.py"]
-    
-for y in x:
-    get_ipython().system(y)
+    display(HTML(f"<style>{data}</style>"))
 
-def dupe_button(desc):
-    button = widgets.Button(description=desc)
-    button.add_class("buttons")
-    return button
+def selection(btn):
+    multi_panel.close()
 
-output = widgets.Output()
-buttons = [dupe_button(desc) for desc in ['A1111', 'Forge', 'ComfyUI']]
-selection_panel = widgets.HBox(buttons, layout=widgets.Layout(
-    width='500px',
-    height='100%',
-    display='flex',
-    flex_flow='row',
-    align_items='center',
-    justify_content='space-between',
-    padding='20px'))
+    with output:
+        if btn == 'A1111':
+            get_ipython().run_line_magic('run', f'{A1111}')
+        elif btn == 'Forge':
+            get_ipython().run_line_magic('run', f'{Forge}')
+        elif btn == 'ComfyUI':
+            get_ipython().run_line_magic('run', f'{ComfyUI}')
 
-selection_panel.add_class("main-panel")
+options = ['A1111', 'Forge', 'ComfyUI']
+buttons = []
 
-def load_css(css):
-    with css.open("r") as file:
-        data = file.read()
+for btn in options:
+    button = widgets.Button(description='')
+    button.add_class(btn.lower())
+    button.on_click(lambda x, btn=btn: selection(btn))
+    buttons.append(button)
 
-    display(HTML(f"<style>{data}</style>"))
+output = widgets.Output()
 
-def selection(b):
-    selection_panel.close()
-    clear_output()
+multi_panel = widgets.HBox(
+    buttons, layout=widgets.Layout(
+        width='600px',
+        height='405px'))
+multi_panel.add_class('multi-panel')
 
-    with output:
-        if b.description == 'A1111':
-            get_ipython().magic(f'run {A1111}')
+def multi_widgets():
+    if not src.exists():
+        src.mkdir(parents=True, exist_ok=True)
 
-        elif b.description == 'Forge':
-            get_ipython().magic(f'run {Forge}')
+    x = [
+        f"curl -sLo {css_setup} https://github.com/gutris1/segsmaker/raw/main/script/multi/setup.css",
+        f"curl -sLo {img} https://github.com/gutris1/segsmaker/raw/main/script/loading.png",
+        f"curl -sLo {mark} https://github.com/gutris1/segsmaker/raw/main/script/multi/marking.py",
+        f"curl -sLo {A1111} https://github.com/gutris1/segsmaker/raw/main/script/multi/A1111.py",
+        f"curl -sLo {Forge} https://huggingface.co/Carlos2312/asd/resolve/main/Forge.py",
+        f"curl -sLo {ComfyUI} https://github.com/gutris1/segsmaker/raw/main/script/multi/ComfyUI.py"]
 
-        elif b.description == 'ComfyUI':
-            get_ipython().magic(f'run {ComfyUI}')
+    for y in x:
+        get_ipython().system(y)
 
-load_css(css)
-display(selection_panel, output)
+    load_css()
+    display(multi_panel, output)
+    os.chdir(home)
 
-for button in buttons:
-    button.on_click(selection)
+multi_widgets()