Update civitaiinfo_convert.py

civitaiinfo_convert.py

@@ -4,37 +4,37 @@ 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()
-
     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"] = 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']}"
+        out["sd version"] = sdversion 
+        out["preferred weight"] = 1
+        with open(jsonname, "w") as f:
+            json.dump(out, f, indent=4)
     except:
-        print("ERROR: json load error :", jsonname)
+        print("ERROR :", file)
         continue
-    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"] = 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']}"
-    out["sd version"] = sdversion 
-    out["preferred weight"] = 1
-    with open(jsonname, "w") as f:
-        json.dump(out, f, indent=4)