| import re |
| import json |
| import glob |
| import os |
|
|
| for file in glob.glob("**/*.civitai.info", recursive=True): |
| try: |
| out = {} |
| jsonname = re.sub(r"\.civitai\.info$", ".json", file) |
| textname = re.sub(r"\.civitai\.info$", ".txt", file) |
| if os.path.isfile(jsonname): |
| continue |
| description = "" |
| if os.path.isfile(textname): |
| with open(textname) as f: |
| description = f.read() |
| f.close() |
|
|
| with open(file) as f: |
| info = json.load(f) |
| print("Processing :",jsonname) |
| sdversion = "Unknown" |
| if 'SD 1' in info['baseModel']: |
| sdversion = "SD1" |
| if 'SD 2' in info['baseModel']: |
| sdversion = "SD2" |
| if 'SDXL' in info['baseModel']: |
| sdversion = "SDXL" |
| |
| |
| out["description"] = description |
| out["activation text"] = ", ".join(info["trainedWords"]) |
| out["notes"] = f"https://civitai.com/models/{info['modelId']}?modelVersionId={info['id']}" |
| out["sd version"] = sdversion |
| out["preferred weight"] = 1 |
| with open(jsonname, "w") as f: |
| json.dump(out, f, indent=4) |
| except: |
| print("ERROR :", file) |
| continue |
|
|
