webscout 8.2.7__py3-none-any.whl → 8.2.8__py3-none-any.whl

webscout/Provider/TTI/artbit/async_artbit.py

@@ -0,0 +1,155 @@
+"""
+AsyncArtbitImager - Your go-to async provider for generating fire images with Artbit! ⚡
+
+Examples:
+    >>> from webscout import AsyncArtbitImager
+    >>> import asyncio
+    >>> 
+    >>> async def example():
+    ...     # Initialize provider
+    ...     provider = AsyncArtbitImager()
+    ...     
+    ...     # Generate a single image
+    ...     images = await provider.generate("Cool art")
+    ...     paths = await provider.save(images)
+    ...     
+    ...     # Generate multiple images with parameters
+    ...     images = await provider.generate(
+    ...         prompt="Epic dragon in cyberpunk city",
+    ...         amount=3,
+    ...         caption_model="sdxl",
+    ...         selected_ratio="1024",
+    ...         negative_prompt="blurry, bad quality"
+    ...     )
+    ...     paths = await provider.save(images, name="dragon", dir="outputs")
+    >>> 
+    >>> # Run the example
+    >>> asyncio.run(example())
+"""
+
+import aiohttp
+import aiofiles
+import asyncio
+import os
+from typing import Union, List
+from webscout.AIbase import AsyncImageProvider
+from webscout.litagent import LitAgent
+
+agent = LitAgent()
+
+class AsyncArtbitImager(AsyncImageProvider):
+    """Your go-to async provider for generating fire images with Artbit! ⚡"""
+
+    def __init__(self, timeout: int = 60, proxies: dict = {}):
+        """Initialize your async Artbit provider with custom settings! ⚙️
+
+        Args:
+            timeout (int): Request timeout in seconds (default: 60)
+            proxies (dict): Proxy settings for requests (default: {})
+        """
+        self.url = "https://artbit.ai/api/generateImage"
+        self.headers = {
+            "User-Agent": agent.random(),
+            "Content-Type": "application/json",
+            "Accept": "application/json"
+        }
+        self.timeout = timeout
+        self.proxies = proxies
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "png"
+
+    async def generate(
+        self, 
+        prompt: str, 
+        amount: int = 1,
+        caption_model: str = "sdxl",
+        selected_ratio: str = "1024",
+        negative_prompt: str = ""
+    ) -> List[str]:
+        """Generate some fire images asynchronously! ⚡
+
+        Args:
+            prompt (str): Your lit image description
+            amount (int): How many images to generate (default: 1)
+            caption_model (str): Which model to use (default: "sdxl")
+            selected_ratio (str): Image size ratio (default: "1024")
+            negative_prompt (str): What you don't want in the image (default: "")
+
+        Returns:
+            List[str]: Your generated image URLs
+        """
+        assert bool(prompt), "Yo fam, prompt can't be empty! 🚫"
+        assert isinstance(amount, int), f"Amount gotta be an integer, not {type(amount)} 🤔"
+        assert amount > 0, "Amount gotta be greater than 0! 📈"
+
+        self.prompt = prompt
+        response: List[str] = []
+
+        payload = {
+            "captionInput": prompt,
+            "captionModel": caption_model,
+            "selectedRatio": selected_ratio,
+            "selectedSamples": str(amount),
+            "negative_prompt": negative_prompt
+        }
+
+        try:
+            async with aiohttp.ClientSession(headers=self.headers) as session:
+                async with session.post(self.url, json=payload, timeout=self.timeout) as resp:
+                    resp.raise_for_status()
+                    response_data = await resp.json()
+                    imgs = response_data.get("imgs", [])
+                    if imgs:
+                        response.extend(imgs)
+
+        except aiohttp.ClientError as e:
+            raise
+
+        return response
+
+    async def save(
+        self,
+        response: List[str],
+        name: str = None,
+        dir: str = os.getcwd(),
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire images asynchronously! 💾
+
+        Args:
+            response (List[str]): Your image URLs to save
+            name (str, optional): Custom name (default: uses prompt)
+            dir (str, optional): Where to save (default: current directory)
+            filenames_prefix (str, optional): Add prefix to filenames
+
+        Returns:
+            List[str]: Where your images were saved
+        """
+        assert isinstance(response, list), f"Response gotta be a list, not {type(response)} 🤔"
+        name = self.prompt if name is None else name
+
+        filenames = []
+        count = 0
+
+        async with aiohttp.ClientSession(headers=self.headers) as session:
+            for img_url in response:
+                def complete_path():
+                    count_value = "" if count == 0 else f"_{count}"
+                    return os.path.join(dir, name + count_value + "." + self.image_extension)
+
+                while os.path.isfile(complete_path()):
+                    count += 1
+
+                absolute_path_to_file = complete_path()
+                filenames.append(filenames_prefix + os.path.split(absolute_path_to_file)[1])
+
+                try:
+                    async with session.get(img_url, timeout=self.timeout) as resp:
+                        resp.raise_for_status()
+                        async with aiofiles.open(absolute_path_to_file, "wb") as fh:
+                            await fh.write(await resp.read())
+
+                except aiohttp.ClientError as e:
+                    raise
+
+        return filenames

webscout/Provider/TTI/artbit/sync_artbit.py

@@ -0,0 +1,148 @@
+"""
+ArtbitImager - Your go-to provider for generating fire images with Artbit! 🔥
+
+Examples:
+    >>> from webscout import ArtbitImager
+    >>> 
+    >>> # Initialize provider
+    >>> provider = ArtbitImager()
+    >>> 
+    >>> # Generate a single image
+    >>> images = provider.generate("Cool art")
+    >>> paths = provider.save(images)
+    >>> 
+    >>> # Generate multiple images with parameters
+    >>> images = provider.generate(
+    ...     prompt="Epic dragon in cyberpunk city",
+    ...     amount=3,
+    ...     caption_model="sdxl",
+    ...     selected_ratio="1024",
+    ...     negative_prompt="blurry, bad quality"
+    ... )
+    >>> paths = provider.save(images, name="dragon", dir="outputs")
+"""
+
+import cloudscraper
+import os
+import requests
+from typing import Union, List
+from webscout.AIbase import ImageProvider
+from webscout.litagent import LitAgent
+
+agent = LitAgent()
+
+class ArtbitImager(ImageProvider):
+    """Your go-to provider for generating fire images with Artbit! 🔥"""
+
+    def __init__(self, timeout: int = 60, proxies: dict = {}):
+        """Initialize your Artbit provider with custom settings! ⚙️
+
+        Args:
+            timeout (int): Request timeout in seconds (default: 60)
+            proxies (dict): Proxy settings for requests (default: {})
+        """
+        self.url = "https://artbit.ai/api/generateImage"
+        self.scraper = cloudscraper.create_scraper()
+        self.scraper.headers.update({"User-Agent": agent.random()})
+        self.scraper.proxies.update(proxies)
+        self.timeout = timeout
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "png"
+
+    def generate(
+        self, 
+        prompt: str, 
+        amount: int = 1,
+        caption_model: str = "sdxl",
+        selected_ratio: str = "1024",
+        negative_prompt: str = ""
+    ) -> List[str]:
+        """Generate some fire images! 🎨
+
+        Args:
+            prompt (str): Your lit image description
+            amount (int): How many images to generate (default: 1)
+            caption_model (str): Which model to use (default: "sdxl")
+            selected_ratio (str): Image size ratio (default: "1024")
+            negative_prompt (str): What you don't want in the image (default: "")
+
+        Returns:
+            List[str]: Your generated image URLs
+        """
+        assert bool(prompt), "Yo fam, prompt can't be empty! 🚫"
+        assert isinstance(amount, int), f"Amount gotta be an integer, not {type(amount)} 🤔"
+        assert amount > 0, "Amount gotta be greater than 0! 📈"
+
+        self.prompt = prompt
+        response: List[str] = []
+
+        payload = {
+            "captionInput": prompt,
+            "captionModel": caption_model,
+            "selectedRatio": selected_ratio,
+            "selectedSamples": str(amount),
+            "negative_prompt": negative_prompt
+        }
+
+        try:
+            resp = self.scraper.post(self.url, json=payload, timeout=self.timeout)
+            resp.raise_for_status()
+
+            response_data = resp.json()
+            imgs = response_data.get("imgs", [])
+            
+            if imgs:
+                response.extend(imgs)
+
+        except requests.RequestException as e:
+            raise
+
+        return response
+
+    def save(
+        self,
+        response: List[str],
+        name: str = None,
+        dir: str = os.getcwd(),
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire images! 💾
+
+        Args:
+            response (List[str]): Your image URLs to save
+            name (str, optional): Custom name (default: uses prompt)
+            dir (str, optional): Where to save (default: current directory)
+            filenames_prefix (str, optional): Add prefix to filenames
+
+        Returns:
+            List[str]: Where your images were saved
+        """
+        assert isinstance(response, list), f"Response gotta be a list, not {type(response)} 🤔"
+        name = self.prompt if name is None else name
+
+        filenames = []
+        count = 0
+
+        for img_url in response:
+            def complete_path():
+                count_value = "" if count == 0 else f"_{count}"
+                return os.path.join(dir, name + count_value + "." + self.image_extension)
+
+            while os.path.isfile(complete_path()):
+                count += 1
+
+            absolute_path_to_file = complete_path()
+            filenames.append(filenames_prefix + os.path.split(absolute_path_to_file)[1])
+
+            try:
+                img_response = requests.get(img_url, stream=True, timeout=self.timeout)
+                img_response.raise_for_status()
+
+                with open(absolute_path_to_file, "wb") as fh:
+                    for chunk in img_response.iter_content(chunk_size=8192):
+                        fh.write(chunk)
+
+            except requests.exceptions.RequestException as e:
+                raise
+
+        return filenames

webscout/Provider/TTI/fastflux/README.md

@@ -0,0 +1,129 @@
+# 🎨 FastFlux Image Generator
+
+Generate amazing images with FastFlux's AI art generator! 🚀
+
+## 🌟 Features
+
+- Fast and reliable image generation
+- Both sync and async implementations
+- Smart retry mechanism
+- Multiple model options
+- Different aspect ratios
+- Proxy support
+- Custom timeouts
+- Easy-to-use interface
+
+## 📦 Installation
+
+```bash
+pip install webscout
+```
+
+## 🚀 Quick Start
+
+### Sync Usage
+
+```python
+from webscout import FastFluxImager
+
+# Initialize the provider
+provider = FastFluxImager()
+
+# Generate a single image
+images = provider.generate("A beautiful sunset over mountains")
+paths = provider.save(images)
+
+# Generate multiple images with parameters
+images = provider.generate(
+    prompt="Epic dragon breathing fire",
+    amount=3,
+    model="flux_1_dev",
+    size="16_9"
+)
+paths = provider.save(images, dir="dragon_pics")
+```
+
+### Async Usage
+
+```python
+from webscout import AsyncFastFluxImager
+import asyncio
+
+async def generate_images():
+    provider = AsyncFastFluxImager()
+    
+    # Generate multiple images
+    images = await provider.generate(
+        "Epic dragon in cyberpunk city",
+        amount=2,
+        model="flux_1_schnell"
+    )
+    paths = await provider.save(images, dir="outputs")
+
+# Run the async function
+asyncio.run(generate_images())
+```
+
+## 🎨 Available Models
+
+- `flux_1_schnell` - Fast generation model (default)
+- `flux_1_dev` - Developer model with detailed outputs
+- `sana_1_6b` - SANA 1.6B model for specialized images
+
+## ⚙️ Configuration
+
+```python
+# Custom settings
+provider = FastFluxImager(
+    timeout=120,  # Longer timeout
+    proxies={
+        'http': 'http://proxy:8080',
+        'https': 'http://proxy:8080'
+    },
+    logging=True  # Enable logging
+)
+
+# Advanced usage
+images = provider.generate(
+    prompt="A shiny red sports car",
+    amount=3,
+    model="flux_1_dev",
+    size="16_9",
+    is_public=False,
+    max_retries=5,
+    retry_delay=3
+)
+```
+
+## 💾 Save Options
+
+```python
+# Save with custom options
+paths = provider.save(
+    images,
+    name="sports_car",  # Custom name
+    dir="my_images",    # Custom directory
+    filenames_prefix="v1_"  # Add prefix
+)
+```
+
+## 🛡️ Error Handling
+
+```python
+try:
+    images = provider.generate("Cool art")
+    paths = provider.save(images)
+except ValueError as e:
+    print(f"Invalid input: {e}")
+except Exception as e:
+    print(f"Generation failed: {e}")
+```
+
+## 💡 Tips
+
+1. Use clear, descriptive prompts
+2. `flux_1_schnell` is faster but `flux_1_dev` gives better quality
+3. Set longer timeouts for better quality models
+4. Enable proxies for better reliability
+5. Use retry mechanism for stability
+6. Save images with meaningful names

webscout/Provider/TTI/fastflux/__init__.py

@@ -0,0 +1,22 @@
+"""
+FastFlux Image Generator - Your go-to provider for generating fire images! 🔥
+
+Examples:
+    >>> # Sync Usage
+    >>> from webscout import FastFluxImager
+    >>> provider = FastFluxImager()
+    >>> images = provider.generate("Cool art")
+    >>> paths = provider.save(images)
+    >>>
+    >>> # Async Usage
+    >>> from webscout import AsyncFastFluxImager
+    >>> async def example():
+    ...     provider = AsyncFastFluxImager()
+    ...     images = await provider.generate("Epic dragon")
+    ...     paths = await provider.save(images)
+"""
+
+from .sync_fastflux import FastFluxImager
+from .async_fastflux import AsyncFastFluxImager
+
+__all__ = ["FastFluxImager", "AsyncFastFluxImager"]