| | 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 |
| |
|
