Update modelos.py

modelos.py

@@ -3,14 +3,18 @@ from ipywidgets import widgets, Layout
 import os
 from nenen88 import download, say, tempe
 
-paths = ["/home/studio-lab-user/forge/asd/cn-1_5.css",
-         "/home/studio-lab-user/asd/asd/cn-1_5.css"]
+bura = "/home/studio-lab-user/forge/asd/cn-1_5.css"
+bura2 = "/home/studio-lab-user/asd/asd/cn-1_5.css"
 
-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>"))
+# 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>"))
 
 url_list = {
     "FastPonyMerge_version75 (Pony)": [
@@ -78,55 +82,92 @@ url_list = {
     "SourceOfTheGoddess_003LCM (1.5)": [
         "https://huggingface.co/Carlos2312/models/resolve/main/sourceOfTheGoddess_003LCM.safetensors sourceOfTheGoddess_003LCM.safetensors"]}
     
-half_list_1 = dict(list(url_list.items())[:len(url_list)//2])
-half_list_2 = dict(list(url_list.items())[len(url_list)//2:])
+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:])
 
-def create_checkbox_group(url_dict):
-    return widgets.VBox([widgets.Checkbox(value=False, description=name, style={'description_width': '0px'}) for name in url_dict])
+cb1 = widgets.VBox(
+    [widgets.Checkbox(value=False, description=name, style={'description_width': '0px'})
+     for name in half_list_1])
+cb1.add_class("checkbox-group1")
 
-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'))
+cb2 = widgets.VBox(
+    [widgets.Checkbox(value=False, description=name, style={'description_width': '0px'})
+     for name in half_list_2])
+cb2.add_class("checkbox-group2")
 
 db = widgets.Button(description="Download")
-sab = widgets.Button(description="Select All")
-usab = widgets.Button(description="Unselect All")
-bs = widgets.Button(description="")
+db.add_class("download-button")
+dbo = widgets.Output()
+cbc = widgets.HBox([cb1, cb2], layout=widgets.Layout(align_items='flex-start'))
+
+gariz3 = """<div class="gradient-cn2">forge</div>"""
+garis3 = widgets.Output()
 
-def toggle_checkboxes(state):
+def sa_cb(b):
     for checkbox in cb1.children + cb2.children:
-        checkbox.value = state
+        checkbox.value = True
 
-sab.on_click(lambda b: toggle_checkboxes(True))
-usab.on_click(lambda b: toggle_checkboxes(False))
+def usa_cb(b):
+    for checkbox in cb1.children + cb2.children:
+        checkbox.value = False
+        
+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")
 
 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 = [url for checkbox, key in zip(cb1.children + cb2.children, url_list.keys()) if checkbox.value for url in url_list[key]]
-    
+    surl = []
+    for checkbox, key in zip(cb1.children + cb2.children, list(url_list.keys())):
+        if checkbox.value:
+            surl.extend(url_list[key])
+            
     widgets.Widget.close(boks2)
     dbo.clear_output()
     
-    display(HTML("<div class='gradient-cn2'>forge</div>"))
+    with garis3:
+        display(HTML(gariz3))
 
-    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}")
+    # 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"
 
-db.on_click(d_b_click)
+    # 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
 
+    # 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)
-display(widgets.Output())
+display(boks2, dbo, garis3)
+db.on_click(d_b_click)