SimplerLLM 0.3.3.3__tar.gz → 0.3.3.4__tar.gz

{simplerllm-0.3.3.3 → simplerllm-0.3.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SimplerLLM
-Version: 0.3.3.3
+Version: 0.3.3.4
 Summary: An easy-to-use Library for interacting with language models.
 Home-page: https://github.com/hassancs91/SimplerLLM
 Author: Hasan Aboul Hasan

{simplerllm-0.3.3.3 → simplerllm-0.3.3.4}/SimplerLLM/image/generation/__init__.py

@@ -8,6 +8,7 @@ from .base import ImageGenerator, ImageProvider, ImageSize
 from .wrappers.openai_wrapper import OpenAIImageGenerator
 from .wrappers.stability_wrapper import StabilityImageGenerator
 from .wrappers.google_wrapper import GoogleImageGenerator
+from .wrappers.seedream_wrapper import SeedreamImageGenerator
 from .providers.image_response_models import ImageGenerationResponse
 
 __all__ = [
@@ -17,5 +18,6 @@ __all__ = [
     'OpenAIImageGenerator',
     'StabilityImageGenerator',
     'GoogleImageGenerator',
+    'SeedreamImageGenerator',
     'ImageGenerationResponse',
 ]

{simplerllm-0.3.3.3 → simplerllm-0.3.3.4}/SimplerLLM/image/generation/base.py

@@ -8,9 +8,10 @@ class ImageProvider(Enum):
     OPENAI_DALL_E = 1
     STABILITY_AI = 2
     GOOGLE_GEMINI = 3
+    SEEDREAM = 4
     # Future providers can be added here:
-    # MIDJOURNEY = 4
-    # REPLICATE = 5
+    # MIDJOURNEY = 5
+    # REPLICATE = 6
 
 
 class ImageSize(Enum):
@@ -97,6 +98,14 @@ class ImageGenerator:
                 api_key=api_key,
                 verbose=verbose,
             )
+        elif provider == ImageProvider.SEEDREAM:
+            from .wrappers.seedream_wrapper import SeedreamImageGenerator
+            return SeedreamImageGenerator(
+                provider=provider,
+                model_name=model_name or "seedream-4-5-251128",
+                api_key=api_key,
+                verbose=verbose,
+            )
         # Future providers can be added here
         # elif provider == ImageProvider.MIDJOURNEY:
         #     from .wrappers.midjourney_wrapper import MidjourneyImageGenerator
@@ -180,6 +189,16 @@ class ImageGenerator:
             }
             return size_map.get(size, "1:1")
 
+        elif provider == ImageProvider.SEEDREAM:
+            # Seedream uses resolution presets (2K, 4K) rather than aspect ratios
+            size_map = {
+                ImageSize.SQUARE: "2K",
+                ImageSize.HORIZONTAL: "2K",
+                ImageSize.VERTICAL: "2K",
+                ImageSize.PORTRAIT_3_4: "2K",
+            }
+            return size_map.get(size, "2K")
+
         # Default fallback
         return "1024x1024"
 

simplerllm-0.3.3.4/SimplerLLM/image/generation/providers/seedream_image.py

