Compare commits
3
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
67de22d772 | ||
|
|
86e67ac11c | ||
|
|
6d90f39412 |
@@ -1,5 +1,3 @@
|
||||
[](https://github.com/theUpsider/ComfyUI-Styles_CSV_Loader/actions/workflows/publish.yml)
|
||||
[](https://github.com/theUpsider/ComfyUI-Styles_CSV_Loader/actions/workflows/test.yml)
|
||||
# Styles CSV Loader Extension for ComfyUI
|
||||
Extension for [ComfyUI](https://github.com/comfyanonymous/ComfyUI) that loads styles from a CSV file.
|
||||
|
||||
@@ -15,15 +13,6 @@ This extension allows users to load styles from a CSV file (styles.csv), primari
|
||||
## Nodes Description
|
||||
Each style is represented as a dictionary with the keys being `style_name` and the values being a list containing `positive_prompt` and `negative_prompt`. The prompts are outputs of this Node.
|
||||
|
||||
### Custom CSV Files
|
||||
The extension now supports loading styles from custom CSV files in addition to the default `styles.csv`. You can specify a custom CSV file path using the optional `csv_file_path` parameter:
|
||||
|
||||
- **Relative paths**: Relative to the ComfyUI root directory (e.g., `"my_styles/portrait_styles.csv"`)
|
||||
- **Absolute paths**: Full file system paths (e.g., `"/path/to/my/custom_styles.csv"`)
|
||||
- **Default behavior**: When no custom path is specified, it uses `styles.csv` from the ComfyUI root directory
|
||||
|
||||
If a selected style is not found in the custom CSV file, the node will fall back to the default styles loaded from `styles.csv`.
|
||||
|
||||
## CSV Format
|
||||
The CSV file should have the following format:
|
||||
```csv
|
||||
@@ -32,23 +21,6 @@ name,prompt,negative_prompt
|
||||
"Another Style","more positive text","more negative text"
|
||||
```
|
||||
|
||||
### Examples
|
||||
You can organize your styles in multiple CSV files for different purposes:
|
||||
|
||||
**portraits.csv**:
|
||||
```csv
|
||||
name,prompt,negative_prompt
|
||||
"Professional Portrait","portrait photography, professional lighting, sharp focus","ugly, deformed, extra limbs"
|
||||
"Vintage Portrait","vintage portrait, sepia tone, classical","modern, digital, colorful"
|
||||
```
|
||||
|
||||
**landscapes.csv**:
|
||||
```csv
|
||||
name,prompt,negative_prompt
|
||||
"Golden Hour","landscape photography, golden hour, wide angle","people, buildings, urban"
|
||||
"Mountain Vista","mountain landscape, dramatic clouds, epic view","flat, boring, low quality"
|
||||
```
|
||||
|
||||
## Development
|
||||
|
||||
### Setting up the development environment
|
||||
@@ -88,4 +60,4 @@ The project maintains 100% test coverage. Coverage reports are generated during
|
||||
## Author
|
||||
- David Fischer
|
||||
- GitHub: [theUpsider](https://github.com/theUpsider)
|
||||
- Support me on [BuyMeACoffee](https://www.buymeacoffee.com/theupsider)
|
||||
- Support me on [BuyMeACoffee](https://www.buymeacoffee.com/theupsider)
|
||||
+2
-1
@@ -5,4 +5,5 @@ import os
|
||||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
if current_dir not in sys.path:
|
||||
sys.path.insert(0, current_dir)
|
||||
from styles_csv_loader import *
|
||||
|
||||
from styles_csv_loader import StylesCSVLoader
|
||||
+1
-1
@@ -1,7 +1,7 @@
|
||||
[project]
|
||||
name = "comfyui-styles_csv_loader"
|
||||
description = "This extension allows users to load styles from a CSV file, primarily for migration purposes from the automatic1111 Stable Diffusion web UI."
|
||||
version = "3.0.0"
|
||||
version = "1.1.1"
|
||||
license = { text = "MIT License" }
|
||||
|
||||
[project.optional-dependencies]
|
||||
|
||||
+4
-24
@@ -45,15 +45,13 @@ class StylesCSVLoader:
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
# Use os.path.normpath to ensure cross-platform compatibility
|
||||
default_styles_path = os.path.normpath(os.path.join(folder_paths.base_path, "styles.csv"))
|
||||
cls.styles_csv = cls.load_styles_csv(default_styles_path)
|
||||
styles_path = os.path.normpath(os.path.join(folder_paths.base_path, "styles.csv"))
|
||||
cls.styles_csv = cls.load_styles_csv(styles_path)
|
||||
return {
|
||||
"required": {
|
||||
"styles": (list(cls.styles_csv.keys()),),
|
||||
},
|
||||
"optional": {
|
||||
"csv_file_path": ("STRING", {"default": "styles.csv", "multiline": False}),
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("STRING", "STRING")
|
||||
@@ -61,25 +59,7 @@ class StylesCSVLoader:
|
||||
FUNCTION = "execute"
|
||||
CATEGORY = "loaders"
|
||||
|
||||
def execute(self, styles, csv_file_path="styles.csv"):
|
||||
# Handle custom CSV file path
|
||||
if csv_file_path != "styles.csv":
|
||||
# Check if path is absolute or relative
|
||||
if os.path.isabs(csv_file_path):
|
||||
custom_styles_path = os.path.normpath(csv_file_path)
|
||||
else:
|
||||
# Relative path from ComfyUI root directory
|
||||
custom_styles_path = os.path.normpath(os.path.join(folder_paths.base_path, csv_file_path))
|
||||
|
||||
# Load styles from custom path
|
||||
custom_styles = self.load_styles_csv(custom_styles_path)
|
||||
if styles in custom_styles:
|
||||
return (custom_styles[styles][0], custom_styles[styles][1])
|
||||
else:
|
||||
# Fall back to default styles if style not found in custom file
|
||||
pass
|
||||
|
||||
# Use default styles
|
||||
def execute(self, styles):
|
||||
return (self.styles_csv[styles][0], self.styles_csv[styles][1])
|
||||
|
||||
|
||||
|
||||
@@ -152,82 +152,6 @@ class TestStylesCSVLoader(unittest.TestCase):
|
||||
|
||||
self.assertEqual(positive, 'positive prompt here')
|
||||
self.assertEqual(negative, 'negative prompt here')
|
||||
|
||||
def test_execute_method_with_custom_csv_path(self):
|
||||
"""Test the execute method with custom CSV file path."""
|
||||
loader = StylesCSVLoader()
|
||||
|
||||
# Mock the default styles_csv attribute
|
||||
loader.styles_csv = {
|
||||
'Default Style': ['default positive', 'default negative']
|
||||
}
|
||||
|
||||
# Test with custom CSV path
|
||||
positive, negative = loader.execute('Cinematic', self.valid_csv_path)
|
||||
|
||||
self.assertEqual(positive, 'cinematic shot, dramatic lighting, film grain')
|
||||
self.assertEqual(negative, 'low quality, blurry, amateur')
|
||||
|
||||
def test_execute_method_with_custom_csv_fallback(self):
|
||||
"""Test that execute method falls back to default when style not found in custom CSV."""
|
||||
loader = StylesCSVLoader()
|
||||
|
||||
# Mock the default styles_csv attribute
|
||||
loader.styles_csv = {
|
||||
'Default Style': ['default positive', 'default negative']
|
||||
}
|
||||
|
||||
# Test with custom CSV path but style that doesn't exist in custom file
|
||||
positive, negative = loader.execute('Default Style', self.valid_csv_path)
|
||||
|
||||
self.assertEqual(positive, 'default positive')
|
||||
self.assertEqual(negative, 'default negative')
|
||||
|
||||
def test_execute_method_with_absolute_path(self):
|
||||
"""Test execute method with absolute path to custom CSV."""
|
||||
loader = StylesCSVLoader()
|
||||
|
||||
# Mock the default styles_csv attribute
|
||||
loader.styles_csv = {
|
||||
'Default Style': ['default positive', 'default negative']
|
||||
}
|
||||
|
||||
# Test with absolute path
|
||||
absolute_path = os.path.abspath(self.valid_csv_path)
|
||||
positive, negative = loader.execute('Cinematic', absolute_path)
|
||||
|
||||
self.assertEqual(positive, 'cinematic shot, dramatic lighting, film grain')
|
||||
self.assertEqual(negative, 'low quality, blurry, amateur')
|
||||
|
||||
@patch('styles_csv_loader.folder_paths')
|
||||
def test_input_types_includes_csv_file_path(self, mock_folder_paths):
|
||||
"""Test that INPUT_TYPES includes the csv_file_path optional parameter."""
|
||||
mock_folder_paths.base_path = self.fixtures_dir
|
||||
|
||||
# Create a temporary styles.csv in the fixtures directory
|
||||
temp_csv_path = os.path.join(self.fixtures_dir, 'styles.csv')
|
||||
with open(temp_csv_path, 'w', encoding='utf-8') as f:
|
||||
f.write('name,prompt,negative_prompt\n')
|
||||
f.write('"Test Style","test prompt","test negative"\n')
|
||||
|
||||
try:
|
||||
input_types = StylesCSVLoader.INPUT_TYPES()
|
||||
|
||||
# Check structure includes optional csv_file_path
|
||||
self.assertIn('required', input_types)
|
||||
self.assertIn('optional', input_types)
|
||||
self.assertIn('csv_file_path', input_types['optional'])
|
||||
|
||||
# Check csv_file_path is a STRING type with default
|
||||
csv_param = input_types['optional']['csv_file_path']
|
||||
self.assertEqual(csv_param[0], 'STRING')
|
||||
self.assertIn('default', csv_param[1])
|
||||
self.assertEqual(csv_param[1]['default'], 'styles.csv')
|
||||
|
||||
finally:
|
||||
# Clean up
|
||||
if os.path.exists(temp_csv_path):
|
||||
os.remove(temp_csv_path)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
Reference in New Issue
Block a user