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

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/Provider/TTI/talkai.py

@@ -1,116 +0,0 @@
-import uuid
-import requests
-import json
-import os
-from typing import Any, Dict, List, Optional
-
-from webscout.AIbase import ImageProvider
-from webscout import exceptions
-
-
-class TalkaiImager(ImageProvider):
-    """
-    Image provider for Talkai.info.
-    """
-
-    def __init__(self, timeout: int = 60, proxies: dict = {}):
-        """Initializes the TalkaiImager class.
-
-        Args:
-            timeout (int, optional): HTTP request timeout in seconds. Defaults to 60.
-            proxies (dict, optional): HTTP request proxies. Defaults to {}.
-        """
-        self.api_endpoint = "https://talkai.info/chat/send/"
-        self.headers = {
-            'accept': 'application/json',
-            'accept-language': 'en-US,en;q=0.9',
-            'content-type': 'application/json',
-            'origin': 'https://talkai.info',
-            'referer': 'https://talkai.info/image/',
-            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
-        }
-        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) -> List[str]:
-        """Generates image URLs from a prompt."""
-        assert bool(prompt), "Prompt cannot be empty."
-        assert isinstance(amount, int) and amount > 0, "Amount must be a positive integer."
-
-        self.prompt = prompt
-        image_urls = []
-
-        try:
-            with self.session.post(self.api_endpoint, json=self._create_payload(prompt), timeout=self.timeout) as response:
-                response.raise_for_status()
-                data = response.json()
-
-                if 'data' in data and len(data['data']) > 0 and 'url' in data['data'][0]:
-                    image_urls.append(data['data'][0]['url'])
-                else:
-                    raise exceptions.InvalidResponseError("No image URL found in API response.")
-
-        except requests.exceptions.RequestException as e:
-            raise exceptions.APIConnectionError(f"Error making API request: {e}") from e
-        except json.JSONDecodeError as e:
-            raise exceptions.InvalidResponseError(f"Invalid JSON response: {e}") from e
-        except Exception as e:
-            raise exceptions.FailedToGenerateResponseError(f"An unexpected error occurred: {e}") from e
-
-        return image_urls
-
-    def _create_payload(self, prompt: str) -> Dict[str, Any]:
-        return {
-            "type": "image",
-            "messagesHistory": [
-                {
-                    "id": str(uuid.uuid4()),
-                    "from": "you",
-                    "content": prompt
-                }
-            ],
-            "settings": {
-                "model": "gpt-4o-mini"  # Or another suitable model if available
-            }
-        }
-
-
-    def save(
-        self,
-        response: List[str],
-        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, url in enumerate(response):
-            try:
-                with self.session.get(url, stream=True, timeout=self.timeout) as r:
-                    r.raise_for_status()
-                    filename = f"{filenames_prefix}{name}_{i}.{self.image_extension}"
-                    filepath = os.path.join(dir, filename)
-                    with open(filepath, 'wb') as f:
-                        for chunk in r.iter_content(chunk_size=8192):
-                            f.write(chunk)
-                    filenames.append(filename)
-            except requests.exceptions.RequestException as e:
-                print(f"Error downloading image from {url}: {e}")
-                filenames.append(None) # Indicate failure to download
-
-        return filenames
-
-
-if __name__ == "__main__":
-    bot = TalkaiImager()
-    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/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