2.0

Movign to latest og, and full transfer to httpx.

Author: ItsCEED

app.py

@@ -1,27 +1,33 @@
-from imaginepy import Imagine, Style, Ratio
+from imaginepy import Imagine
+from imaginepy.constants import *
 
 
 def main():
-    imagine = Imagine(style=Style.ANIME_V2) #initialize with style
+    imagine = Imagine()
 
     img_data = imagine.sdprem(
         prompt="Woman sitting on a table, looking at the sky, seen from behind",
-        style=Style.ANIME_V2,
-        ratio=Ratio.RATIO_16X9
+        style=Style.NO_STYLE,
+        ratio=Ratio.RATIO_16X9,
+        negative="",
+        seed=1000,
+        cfg=16,
+        model=Model.REALISTIC,
+        asbase64=False  # default is false, putting it here as presentation.
     )
 
     if img_data is None:
         print("An error occurred while generating the image.")
         return
 
-    img_data = imagine.upscale(image=img_data)
+    img_data = imagine.upscale(img_data)
 
     if img_data is None:
         print("An error occurred while upscaling the image.")
         return
 
     try:
-        with open("example.png", mode="wb") as img_file:
+        with open("example.jpeg", mode="wb") as img_file:
             img_file.write(img_data)
     except Exception as e:
         print(f"An error occurred while writing the image to file: {e}")

app_async.py

@@ -1,14 +1,19 @@
 import asyncio
-from imaginepy import AsyncImagine, Style, Ratio
+from imaginepy import AsyncImagine
+from imaginepy.constants import *
 
 
 async def main():
-    imagine = AsyncImagine(style=Style.ANIME_V2) #initialize with style
-
-    img_data = await imagine.sdprem(
+    imagine = AsyncImagine()
+    img_data = imagine.sdprem(
         prompt="Woman sitting on a table, looking at the sky, seen from behind",
-        style=Style.ANIME_V2,
-        ratio=Ratio.RATIO_16X9
+        style=Style.NO_STYLE,
+        ratio=Ratio.RATIO_16X9,
+        negative="",
+        seed=1000,
+        cfg=16,
+        model=Model.REALISTIC,
+        asbase64=False  # default is false, putting it here as presentation.
     )
 
     if img_data is None:
@@ -26,9 +31,8 @@ async def main():
             img_file.write(img_data)
     except Exception as e:
         print(f"An error occurred while writing the image to file: {e}")
-        
+
     await imagine.close()
-    
 
 
 if __name__ == "__main__":

example.jpeg

@@ -1,5 +1,5 @@
-from .sync_imagine import Imagine
-from .async_imagine import AsyncImagine
+from .sync_imagine import Imagine, DeviantArt
+from .async_imagine import AsyncImagine, DeviantArt
 from .constants import *
 
-__version__ = "1.1.0"
+__version__ = "2.0.0"

imaginepy/__init__.py

@@ -0,0 +1,14 @@
+class ImaginePyException(Exception):
+    """Exceptions used by pyimagine."""
+
+
+class InvalidWord(ImaginePyException):
+    """Used word is invalid."""
+
+
+class InvalidSize(ImaginePyException):
+    """Both sizes must be the same."""
+
+
+class BannedContent(ImaginePyException):
+    """Content wanting to be generated is banned."""

imaginepy/async_imagine.py

@@ -0,0 +1,59 @@
+import random
+from io import BytesIO
+
+from PIL import Image
+
+
+def bytes2png(content: bytes) -> bytes:
+    # Convert the image to PNG format
+    src_image = Image.open(BytesIO(content))
+    png_image = src_image.convert("RGBA")
+
+    # Save the PNG image to bytes
+    png_data = BytesIO()
+    png_image.save(png_data, format="PNG")
+    png_data.seek(0)
+    return png_data.getvalue()
+
+
+def clear_dict(value: dict) -> dict:
+    return {key: value for key, value in value.items() if value is not None} if value else None
+
+
+def get_cfg(value: float) -> float:
+    if value < 0.0 or value > 16.0:
+        raise ValueError(f"Invalid CFG, must be in range (0; 16), {value}")
+    return value
+
+
+def get_steps(value: int) -> int:
+    if value < 0 or value > 50:
+        raise ValueError(f"Invalid steps, must be in range (0; 50), {value}")
+    return value
+
+
+def get_word(value: str) -> str:
+    chars = list(value.lower())
+    size = len(chars) - 1
+    vowel = chars[0] in "aeiouy"
+
+    for i in range(size + 1):
+        if i > 0 and chars[i] == chars[i - 1]:
+            chars.insert(i + 1, chars[i])
+            return "".join(chars)
+        if i == size // 2 and size % 2 == 1:
+            chars.insert(i + 1, chars[i])
+            return "".join(chars)
+
+    if vowel:
+        chars.insert(2, chars[1])
+    else:
+        i = random.randint(0, size)
+        chars.insert(i + 1, chars[i])
+    return "".join(chars)
+
+
+def same_size(src1: bytes, src2: bytes) -> bool:
+    width1, height1 = Image.open(BytesIO(src1)).size
+    width2, height2 = Image.open(BytesIO(src2)).size
+    return width1 == width2 and height1 == height2

imaginepy/constants.py

@@ -5,28 +5,39 @@
 
 setup(
     name='imaginepy',
-    version='1.1.0',
+    version='2.0.0',
     author='CEED',
     author_email='',
     description='Python library to create Art with AI.',
     long_description=long_description,
     long_description_content_type="text/markdown",
     url='https://github.com/ItsCEED/imaginepy',
+    license="GPL-3.0-only",
     packages=find_packages(),
+    keywords=[
+        "art",
+        "image",
+        "ai",
+        "stable-diffusion"
+    ],
     classifiers=[
-        "Environment :: Console",
-        "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
+        "Development Status :: 5 - Production/Stable",
+        "Intended Audience :: Developers",
+        "Intended Audience :: End Users/Desktop",
+        "Natural Language :: English",
         "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3.4",
-        "Programming Language :: Python :: 3.5",
-        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3",
         "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "Topic :: Utilities"
     ],
     python_requires='>=3.7',
     install_requires=[
-        'requests',
+        'aiohttp',
         'requests_toolbelt',
         'langdetect',
+        'pybase64'
     ],
-)
+)