InvokeAI 2.2.4.5__py3-none-any.whl

scripts/create_root.py

@@ -0,0 +1,43 @@
+"""
+script to create the necesarry folder structure
+"""
+import shutil
+import os
+# from ldm.invoke.globals import Globals
+
+def main():
+    """
+    main function
+    """
+    osp = os.path
+    dirs_to_copy = 'configs', 'frontend', 'scripts'
+    done = int()
+
+    def copy_from_venv(folder):
+        """
+        function to copy the requested directory tree from the venv
+        """
+        src_folder = osp.abspath(osp.join(os.environ.get('VIRTUAL_ENV'), folder))
+        dest_folder = osp.abspath(osp.join(os.environ.get('VIRTUAL_ENV'), '..', folder))
+        # dest_folder = osp.join(Globals.root, folder)
+        print(f'trying to copy {src_folder} to {dest_folder}')
+        shutil.copytree(src_folder, dest_folder)
+
+    for tree in dirs_to_copy:
+        try:
+            copy_from_venv(folder = tree)
+            done = done + 1
+        except FileExistsError as err:
+            print(err)
+        except FileNotFoundError as err:
+            print(err)
+        except RuntimeError as err:
+            print(err)
+
+    if int(len(dirs_to_copy)) == done:
+        from scripts import configure_invokeai
+        configure_invokeai.main()
+
+
+if __name__ == "__main__":
+    main()

scripts/dream.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+# Copyright (c) 2022 Lincoln D. Stein (https://github.com/lstein)
+
+import warnings
+import ldm.invoke.CLI
+warnings.warn("dream.py is being deprecated, please run invoke.py for the "
+              "new UI/API or legacy_api.py for the old API",
+              DeprecationWarning)
+ldm.invoke.CLI.main()
+

scripts/images2prompt.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+'''This script reads the "Invoke" Stable Diffusion prompt embedded in files generated by invoke.py'''
+
+import sys
+from PIL import Image,PngImagePlugin
+
+if len(sys.argv) < 2:
+    print("Usage: file2prompt.py <file1.png> <file2.png> <file3.png>...")
+    print("This script opens up the indicated invoke.py-generated PNG file(s) and prints out the prompt used to generate them.")
+    exit(-1)
+
+filenames = sys.argv[1:]
+for f in filenames:
+    try:
+        im = Image.open(f)
+        try:
+            prompt = im.text['Dream']
+        except KeyError:
+            prompt = ''
+        print(f'{f}: {prompt}')
+    except FileNotFoundError:
+        sys.stderr.write(f'{f} not found\n')
+        continue
+    except PermissionError:
+        sys.stderr.write(f'{f} could not be opened due to inadequate permissions\n')
+        continue
+
+
+
+

scripts/invoke.py

@@ -0,0 +1,5 @@
+#!/usr/bin/env python
+
+import ldm.invoke.CLI
+ldm.invoke.CLI.main()
+