webscout 6.3__py3-none-any.whl → 6.5__py3-none-any.whl

webscout/Provider/TTI/aiforce.py

@@ -1,160 +0,0 @@
-import requests
-import os
-import time
-from typing import List, Optional
-from string import punctuation
-from random import choice
-from requests.exceptions import RequestException
-
-from webscout.AIbase import ImageProvider
-
-class AiForceimager(ImageProvider):
-    """Image provider for Airforce API"""
-
-    AVAILABLE_MODELS = [
-        "flux",
-        "flux-realism",
-        "flux-anime",
-        "flux-3d",
-        "flux-disney",
-        "flux-pixel",
-        "flux-4o",
-        "any-dark"
-]
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        """Initializes the AiForceimagger class.
-
-        Args:
-            api_token (str, optional): Your Airforce API token. If None, it will use the environment variable "AIRFORCE_API_TOKEN". 
-                                      Defaults to None.
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies (socks). Defaults to {}.
-        """
-        self.api_endpoint = "https://api.airforce/imagine2"
-        self.headers = {
-            "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8",
-            "Accept-Language": "en-US,en;q=0.5",
-            "Accept-Encoding": "gzip, deflate",
-            "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:122.0) Gecko/20100101 Firefox/122.0",
-        }
-        self.session = requests.Session()
-        self.session.headers.update(self.headers)
-        self.session.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, 
-        additives: bool = True,
-        model: str = "flux-realism", 
-        width: int = 768, 
-        height: int = 768, 
-        seed: Optional[int] = None,
-        max_retries: int = 3, 
-        retry_delay: int = 5
-    ) -> List[bytes]:
-        """Generate image from prompt
-
-        Args:
-            prompt (str): Image description.
-            amount (int, optional): Total images to be generated. Defaults to 1.
-            additives (bool, optional): Try to make each prompt unique. Defaults to True.
-            model (str, optional): The model to use for image generation. 
-                                    Defaults to "flux". Available options: "flux", "flux-realism".
-            width (int, optional): Width of the generated image. Defaults to 768.
-            height (int, optional): Height of the generated image. Defaults to 768.
-            seed (int, optional): Seed for the random number generator. Defaults to None.
-            max_retries (int, optional): Maximum number of retry attempts. Defaults to 3.
-            retry_delay (int, optional): Delay between retries in seconds. Defaults to 5.
-
-        Returns:
-            List[bytes]: List of generated images as bytes.
-        """
-        assert bool(prompt), "Prompt cannot be null"
-        assert isinstance(amount, int), f"Amount should be an integer only not {type(amount)}"
-        assert amount > 0, "Amount should be greater than 0"
-        assert model in self.AVAILABLE_MODELS, f"Model should be one of {self.AVAILABLE_MODELS}"
-
-        ads = lambda: (
-            ""
-            if not additives
-            else choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-        )
-
-        self.prompt = prompt
-        response = []
-        for _ in range(amount):
-            url = f"{self.api_endpoint}?model={model}&prompt={prompt}&size={width}:{height}"
-            if seed:
-                url += f"&seed={seed}"
-            
-            for attempt in range(max_retries):
-                try:
-                    resp = self.session.get(url, timeout=self.timeout)
-                    resp.raise_for_status()
-                    response.append(resp.content)
-                    break
-                except RequestException as e:
-                    if attempt == max_retries - 1:
-                        print(f"Failed to generate image after {max_retries} attempts: {e}")
-                        raise
-                    else:
-                        print(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds...")
-                        time.sleep(retry_delay)
-
-        return response
-
-    def save(
-        self,
-        response: List[bytes],
-        name: str = None,
-        dir: str = os.getcwd(),
-        filenames_prefix: str = "",
-    ) -> List[str]:
-        """Save generated images
-
-        Args:
-            response (List[bytes]): List of generated images as bytes.
-            name (str): Filename for the images. Defaults to the last prompt.
-            dir (str, optional): Directory for saving images. Defaults to os.getcwd().
-            filenames_prefix (str, optional): String to be prefixed at each filename to be returned.
-
-        Returns:
-            List[str]: List of saved filenames.
-        """
-        assert isinstance(response, list), f"Response should be of {list} not {type(response)}"
-        name = self.prompt if name is None else name
-
-        filenames = []
-        count = 0
-        for image 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])
-
-            with open(absolute_path_to_file, "wb") as fh:
-                fh.write(image)
-
-        return filenames
-
-if __name__ == "__main__":
-    bot = AiForceimager()
-    try:
-        resp = bot.generate("A shiny red sports car speeding down a scenic mountain road", 1, model="flux-realism")
-        print(bot.save(resp))
-    except Exception as e:
-        print(f"An error occurred: {e}")

