webscout 6.2b0__py3-none-any.whl → 6.4__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}")

webscout/models.py

@@ -1,23 +0,0 @@
-from dataclasses import dataclass
-from typing import Dict, Optional
-
-
-@dataclass
-class MapsResult:
-    """Represents a result from the maps search."""
-
-    title: Optional[str] = None
-    address: Optional[str] = None
-    country_code: Optional[str] = None
-    latitude: Optional[str] = None
-    longitude: Optional[str] = None
-    url: Optional[str] = None
-    desc: Optional[str] = None
-    phone: Optional[str] = None
-    image: Optional[str] = None
-    source: Optional[str] = None
-    hours: Optional[Dict[str, str]] = None
-    category: Optional[str] = None
-    facebook: Optional[str] = None
-    instagram: Optional[str] = None
-    twitter: Optional[str] = None

webscout-6.2b0.dist-info/LICENSE.md

@@ -1,50 +0,0 @@
-************************************************
-**** HelpingAI License ****
-************************************************
-
-Version 2.0
-
-Developed by Abhay Koul
-
-### Preamble
-
-The HelpingAI License governs the use of HelpingAI's digital assets, including but not limited to software, scripts, datasets, documents, images, audio recordings, videos. The HelpingAI License aims to provide clear, comprehensive terms for accessing, modifying, and sharing resources, while promoting ethical development practices.
-
-### Grant of Rights
-
-Under the HelpingAI License, HelpingAI grants you the rights to copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Content, provided you comply with the terms and conditions outlined in this document.
-
-### Terms and Conditions
-
-To exercise the rights granted in the previous section, you must adhere to the following terms and conditions:
-
-2.1. **Redistribution of Source Code.**
- If you redistribute the Source Code, you must include the complete HelpingAI License with your distribution. You must also add clear notifications in all modified files stating:
-
- > "This Work is released under the HelpingAI License v2.0."
-
-2.2. **Distribution in Binary Form.**
- If you distribute Binaries derived from the Source Code, you must include the following statement in your distribution:
-
- > "This Work is based on the HelpingAI Licensed Work, under the HelpingAI License v2.0."
-
-2.3. **Notification of Changes.**
- You must clearly indicate any modifications you make to the Source Code or Documentation, including detailed comments about the nature and extent of the changes. Include the date and originator of the modifications.
-
-2.4. **Branding Attribution.**
- You must not remove or alter any HelpingAI branding, logos, or notices included in the Content without explicit prior consent from HelpingAI.
-
-2.5. **Disclaimer of Warranty.**
- The Content is provided "AS IS," without any implied warranties, including but not limited to warranties of merchantability, fitness for a particular purpose, and non-infringement.
-
-2.6. **Limitation of Liability.**
- To the maximum extent permitted by law, neither HelpingAI nor any contributor shall be liable for any loss, personal injury, property damage, or any indirect, special, incidental, or consequential damages arising from or related to the use of the Content.
-
-2.7. **Governing Law.**
- This HelpingAI License shall be governed and construed in accordance with the laws of the jurisdiction where HelpingAI primarily operates.
-
-### Definitions
-
-3.1. **"Source Code"** refers to the preferred form for making modifications to the Content, typically represented by human-readable programming languages, scripts, or documentation formats.
-
-3.2. **"Binaries"** refers to compiled forms of the Source Code, such as executables, libraries, or similar artifacts produced from the Source Code.