mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-25 23:25:43 -03:00
feat: enhance Civitai resource extraction in StandardMetadataParser for improved JSON handling. Fixes https://github.com/willmiao/ComfyUI-Lora-Manager/issues/141
This commit is contained in:
@@ -403,9 +403,25 @@ class StandardMetadataParser(RecipeMetadataParser):
|
|||||||
|
|
||||||
# Extract Civitai resources
|
# Extract Civitai resources
|
||||||
if 'Civitai resources:' in user_comment:
|
if 'Civitai resources:' in user_comment:
|
||||||
resources_part = user_comment.split('Civitai resources:', 1)[1]
|
resources_part = user_comment.split('Civitai resources:', 1)[1].strip()
|
||||||
if '],' in resources_part:
|
|
||||||
resources_json = resources_part.split('],', 1)[0] + ']'
|
# Look for the opening and closing brackets to extract the JSON array
|
||||||
|
if resources_part.startswith('['):
|
||||||
|
# Find the position of the closing bracket
|
||||||
|
bracket_count = 0
|
||||||
|
end_pos = -1
|
||||||
|
|
||||||
|
for i, char in enumerate(resources_part):
|
||||||
|
if char == '[':
|
||||||
|
bracket_count += 1
|
||||||
|
elif char == ']':
|
||||||
|
bracket_count -= 1
|
||||||
|
if bracket_count == 0:
|
||||||
|
end_pos = i
|
||||||
|
break
|
||||||
|
|
||||||
|
if end_pos != -1:
|
||||||
|
resources_json = resources_part[:end_pos+1]
|
||||||
try:
|
try:
|
||||||
resources = json.loads(resources_json)
|
resources = json.loads(resources_json)
|
||||||
# Filter loras and checkpoints
|
# Filter loras and checkpoints
|
||||||
|
|||||||
Reference in New Issue
Block a user