Update modelos.py

modelos.py

@@ -3,18 +3,14 @@ from ipywidgets import widgets, Layout
 import os
 from nenen88 import download, say, tempe
 
-bura = "/home/studio-lab-user/forge/asd/cn-1_5.css"
-bura2 = "/home/studio-lab-user/asd/asd/cn-1_5.css"
+paths = ["/home/studio-lab-user/forge/asd/cn-1_5.css",
+         "/home/studio-lab-user/asd/asd/cn-1_5.css"]
 
-# Check if the file exists at the first location
-if os.path.exists(bura):
-    file_path = bura
-else:
-    file_path = bura2
-
-with open(file_path, "r") as oppai:
-    susu = oppai.read()
-display(HTML(f"<style>{susu}</style>"))
+file_path = next((path for path in paths if os.path.exists(path)), None)
+if file_path:
+    with open(file_path, "r") as oppai:
+        susu = oppai.read()
+    display(HTML(f"<style>{susu}</style>"))
 
 url_list = {
     "FastPonyMerge_version75 (Pony)": [
@@ -82,92 +78,55 @@ url_list = {
     "SourceOfTheGoddess_003LCM (1.5)": [
         "https://huggingface.co/Carlos2312/models/resolve/main/sourceOfTheGoddess_003LCM.safetensors sourceOfTheGoddess_003LCM.safetensors"]}
     
-list_half = len(url_list) // 2
-half_list_1 = dict(list(url_list.items())[:list_half])
-half_list_2 = dict(list(url_list.items())[list_half:])
-
-cb1 = widgets.VBox(
-    [widgets.Checkbox(value=False, description=name, style={'description_width': '0px'})
-     for name in half_list_1])
-cb1.add_class("checkbox-group1")
+half_list_1 = dict(list(url_list.items())[:len(url_list)//2])
+half_list_2 = dict(list(url_list.items())[len(url_list)//2:])
 
-cb2 = widgets.VBox(
-    [widgets.Checkbox(value=False, description=name, style={'description_width': '0px'})
-     for name in half_list_2])
-cb2.add_class("checkbox-group2")
+def create_checkbox_group(url_dict):
+    return widgets.VBox([widgets.Checkbox(value=False, description=name, style={'description_width': '0px'}) for name in url_dict])
 
-db = widgets.Button(description="Download")
-db.add_class("download-button")
-dbo = widgets.Output()
+cb1 = create_checkbox_group(half_list_1)
+cb2 = create_checkbox_group(half_list_2)
 cbc = widgets.HBox([cb1, cb2], layout=widgets.Layout(align_items='flex-start'))
 
-gariz3 = """<div class="gradient-cn2">forge</div>"""
-garis3 = widgets.Output()
-
-def sa_cb(b):
-    for checkbox in cb1.children + cb2.children:
-        checkbox.value = True
-
-def usa_cb(b):
-    for checkbox in cb1.children + cb2.children:
-        checkbox.value = False
-        
+db = widgets.Button(description="Download")
 sab = widgets.Button(description="Select All")
-sab.add_class("select-all-button")
-sab.on_click(sa_cb)
-
 usab = widgets.Button(description="Unselect All")
-usab.add_class("unselect-all-button")
-usab.on_click(usa_cb)
-
 bs = widgets.Button(description="")
-bs.add_class("border-style")
+
+def toggle_checkboxes(state):
+    for checkbox in cb1.children + cb2.children:
+        checkbox.value = state
+
+sab.on_click(lambda b: toggle_checkboxes(True))
+usab.on_click(lambda b: toggle_checkboxes(False))
 
 bl = widgets.HBox([sab, usab, db, bs])
-boks2 = widgets.VBox([bl, cbc], layout=Layout(
-    display='flex',
-    flex_flow='column',
-    width='630px',
-    height='455px',
-    align_items='center',
-    padding='10px'))
+boks2 = widgets.VBox([bl, cbc], layout=Layout(display='flex', flex_flow='column', width='630px', height='455px', align_items='center', padding='10px'))
 boks2.add_class("boks2")
-        
+
 def d_b_click(b):
-    surl = []
-    for checkbox, key in zip(cb1.children + cb2.children, list(url_list.keys())):
-        if checkbox.value:
-            surl.extend(url_list[key])
-            
+    surl = [url for checkbox, key in zip(cb1.children + cb2.children, url_list.keys()) if checkbox.value for url in url_list[key]]
+    
     widgets.Widget.close(boks2)
     dbo.clear_output()
     
-    with garis3:
-        display(HTML(gariz3))
+    display(HTML("<div class='gradient-cn2'>forge</div>"))
 
-    # Define the two directory paths
-    dir_path_1 = "/home/studio-lab-user/forge/models/Stable-diffusion/tmp_ckpt"
-    dir_path_2 = "/home/studio-lab-user/asd/models/Stable-diffusion/tmp_ckpt"
+    dir_paths = ["/home/studio-lab-user/forge/models/Stable-diffusion/tmp_ckpt",
+                 "/home/studio-lab-user/asd/models/Stable-diffusion/tmp_ckpt"]
+
+    directory_path = next((path for path in dir_paths if os.path.exists(path)), None)
+    
+    if directory_path:
+        with dbo:
+            say("【{red} Downloading{cyan} Models{magenta} Models{yellow} 】{red}")
+            os.chdir(directory_path)
+            for url in surl:
+                download(url)
+            say("【{red} Done{d} 】{red}")
 
-    # Check if the directory exists at the first location
-    if os.path.exists(dir_path_1):
-        directory_path = dir_path_1
-    else:
-        directory_path = dir_path_2
+db.on_click(d_b_click)
 
-    # Change the current directory to the selected directory
-    with dbo:
-        say("【{red} Downloading{cyan} Models{magenta} Models{yellow} 】{red}")
-        os.chdir(directory_path)
-        
-        for url in surl:
-            download(url)
-            
-        with garis3:
-            garis3.clear_output()
-            
-        say("【{red} Done{d} 】{red}")
-            
 tempe()
-display(boks2, dbo, garis3)
-db.on_click(d_b_click)
+display(boks2)
+display(widgets.Output())