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

webscout/Provider/TTI/Nexra.py

@@ -1,120 +0,0 @@
-import requests
-import json
-import os
-import time
-from typing import List, Optional
-from requests.exceptions import RequestException
-
-from webscout.AIbase import ImageProvider
-
-class NexraImager(ImageProvider):
-    """Image provider for Nexra API"""
-
-    AVAILABLE_MODELS = {
-        "standard": ["emi", "stablediffusion-1.5", "stablediffusion-2.1", "sdxl-lora", "dalle", "dalle2", "dalle-mini", "flux", "midjourney"],
-        "prodia": [
-            "dreamshaperXL10_alpha2.safetensors [c8afe2ef]",
-            "dynavisionXL_0411.safetensors [c39cc051]",
-            "juggernautXL_v45.safetensors [e75f5471]",
-            "realismEngineSDXL_v10.safetensors [af771c3f]",
-            "sd_xl_base_1.0.safetensors [be9edd61]",
-            "animagineXLV3_v30.safetensors [75f2f05b]",
-            "sd_xl_base_1.0_inpainting_0.1.safetensors [5679a81a]",
-            "turbovisionXL_v431.safetensors [78890989]",
-            "devlishphotorealism_sdxl15.safetensors [77cba69f]",
-            "realvisxlV40.safetensors [f7fdcb51]"
-        ]
-    }
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        self.url = "https://nexra.aryahcr.cc/api/image/complements"
-        self.headers = {"Content-Type": "application/json"}
-        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, model: str = "flux", amount: int = 1,
-        max_retries: int = 3, retry_delay: int = 5,
-        additional_params: Optional[dict] = None
-    ) -> List[bytes]:
-        assert bool(prompt), "Prompt cannot be null"
-        assert isinstance(amount, int) and amount > 0, "Amount should be a positive integer"
-
-        self.prompt = prompt
-        response = []
-
-        payload = {
-            "prompt": prompt,
-            "model": "prodia" if model in self.AVAILABLE_MODELS["prodia"] else model,
-        }
-
-        if model in self.AVAILABLE_MODELS["prodia"]:
-            payload["data"] = {
-                "model": model,
-                "steps": 25,
-                "cfg_scale": 7,
-                "sampler": "DPM++ 2M Karras",
-                "negative_prompt": ""
-            }
-        if additional_params:
-            payload.update(additional_params)
-
-        for _ in range(max_retries):
-            try:
-                resp = self.session.post(self.url, json=payload, timeout=self.timeout)
-                resp.raise_for_status()
-
-                # Remove leading underscores and then parse JSON
-                response_data = json.loads(resp.text.lstrip("_"))
-
-                if response_data.get("status") and "images" in response_data:
-                    for image_url in response_data["images"]:
-                        img_resp = requests.get(image_url)
-                        img_resp.raise_for_status()
-                        response.append(img_resp.content)
-                    break
-                else:
-                    raise Exception("Failed to generate image: " + str(response_data))
-            except json.JSONDecodeError as json_err:
-                print(f"JSON Decode Error: {json_err}")
-                print(f"Raw response: {resp.text}")
-                if _ == max_retries - 1:
-                    raise
-            except RequestException as e:
-                print(f"Request Exception: {e}")
-                if _ == max_retries - 1:
-                    raise
-            print(f"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]:
-        assert isinstance(response, list), f"Response should be a list, not {type(response)}"
-        name = self.prompt if name is None else name
-
-        filenames = []
-        for i, image in enumerate(response):
-            filename = f"{filenames_prefix}{name}_{i}.{self.image_extension}"
-            filepath = os.path.join(dir, filename)
-
-            with open(filepath, "wb") as fh:
-                fh.write(image)
-            filenames.append(filename)
-
-        return filenames
-
-if __name__ == "__main__":
-    bot = NexraImager()
-    resp_standard = bot.generate("AI-generated image - webscout", "midjourney", 1)
-    print(bot.save(resp_standard))

webscout/Provider/TTI/PollinationsAI.py

@@ -1,138 +0,0 @@
-import requests
-import os
-import time
-from typing import List
-from string import punctuation
-from random import choice
-from requests.exceptions import RequestException
-
-from webscout.AIbase import ImageProvider
-
-class PollinationsAI(ImageProvider):
-    """Image provider for pollinations.ai"""
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        """Initializes the PollinationsAI class.
-
-        Args:
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies (socks). Defaults to {}.
-        """
-        self.image_gen_endpoint = "https://image.pollinations.ai/prompt/{prompt}?width={width}&height={height}&model={model}"
-        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 = "jpeg"
-
-    def generate(
-        self, prompt: str, amount: int = 1, additives: bool = True,
-        width: int = 768, height: int = 768, model: str = "flux",
-        max_retries: int = 3, retry_delay: int = 5
-    ) -> 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.
-            width (int, optional): Width of the generated image. Defaults to 768.
-            height (int, optional): Height of the generated image. Defaults to 768.
-            model (str, optional): The model to use for image generation. Defaults to "flux".
-            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"
-
-        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 = self.image_gen_endpoint.format(
-                prompt=prompt + ads(), width=width, height=height, model=model
-            )
-            
-            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 = PollinationsAI()
-    try:
-        resp = bot.generate("AI-generated image - webscout", 1)
-        print(bot.save(resp))
-    except Exception as e:
-        print(f"An error occurred: {e}")

webscout/Provider/TTI/WebSimAI.py

@@ -1,142 +0,0 @@
-import requests
-import os
-from typing import List
-
-from webscout.AIbase import ImageProvider
-
-class WebSimAI(ImageProvider):
-    """
-    Image provider for WebSim.ai.
-    """
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        """Initializes the WebSimAI class.
-
-        Args:
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies (socks). Defaults to {}.
-        """
-        self.url = "https://websim.ai/api/image_gen"
-        self.headers = {
-            "Accept": "*/*",
-            "Accept-Encoding": "gzip, deflate, br, zstd",
-            "Accept-Language": "en-US,en;q=0.9,en-IN;q=0.8",
-            "Content-Type": "application/json",
-            "User-Agent": (
-                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
-                "AppleWebKit/537.36 (KHTML, like Gecko) "
-                "Chrome/129.0.0.0 Safari/537.36 Edg/129.0.0.0"
-            ),
-            "Origin": "https://websim.ai",
-            "Referer": "https://websim.ai/p/a5yvwmtj8qz6ayx4tlg1"
-        }
-        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, 
-        width: int = 1024, 
-        height: int = 756
-    ) -> List[bytes]:
-        """Generate image from prompt
-
-        Args:
-            prompt (str): Image description.
-            amount (int, optional): Total images to be generated. Defaults to 1.
-            width (int, optional): Width of the generated image. Defaults to 1024.
-            height (int, optional): Height of the generated image. Defaults to 756.
-
-        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):
-            payload = {
-                "prompt": prompt,
-                "width": width,
-                "height": height,
-                "site_id": "KcWvHOHNBP2PmWUYZ",
-            }
-
-            try:
-                resp = self.session.post(self.url, headers=self.headers, json=payload, timeout=self.timeout)
-                resp.raise_for_status()  # Raises HTTPError for bad responses
-
-                response_data = resp.json()
-                image_url = response_data.get("url")
-                if not image_url:
-                    print("No image URL found in the response.")
-                    continue
-
-                image_response = requests.get(image_url)
-                image_response.raise_for_status()
-                response.append(image_response.content)
-
-            except requests.exceptions.HTTPError as http_err:
-                print(f"HTTP error occurred: {http_err} - {response.text}")
-                return []  # Return an empty list on error
-            except requests.exceptions.RequestException as req_err:
-                print(f"Request error occurred: {req_err}")
-                return []  # Return an empty list on error
-
-        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 = WebSimAI()
-    try:
-        resp = bot.generate("A shiny red sports car speeding down a scenic mountain road", 1)
-        print(bot.save(resp))
-    except Exception as e:
-        print(f"An error occurred: {e}")

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}")