@@ -0,0 +1,342 @@
+from dotenv import load_dotenv
+import os
+import time
+import requests
+import base64
+from .image_response_models import ImageGenerationResponse
+
+# Load environment variables
+load_dotenv(override=True)
+
+MAX_RETRIES = int(os.getenv("MAX_RETRIES", 3))
+RETRY_DELAY = int(os.getenv("RETRY_DELAY", 2))
+
+# BytePlus Seedream API endpoint
+SEEDREAM_API_ENDPOINT = "https://ark.ap-southeast.bytepluses.com/api/v3/images/generations"
+
+
+def generate_image(
+    prompt,
+    model_name="seedream-4-5-251128",
+    size="2K",
+    watermark=False,
+    response_format="url",
+    output_path=None,
+    full_response=False,
+    api_key=None,
+    verbose=False,
+    image=None,
+):
+    """
+    Generate image from text prompt using BytePlus Seedream API.
+
+    Args:
+        prompt: Text description of the desired image
+        model_name: Model to use (default: seedream-4-5-251128)
+        size: Image size - "2K", "4K", or custom dimensions (e.g., "2048x2048")
+        watermark: If True, adds "AI generated" watermark to bottom-right corner
+        response_format: "url" for image URL, "b64_json" for base64 encoded data
+        output_path: Optional file path to save image
+        full_response: If True, returns ImageGenerationResponse with metadata
+        api_key: BytePlus ARK API key (uses ARK_API_KEY env var if not provided)
+        verbose: If True, prints progress information
+        image: Optional URL of reference image for image-to-image generation
+
+    Returns:
+        If full_response=True: ImageGenerationResponse object
+        If output_path provided: file path string
+        If response_format="url": image URL string
+        Otherwise: image bytes
+    """
+    start_time = time.time() if full_response else None
+
+    # Get API key from parameter or environment
+    api_key = api_key or os.getenv("ARK_API_KEY", "")
+    if not api_key:
+        raise ValueError("ARK_API_KEY not found in environment variables or parameters")
+
+    if verbose:
+        print(f"[Seedream] Generating image with model={model_name}, size={size}")
+        if image:
+            print(f"[Seedream] Using reference image: {image}")
+
+    for attempt in range(MAX_RETRIES):
+        try:
+            # Build request headers
+            headers = {
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {api_key}",
+            }
+
+            # Build request data
+            data = {
+                "model": model_name,
+                "prompt": prompt,
+                "size": size,
+                "watermark": watermark,
+            }
+
+            # Add reference image for image-to-image generation
+            if image:
+                data["image"] = image
+
+            # Make API request
+            response = requests.post(
+                SEEDREAM_API_ENDPOINT,
+                headers=headers,
+                json=data,
+                timeout=120
+            )
+
+            # Check for errors
+            if response.status_code != 200:
+                error_msg = f"Seedream API error (status {response.status_code})"
+                try:
+                    error_data = response.json()
+                    error_msg += f": {error_data}"
+                except:
+                    error_msg += f": {response.text}"
+                raise Exception(error_msg)
+
+            # Parse response
+            response_data = response.json()
+
+            if verbose:
+                print(f"[Seedream] Response received: {response_data.get('model', 'unknown model')}")
+
+            # Extract image data from response
+            if "data" not in response_data or len(response_data["data"]) == 0:
+                raise Exception("No image data found in Seedream response")
+
+            image_info = response_data["data"][0]
+            image_url = image_info.get("url")
+            image_size_str = image_info.get("size", size)
+
+            if not image_url:
+                raise Exception("No image URL found in Seedream response")
+
+            # Handle output based on format and output_path
+            if output_path or response_format == "b64_json":
+                # Download the image
+                img_response = requests.get(image_url, timeout=60)
+                img_response.raise_for_status()
+                image_bytes = img_response.content
+
+                if output_path:
+                    # Save to file
+                    os.makedirs(os.path.dirname(output_path), exist_ok=True) if os.path.dirname(output_path) else None
+                    with open(output_path, 'wb') as f:
+                        f.write(image_bytes)
+                    image_data = output_path
+                    file_size = len(image_bytes)
+                    if verbose:
+                        print(f"[Seedream] Image saved to: {output_path} ({file_size} bytes)")
+                else:
+                    image_data = image_bytes
+                    file_size = len(image_bytes)
+                    if verbose:
+                        print(f"[Seedream] Image downloaded ({file_size} bytes)")
+            else:
+                # Return URL
+                image_data = image_url
+                file_size = None
+                if verbose:
+                    print(f"[Seedream] Image URL received: {image_url[:50]}...")
+
+            # Return full response with metadata if requested
+            if full_response:
+                end_time = time.time()
+                process_time = end_time - start_time
+
+                # Extract usage info
+                usage = response_data.get("usage", {})
+
+                return ImageGenerationResponse(
+                    image_data=image_data,
+                    model=model_name,
+                    prompt=prompt,
+                    revised_prompt=None,
+                    size=image_size_str,
+                    quality=None,
+                    style=None,
+                    process_time=process_time,
+                    provider="SEEDREAM",
+                    file_size=file_size,
+                    output_path=output_path,
+                    llm_provider_response=response_data,
+                )
+
+            return image_data
+
+        except Exception as e:
+            if attempt < MAX_RETRIES - 1:
+                if verbose:
+                    print(f"[Seedream] Attempt {attempt + 1} failed: {e}. Retrying...")
+                time.sleep(RETRY_DELAY * (2**attempt))
+            else:
+                error_msg = f"Failed to generate image after {MAX_RETRIES} attempts due to: {e}"
+                raise Exception(error_msg)
+
+
+async def generate_image_async(
+    prompt,
+    model_name="seedream-4-5-251128",
+    size="2K",
+    watermark=False,
+    response_format="url",
+    output_path=None,
+    full_response=False,
+    api_key=None,
+    verbose=False,
+    image=None,
+):
+    """
+    Async version: Generate image from text prompt using BytePlus Seedream API.
+
+    Args:
+        prompt: Text description of the desired image
+        model_name: Model to use (default: seedream-4-5-251128)
+        size: Image size - "2K", "4K", or custom dimensions (e.g., "2048x2048")
+        watermark: If True, adds "AI generated" watermark to bottom-right corner
+        response_format: "url" for image URL, "b64_json" for base64 encoded data
+        output_path: Optional file path to save image
+        full_response: If True, returns ImageGenerationResponse with metadata
+        api_key: BytePlus ARK API key (uses ARK_API_KEY env var if not provided)
+        verbose: If True, prints progress information
+        image: Optional URL of reference image for image-to-image generation
+
+    Returns:
+        If full_response=True: ImageGenerationResponse object
+        If output_path provided: file path string
+        If response_format="url": image URL string
+        Otherwise: image bytes
+    """
+    import asyncio
+
+    loop = asyncio.get_event_loop()
+    return await loop.run_in_executor(
+        None,
+        lambda: generate_image(
+            prompt=prompt,
+            model_name=model_name,
+            size=size,
+            watermark=watermark,
+            response_format=response_format,
+            output_path=output_path,
+            full_response=full_response,
+            api_key=api_key,
+            verbose=verbose,
+            image=image,
+        )
+    )
+
+
+def edit_image(
+    image_source,
+    edit_prompt,
+    model_name="seedream-4-5-251128",
+    size="2K",
+    watermark=False,
+    response_format="url",
+    output_path=None,
+    full_response=False,
+    api_key=None,
+    verbose=False,
+):
+    """
+    Edit an existing image using text instructions with BytePlus Seedream API.
+
+    Args:
+        image_source: URL of the source image to edit
+        edit_prompt: Text instructions for how to edit the image
+        model_name: Model to use (default: seedream-4-5-251128)
+        size: Image size - "2K", "4K", or custom dimensions (e.g., "2048x2048")
+        watermark: If True, adds "AI generated" watermark to bottom-right corner
+        response_format: "url" for image URL, "b64_json" for base64 encoded data
+        output_path: Optional file path to save edited image
+        full_response: If True, returns ImageGenerationResponse with metadata
+        api_key: BytePlus ARK API key (uses ARK_API_KEY env var if not provided)
+        verbose: If True, prints progress information
+
+    Returns:
+        If full_response=True: ImageGenerationResponse object
+        If output_path provided: file path string
+        If response_format="url": image URL string
+        Otherwise: edited image bytes
+
+    Example:
+        >>> edited_url = edit_image(
+        ...     image_source="https://example.com/original.jpg",
+        ...     edit_prompt="Change the background to a sunset sky",
+        ...     size="2K"
+        ... )
+    """
+    if verbose:
+        print(f"[Seedream] Editing image with prompt: {edit_prompt[:50]}...")
+
+    # Use generate_image with the image parameter for image-to-image editing
+    return generate_image(
+        prompt=edit_prompt,
+        model_name=model_name,
+        size=size,
+        watermark=watermark,
+        response_format=response_format,
+        output_path=output_path,
+        full_response=full_response,
+        api_key=api_key,
+        verbose=verbose,
+        image=image_source,
+    )
+
+
+async def edit_image_async(
+    image_source,
+    edit_prompt,
+    model_name="seedream-4-5-251128",
+    size="2K",
+    watermark=False,
+    response_format="url",
+    output_path=None,
+    full_response=False,
+    api_key=None,
+    verbose=False,
+):
+    """
+    Async version: Edit an existing image using text instructions with BytePlus Seedream API.
+
+    Args:
+        image_source: URL of the source image to edit
+        edit_prompt: Text instructions for how to edit the image
+        model_name: Model to use (default: seedream-4-5-251128)
+        size: Image size - "2K", "4K", or custom dimensions (e.g., "2048x2048")
+        watermark: If True, adds "AI generated" watermark to bottom-right corner
+        response_format: "url" for image URL, "b64_json" for base64 encoded data
+        output_path: Optional file path to save edited image
+        full_response: If True, returns ImageGenerationResponse with metadata
+        api_key: BytePlus ARK API key (uses ARK_API_KEY env var if not provided)
+        verbose: If True, prints progress information
+
+    Returns:
+        If full_response=True: ImageGenerationResponse object
+        If output_path provided: file path string
+        If response_format="url": image URL string
+        Otherwise: edited image bytes
+    """
+    import asyncio
+
+    loop = asyncio.get_event_loop()
+    return await loop.run_in_executor(
+        None,
+        lambda: edit_image(
+            image_source=image_source,
+            edit_prompt=edit_prompt,
+            model_name=model_name,
+            size=size,
+            watermark=watermark,
+            response_format=response_format,
+            output_path=output_path,
+            full_response=full_response,
+            api_key=api_key,
+            verbose=verbose,
+        )
+    )