minor changes..

special/dl_display_results.py

@@ -117,9 +117,6 @@ hr {
     transition: all 0.15s ease-in-out;
 }
 
-/* setting the height of "outputs" containers */
-.outputs_dl { max-height: 400px; }
-
 /* good use of space */
 .extension {
   display: grid;
@@ -166,45 +163,52 @@ display(HTML(CSS))
 
 
 # ==================== WIDGETS ====================
-# --- global widgets ---
 HR = widgets.HTML('<hr>')
 
 header_widget = widgets.HTML(value='''
-<div class="header_dl">DOWNLOAD RESULTS <span style="color: grey; opacity: 0.25;">| v0.2.8</span></div>
+<div><span class="header_dl">DOWNLOAD RESULTS</span> <span style="color: grey; opacity: 0.25;">| v0.3</span></div>
 ''')
 
-# Adding content to "outputs_widgets_list"
+## Functions
 def output_container_generator(header, items):
     header_widget = widgets.HTML(value=f'<div class="header_outputs_dl">{header} ➤</div>')
     content = [widgets.HTML(value=f'<div class="items_dl">{item}</div>') for item in items]
     container_widget = widgets.VBox([header_widget] + content).add_class("outputs_dl")
     return container_widget
 
+def get_files_list(directory, extensions, exclude_dirs=()):
+    return [file for file in os.listdir(directory) if file.endswith(extensions)]
+
+def get_folders_list(directory):
+    return [folder for folder in os.listdir(directory)]
+
+def get_controlnets_list(directory, filter_pattern):
+    filter_name = re.compile(filter_pattern)
+    return [
+        filter_name.match(file).group(1) if filter_name.match(file) else file
+        for file in os.listdir(directory)
+        if not file.endswith(('.txt', '.yaml'))
+    ]
+
+
 # Models
-models_list = [file for file in os.listdir(models_dir) if file.endswith('.safetensors')]
-models_widget = output_container_generator('Models', models_list).add_class("model")
+models_list = get_files_list(models_dir, '.safetensors')
+models_widget = output_container_generator('Models', models_list)
 # Vaes
-vaes_list = [file for file in os.listdir(vaes_dir) if file.endswith('.safetensors')]
-vaes_widget = output_container_generator('VAEs', vaes_list).add_class("vae")
+vaes_list = get_files_list(vaes_dir, '.safetensors')
+vaes_widget = output_container_generator('VAEs', vaes_list)
 # Embeddings
-embeddings_list = [file for file in os.listdir(embeddings_dir) if file.endswith(('.safetensors', '.pt')) and not os.path.isdir(os.path.join(embeddings_dir, file))]
-embeddings_widget = output_container_generator('Embeddings', embeddings_list).add_class("embedding")
+embeddings_list = get_files_list(embeddings_dir, ('.safetensors', '.pt'))
+embeddings_widget = output_container_generator('Embeddings', embeddings_list)
 # LoRAs
-# loras_list = [file for file in os.listdir(loras_dir) if not file.endswith('.txt')]
-loras_list = [file for file in os.listdir(loras_dir) if file.endswith('.safetensors')]
-loras_widget = output_container_generator('LoRAs', loras_list).add_class("lora")
+loras_list = get_files_list(loras_dir, '.safetensors')
+loras_widget = output_container_generator('LoRAs', loras_list)
 # Extensions
-extensions_list = [folder for folder in os.listdir(extensions_dir) if os.path.isdir(os.path.join(extensions_dir, folder))]
-extensions_widget = output_container_generator('Extensions', extensions_list).add_class("extension")
+extensions_list = get_folders_list(extensions_dir)
+extensions_widget = output_container_generator('Extensions', extensions_list).add_class("extension") # for fix height
 # ControlNet
-# controlnets_list = [file for file in os.listdir(control_dir) if not file.endswith(('.txt', '.yaml'))]
-filter_name = re.compile(r'^[^_]*_[^_]*_[^_]*_(.*)_fp16\.safetensors$')
-controlnets_list = [
-    filter_name.match(file).group(1) if filter_name.match(file) else file
-    for file in os.listdir(control_dir)
-    if not file.endswith(('.txt', '.yaml'))
-]
-controlnets_widget = output_container_generator('ControlNets', controlnets_list).add_class("controlnet")
+controlnets_list = get_controlnets_list(control_dir, r'^[^_]*_[^_]*_[^_]*_(.*)_fp16\.safetensors')
+controlnets_widget = output_container_generator('ControlNets', controlnets_list)
 
 
 ## Sorting and Output
@@ -220,5 +224,4 @@ outputs_widgets_list = [widget for widget, widget_list in widgets_dict.items() i
 result_output_widget = widgets.HBox(outputs_widgets_list).add_class("result_output_dl")
 
 container_widget = widgets.VBox([header_widget, HR, result_output_widget, HR]).add_class("container_dl")
-
 display(container_widget)