Upload civitaiinfo_convert.py

civitaiinfo_convert.py

@@ -0,0 +1,30 @@
+import re
+import json
+import glob
+import os
+
+for file in glob.glob("**/*.civitai.info", recursive=True):
+    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()
+
+    print(jsonname)
+    with open(file) as f:
+        info = json.load(f)
+        
+#    out["description"] = info.get("description", "")
+#    if not out["description"]:
+    out["description"] = description
+    out["activation text"] = ", ".join(info["trainedWords"])
+    out["notes"] = f"https://civitai.com/models/{info['modelId']}?modelVersionId={info['id']}"
+    
+    print(jsonname)
+    with open(jsonname, "w") as f:
+        json.dump(out, f, indent=4)