webscout/Provider/TTI/artbit.py

@@ -1,141 +0,0 @@
-import cloudscraper
-import os
-import requests
-
-from typing import List
-
-from webscout.AIbase import ImageProvider
-
-class ArtbitImager(ImageProvider):
-    """
-    Image provider for Artbit.ai.
-    """
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        """Initializes the ArtbitImager class.
-
-        Args:
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies. Defaults to {}.
-        """
-        self.url = "https://artbit.ai/api/generateImage"
-        self.scraper = cloudscraper.create_scraper()
-        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 image from prompt
-
-        Args:
-            prompt (str): Image description.
-            amount (int, optional): Total images to be generated. Defaults to 1.
-            caption_model (str, optional): Caption model to use. Defaults to "sdxl".
-            selected_ratio (str, optional): Image ratio. Defaults to "1024".
-            negative_prompt (str, optional): Negative prompt. Defaults to "".
-
-        Returns:
-            List[str]: List of generated image URLs.
-        """
-        assert bool(prompt), "Prompt cannot be null"
-        assert isinstance(amount, int), f"Amount should be an integer only not {type(amount)}"
-        assert amount > 0, "Amount should 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:
-            # Sending the POST request using CloudScraper
-            resp = self.scraper.post(self.url, json=payload, timeout=self.timeout)
-            resp.raise_for_status()  # Check for HTTP errors
-
-            # Parsing the JSON response
-            response_data = resp.json()
-
-            # Extracting image URLs from the response
-            imgs = response_data.get("imgs", [])
-            if imgs:
-                response.extend(imgs)
-            else:
-                print("No images found in the response.")
-
-        except requests.RequestException as e:
-            print(f"An error occurred while making the request: {e}")
-            raise
-
-        return response
-
-    def save(
-        self,
-        response: List[str],  # List of image URLs
-        name: str = None,
-        dir: str = os.getcwd(),
-        filenames_prefix: str = "",
-    ) -> List[str]:
-        """Save generated images
-
-        Args:
-            response (List[str]): List of generated image URLs.
-            name (str): Filename for the images. Defaults to the last prompt.
-            dir (str, optional): Directory for saving images. Defaults to os.getcwd().
-            filenames_prefix (str, optional): String to be prefixed at each filename to be returned.
-
-        Returns:
-            List[str]: List of saved filenames.
-        """
-        assert isinstance(response, list), f"Response should be of {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])
-
-            # Download and save the image
-            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:
-                print(f"An error occurred while downloading image from {img_url}: {e}")
-                raise
-
-        return filenames
-
-if __name__ == "__main__":
-    bot = ArtbitImager()
-    try:
-        resp = bot.generate(
-            "A shiny red sports car speeding down a scenic mountain road with a clear blue sky in the background, surrounded by lush green trees.",
-            amount=3
-        )
-        print(bot.save(resp))
-    except Exception as e:
-        print(f"An error occurred: {e}")

webscout/Provider/TTI/deepinfra.py

@@ -1,148 +0,0 @@
-import requests
-import os
-from typing import List
-from string import punctuation
-from random import choice
-from random import randint
-import base64
-
-from webscout.AIbase import ImageProvider
-
-class DeepInfraImager(ImageProvider):
-    """DeepInfra Image provider"""
-
-    def __init__(
-        self,
-        model: str = "black-forest-labs/FLUX-1-schnell",
-        timeout: int = 60,
-        proxies: dict = {},
-    ):
-        """Initializes `DeepInfraImager`
-
-        Args:
-            model (str, optional): The name of the DeepInfra model to use. 
-                                        Defaults to "black-forest-labs/FLUX-1-schnell".
-            timeout (int, optional): Http request timeout. Defaults to 60 seconds.
-            proxies (dict, optional): Http request proxies (socks). Defaults to {}.
-        """
-        self.image_gen_endpoint: str = f"https://api.deepinfra.com/v1/inference/{model}"
-        self.headers = {
-            "Accept": "application/json, text/plain, */*",
-            "Accept-Language": "en-US,en;q=0.9,en-IN;q=0.8",
-            "Accept-Encoding": "gzip, deflate, br, zstd",
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0",
-            "DNT": "1",
-            "Origin": "https://deepinfra.com",
-            "Referer": "https://deepinfra.com/",
-            "Sec-CH-UA": '"Chromium";v="128", "Not;A=Brand";v="24", "Microsoft Edge";v="128"',
-            "Sec-CH-UA-Mobile": "?0",
-            "Sec-CH-UA-Platform": '"Windows"',
-            "Sec-Fetch-Dest": "empty",
-            "Sec-Fetch-Mode": "cors",
-            "Sec-Fetch-Site": "same-site"
-        }
-        self.session = requests.Session()
-        self.session.headers.update(self.headers)
-        self.session.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, additives: bool = True, 
-        num_inference_steps: int = 25, guidance_scale: float = 7.5, 
-        width: int = 1024, height: int = 1024, seed: int = None
-    ) -> list[bytes]:
-        """Generate image from prompt
-
-        Args:
-            prompt (str): Image description.
-            amount (int): Total images to be generated. Defaults to 1.
-            additives (bool, optional): Try to make each prompt unique. Defaults to True.
-            num_inference_steps (int, optional): Number of inference steps. Defaults to 39.
-            guidance_scale (float, optional): Guidance scale for image generation. Defaults to 13.3.
-            width (int, optional): Width of the generated image. Defaults to 1024.
-            height (int, optional): Height of the generated image. Defaults to 1024.
-            seed (int, optional): Random seed for image generation. If None, a random seed is used. 
-                                  Defaults to None.
-
-        Returns:
-            list[bytes]: List of generated images as bytes.
-        """
-        assert bool(prompt), "Prompt cannot be null"
-        assert isinstance(amount, int), f"Amount should be an integer only not {type(amount)}"
-        assert amount > 0, "Amount should be greater than 0"
-
-        ads = lambda: (
-            ""
-            if not additives
-            else choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-            + choice(punctuation)
-        )
-
-        self.prompt = prompt
-        response = []
-        for _ in range(amount):
-            payload = {
-                "prompt": prompt + ads(),
-                "num_inference_steps": num_inference_steps,
-                "guidance_scale": guidance_scale,
-                "width": width,
-                "height": height,
-                "seed": seed if seed is not None else randint(1, 10000), 
-            }
-            resp = self.session.post(url=self.image_gen_endpoint, json=payload, timeout=self.timeout)
-            resp.raise_for_status()
-            # Extract base64 encoded image data and decode it
-            image_data = resp.json()['images'][0].split(",")[1]
-            image_bytes = base64.b64decode(image_data)
-            response.append(image_bytes)
-
-        return response
-
-    def save(
-        self,
-        response: list[bytes],
-        name: str = None,
-        dir: str = os.getcwd(),
-        filenames_prefix: str = "",
-    ) -> list[str]:
-        """Save generated images
-
-        Args:
-            response (list[bytes]): List of generated images as bytes.
-            name (str):  Filename for the images. Defaults to last prompt.
-            dir (str, optional): Directory for saving images. Defaults to os.getcwd().
-            filenames_prefix (str, optional): String to be prefixed at each filename to be returned.
-        """
-        assert isinstance(response, list), f"Response should be of {list} not {type(response)}"
-        name = self.prompt if name is None else name
-
-        filenames: list = []
-        count = 0
-        for image 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])
-
-            with open(absolute_path_to_file, "wb") as fh:
-                fh.write(image)
-
-        return filenames
-
-
-if __name__ == "__main__":
-    bot = DeepInfraImager()
-    resp = bot.generate("AI-generated image - webscout", 1)
-    print(bot.save(resp))

webscout/Provider/TTI/huggingface.py

@@ -1,155 +0,0 @@
-import os
-import requests
-import io
-from PIL import Image
-from typing import Any, List, Optional, Dict
-from webscout.AIbase import ImageProvider
-
-class HFimager(ImageProvider):
-    """
-    Image provider for Hugging Face Inference API.
-    """
-
-    def __init__(
-        self,
-        api_token: str = None,
-        timeout: int = 60,
-        proxies: dict = {}
-    ):
-        """Initializes the HFimager class.
-
-        Args:
-            api_token (str, optional): Hugging Face API token. If None, it will use the environment variable "HUGGINGFACE_API_TOKEN".
-                                       Defaults to None.
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies. Defaults to {}.
-        """
-        self.base_url = "https://api-inference.huggingface.co/models/"
-        self.api_token = api_token or os.environ["HUGGINGFACE_API_TOKEN"]
-        self.headers = {"Authorization": f"Bearer {self.api_token}"}
-        self.session = requests.Session()
-        self.session.headers.update(self.headers)
-        self.session.proxies.update(proxies)
-        self.timeout = timeout
-        self.prompt: str = "AI-generated image - webscout"
-        self.image_extension: str = "jpg"
-
-    def generate(
-        self,
-        prompt: str,
-        amount: int = 1,
-        model: str = "black-forest-labs/FLUX.1-dev",
-        guidance_scale: Optional[float] = None,
-        negative_prompt: Optional[str] = None,
-        num_inference_steps: Optional[int] = None,
-        width: Optional[int] = None,
-        height: Optional[int] = None,
-        scheduler: Optional[str] = None,
-        seed: Optional[int] = None,
-    ) -> List[bytes]:
-        """
-        Generate image from prompt.
-
-        Args:
-            prompt (str): Image description.
-            amount (int): Total images to be generated. Defaults to 1.
-            model (str): Hugging Face model name. Defaults to "black-forest-labs/FLUX.1-dev".
-            guidance_scale (float, optional): Guidance scale value. Defaults to None.
-            negative_prompt (str, optional): Negative prompt. Defaults to None.
-            num_inference_steps (int, optional): Number of inference steps. Defaults to None.
-            width (int, optional): Width of the output image. Defaults to None.
-            height (int, optional): Height of the output image. Defaults to None.
-            scheduler (str, optional): Scheduler to use. Defaults to None.
-            seed (int, optional): Seed for random number generator. Defaults to None.
-
-        Returns:
-            List[bytes]: List of generated images as bytes.
-        """
-        assert bool(prompt), "Prompt cannot be null"
-        assert isinstance(amount, int), f"Amount should be an integer only not {type(amount)}"
-        assert amount > 0, "Amount should be greater than 0"
-
-        self.prompt = prompt
-        response = []
-
-        for _ in range(amount):
-            url = self.base_url + model
-
-            # Create the base payload with the prompt
-            payload: Dict[str, Any] = {"inputs": prompt}
-
-            # Add optional parameters to the payload if provided
-            parameters = {}
-            if guidance_scale is not None:
-                parameters["guidance_scale"] = guidance_scale
-            if negative_prompt is not None:
-                parameters["negative_prompt"] = negative_prompt
-            if num_inference_steps is not None:
-                parameters["num_inference_steps"] = num_inference_steps
-            if width is not None and height is not None:
-                parameters["target_size"] = {"width": width, "height": height}
-            if scheduler is not None:
-                parameters["scheduler"] = scheduler
-            if seed is not None:
-                parameters["seed"] = seed
-
-            # Add the parameters to the payload if any are set
-            if parameters:
-                payload["parameters"] = parameters
-
-            try:
-                resp = self.session.post(url, headers=self.headers, json=payload, timeout=self.timeout)
-                resp.raise_for_status()
-                response.append(resp.content)
-            except requests.RequestException as e:
-                print(f"Failed to generate image: {e}")
-                raise
-
-        return response
-
-    def save(
-        self,
-        response: List[bytes],
-        name: str = None,
-        dir: str = os.getcwd(),
-        filenames_prefix: str = "",
-    ) -> List[str]:
-        """Save generated images
-
-        Args:
-            response (List[bytes]): List of generated images as bytes.
-            name (str): Filename for the images. Defaults to the last prompt.
-            dir (str, optional): Directory for saving images. Defaults to os.getcwd().
-            filenames_prefix (str, optional): String to be prefixed at each filename to be returned.
-
-        Returns:
-            List[str]: List of saved filenames.
-        """
-        assert isinstance(response, list), f"Response should be of {list} not {type(response)}"
-        name = self.prompt if name is None else name
-
-        filenames = []
-        count = 0
-        for image_bytes 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])
-
-            with open(absolute_path_to_file, "wb") as fh:
-                fh.write(image_bytes)
-
-        return filenames
-
-if __name__ == "__main__":
-    bot = HFimager(api_token='your huggingface API')
-    try:
-        resp = bot.generate("A shiny red sports car speeding down a scenic mountain road with a clear blue sky in the background, surrounded by lush green trees.", 1)
-        print(bot.save(resp, name="test"))
-    except Exception as e:
-        print(f"An error occurred: {e}")