webscout 7.2__py3-none-any.whl → 7.3__py3-none-any.whl

webscout/Provider/Deepinfra.py

@@ -9,7 +9,8 @@ from webscout.AIutel import AwesomePrompts, sanitize_stream
 from webscout.AIbase import Provider, AsyncProvider
 from webscout import exceptions
 from webscout import LitAgent
-from webscout.Litlogger import Logger, LogFormat
+from webscout.Litlogger import Logger, LogFormat, ConsoleHandler
+from webscout.Litlogger.core.level import LogLevel
 
 class DeepInfra(Provider):
     """
@@ -79,15 +80,20 @@ class DeepInfra(Provider):
         )
         self.conversation.history_offset = history_offset
 
-        # Initialize logger if enabled
-        self.logger = Logger(
-            name="DeepInfra",
-            format=LogFormat.MODERN_EMOJI,
-
-        ) if logging else None
-
-        if self.logger:
-            self.logger.info("DeepInfra initialized successfully")
+        # Initialize logger with proper configuration
+        if logging:
+            console_handler = ConsoleHandler(
+                level=LogLevel.DEBUG,
+            )
+            
+            self.logger = Logger(
+                name="DeepInfra",                
+                level=LogLevel.DEBUG,
+                handlers=[console_handler]
+            )
+            self.logger.info("DeepInfra initialized successfully ✨")
+        else:
+            self.logger = None
 
     def ask(
         self,
@@ -97,6 +103,9 @@ class DeepInfra(Provider):
         optimizer: str = None,
         conversationally: bool = False,
     ) -> Union[Dict[str, Any], Generator]:
+        if self.logger:
+            self.logger.debug(f"Processing request - Stream: {stream}, Optimizer: {optimizer}")
+        
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
             if optimizer in self.__available_optimizers:
@@ -104,7 +113,7 @@ class DeepInfra(Provider):
                     conversation_prompt if conversationally else prompt
                 )
                 if self.logger:
-                    self.logger.debug(f"Applied optimizer: {optimizer}")
+                    self.logger.info(f"Applied optimizer: {optimizer} 🔧")
             else:
                 if self.logger:
                     self.logger.error(f"Invalid optimizer requested: {optimizer}")
@@ -120,25 +129,30 @@ class DeepInfra(Provider):
             "stream": stream
         }
 
+        if self.logger:
+            self.logger.debug(f"Sending request to model: {self.model} 🚀")
+
         def for_stream():
             if self.logger:
-                self.logger.debug("Sending streaming request to DeepInfra API...")
+                self.logger.info("Starting stream processing ⚡")
             try:
                 with requests.post(self.url, headers=self.headers, data=json.dumps(payload), stream=True, timeout=self.timeout) as response:
                     if response.status_code != 200:
                         if self.logger:
-                            self.logger.error(f"Request failed with status code {response.status_code}")
-
-                        raise exceptions.FailedToGenerateResponseError(f"Request failed with status code {response.status_code}")
-                    if self.logger:
-                        self.logger.debug(response.text)
+                            self.logger.error(f"Request failed with status {response.status_code} ❌")
+                        raise exceptions.FailedToGenerateResponseError(
+                            f"Request failed with status code {response.status_code}"
+                        )
+                    
                     streaming_text = ""
                     for line in response.iter_lines(decode_unicode=True):
                         if line:
                             line = line.strip()
                             if line.startswith("data: "):
-                                json_str = line[6:]  # Remove "data: " prefix
+                                json_str = line[6:]
                                 if json_str == "[DONE]":
+                                    if self.logger:
+                                        self.logger.info("Stream completed successfully ✅")
                                     break
                                 try:
                                     json_data = json.loads(json_str)
@@ -151,17 +165,19 @@ class DeepInfra(Provider):
                                             yield resp if raw else resp
                                 except json.JSONDecodeError:
                                     if self.logger:
-                                        self.logger.error("JSON decode error in streaming data")
-                                    pass
+                                        self.logger.error("Failed to decode JSON response 🔥")
+                                    continue
+                    
                     self.conversation.update_chat_history(prompt, streaming_text)
-                    if self.logger:
-                        self.logger.info("Streaming response completed successfully")
+                    
             except requests.RequestException as e:
                 if self.logger:
-                    self.logger.error(f"Request failed: {e}")
-                raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
+                    self.logger.error(f"Request failed: {str(e)} 🔥")
+                raise exceptions.FailedToGenerateResponseError(f"Request failed: {str(e)}")
 
         def for_non_stream():
+            if self.logger:
+                self.logger.debug("Processing non-stream request")
             for _ in for_stream():
                 pass
             return self.last_response

webscout/Provider/TTI/FreeAIPlayground/__init__.py

@@ -0,0 +1,9 @@
+"""
+FreeAI Provider Package
+Provides access to various AI models for image generation including DALL-E 3 and Flux models
+"""
+
+from .sync_freeaiplayground import FreeAIImager
+from .async_freeaiplayground import AsyncFreeAIImager
+
+__all__ = ['FreeAIImager', 'AsyncFreeAIImager']

webscout/Provider/TTI/FreeAIPlayground/async_freeaiplayground.py

@@ -0,0 +1,206 @@
+import aiohttp
+import asyncio
+import os
+from typing import List, Union, AsyncGenerator
+from string import punctuation
+from random import choice
+import aiofiles
+
+from webscout.AIbase import AsyncImageProvider
+from webscout.litagent import LitAgent  # Import our fire user agent generator 🔥
+from webscout.Litlogger import Logger  # For that cyberpunk logging swag ⚡
+
+# Initialize our fire logger 🚀
+logger = Logger("AsyncFreeAIPlayground")
+
+class AsyncFreeAIImager(AsyncImageProvider):
+    """
+    Async FreeAI Image Provider - Your go-to for fire AI art! 🎨
+    """
+    
+    AVAILABLE_MODELS = [
+        "dall-e-3",
+        "Flux Pro Ultra",
+        "Flux Pro",
+        "Flux Pro Ultra Raw",
+        "Flux Schnell",
+        "Flux Realism",
+        "grok-2-aurora"
+    ]
+
+    def __init__(
+        self,
+        model: str = "dall-e-3",  # Updated default model
+        timeout: int = 60,
+        proxies: dict = {},
+        logging: bool = True
+    ):
+        """Initialize your async FreeAIPlayground provider with custom settings! ⚙️
+
+        Args:
+            model (str): Which model to use (default: dall-e-3)
+            timeout (int): Request timeout in seconds (default: 60)
+            proxies (dict): Proxy settings for requests (default: {})
+            logging (bool): Enable fire logging (default: True)
+        """
+        self.image_gen_endpoint: str = "https://api.freeaichatplayground.com/v1/images/generations"
+        self.headers = {
+            "Accept": "application/json",
+            "Accept-Language": "en-US,en;q=0.9",
+            "Content-Type": "application/json",
+            "User-Agent": LitAgent().random(),  # Using our fire random agent! 🔥
+            "Origin": "https://freeaichatplayground.com",
+            "Referer": "https://freeaichatplayground.com/",
+        }
+        self.timeout = aiohttp.ClientTimeout(total=timeout)
+        self.model = model
+        self.proxies = proxies
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "png"
+        self.logging = logging
+        if self.logging:
+            logger.info("AsyncFreeAIPlayground initialized! Ready to create some fire art! 🚀")
+
+    async def generate(
+        self, prompt: str, amount: int = 1, additives: bool = True,
+        size: str = "1024x1024", quality: str = "standard",
+        style: str = "vivid", max_retries: int = 3, retry_delay: int = 5
+    ) -> List[bytes]:
+        """Generate some fire images from your prompt! 🎨
+
+        Args:
+            prompt (str): Your creative prompt
+            amount (int): How many images to generate
+            additives (bool): Add random characters to make prompts unique
+            size (str): Image size (1024x1024, 1024x1792, 1792x1024)
+            quality (str): Image quality (standard, hd)
+            style (str): Image style (vivid, natural)
+            max_retries (int): Max retry attempts if generation fails
+            retry_delay (int): Delay between retries in seconds
+
+        Returns:
+            List[bytes]: Your 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)
+        )
+
+        if self.logging:
+            logger.info(f"Generating {amount} images... 🎨")
+
+        self.prompt = prompt
+        response = []
+        
+        async with aiohttp.ClientSession(headers=self.headers, timeout=self.timeout) as session:
+            for i in range(amount):
+                payload = {
+                    "model": self.model,
+                    "prompt": prompt + ads(),
+                    "n": 1,
+                    "size": size,
+                    "quality": quality,
+                    "style": style
+                }
+                
+                for attempt in range(max_retries):
+                    try:
+                        async with session.post(self.image_gen_endpoint, json=payload) as resp:
+                            resp.raise_for_status()
+                            data = await resp.json()
+                            image_url = data['data'][0]['url']
+                            
+                            # Get the image data from the URL
+                            async with session.get(image_url) as img_resp:
+                                img_resp.raise_for_status()
+                                image_bytes = await img_resp.read()
+                                response.append(image_bytes)
+                                
+                            if self.logging:
+                                logger.success(f"Generated image {len(response)}/{amount}! 🎨")
+                            break
+                    except Exception as e:
+                        if attempt == max_retries - 1:
+                            if self.logging:
+                                logger.error(f"Failed to generate image after {max_retries} attempts: {e} 😢")
+                            raise
+                        if self.logging:
+                            logger.warning(f"Attempt {attempt + 1} failed, retrying in {retry_delay}s... 🔄")
+                        await asyncio.sleep(retry_delay)
+
+        if self.logging:
+            logger.success("All images generated successfully! 🎉")
+        return response
+
+    async def save(
+        self,
+        response: Union[List[bytes], AsyncGenerator[bytes, None]],
+        name: str = None,
+        dir: str = os.getcwd(),
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire images! 💾
+
+        Args:
+            response (Union[List[bytes], AsyncGenerator[bytes, None]]): Image data
+            name (str, optional): Base name for saved files
+            dir (str, optional): Where to save the images
+            filenames_prefix (str, optional): Prefix for filenames
+
+        Returns:
+            List[str]: List of saved filenames
+        """
+        if not os.path.exists(dir):
+            os.makedirs(dir)
+            if self.logging:
+                logger.info(f"Created directory: {dir} 📁")
+
+        name = self.prompt if name is None else name
+        saved_paths = []
+
+        async def save_single_image(image_bytes: bytes, index: int) -> str:
+            filename = f"{filenames_prefix}{name}_{index}.{self.image_extension}"
+            filepath = os.path.join(dir, filename)
+            
+            async with aiofiles.open(filepath, "wb") as f:
+                await f.write(image_bytes)
+            
+            if self.logging:
+                logger.success(f"Saved image to: {filepath} 💾")
+            return filename
+
+        if isinstance(response, list):
+            image_list = response
+        else:
+            image_list = [chunk async for chunk in response]
+
+        if self.logging:
+            logger.info(f"Saving {len(image_list)} images... 💾")
+
+        tasks = [save_single_image(img, i) for i, img in enumerate(image_list)]
+        saved_paths = await asyncio.gather(*tasks)
+
+        if self.logging:
+            logger.success(f"All images saved successfully! Check {dir} 🎉")
+        return saved_paths
+
+
+if __name__ == "__main__":
+    async def main():
+        bot = AsyncFreeAIImager()
+        try:
+            resp = await bot.generate("A shiny red sports car speeding down a scenic mountain road", 1)
+            paths = await bot.save(resp)
+            print(paths)
+        except Exception as e:
+            if bot.logging:
+                logger.error(f"An error occurred: {e} 😢")
+
+    asyncio.run(main())

webscout/Provider/TTI/FreeAIPlayground/sync_freeaiplayground.py

@@ -0,0 +1,192 @@
+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
+from webscout.litagent import LitAgent  # Import our fire user agent generator 🔥
+from webscout.Litlogger import Logger  # For that cyberpunk logging swag ⚡
+
+# Initialize our fire logger 🚀
+logger = Logger("FreeAIPlayground")
+
+class FreeAIImager(ImageProvider):
+    """
+    FreeAI Image Provider - Your go-to for fire AI art! 🎨
+    """
+    
+    AVAILABLE_MODELS = [
+        "dall-e-3",
+        "Flux Pro Ultra",
+        "Flux Pro",
+        "Flux Pro Ultra Raw",
+        "Flux Schnell",
+        "Flux Realism",
+        "grok-2-aurora"
+    ]
+
+    def __init__(
+        self,
+        model: str = "dall-e-3",  # Updated default model
+        timeout: int = 60,
+        proxies: dict = {},
+        logging: bool = True
+    ):
+        """Initialize your FreeAIPlayground provider with custom settings! ⚙️
+
+        Args:
+            model (str): Which model to use (default: dall-e-3)
+            timeout (int): Request timeout in seconds (default: 60)
+            proxies (dict): Proxy settings for requests (default: {})
+            logging (bool): Enable fire logging (default: True)
+        """
+        self.image_gen_endpoint: str = "https://api.freeaichatplayground.com/v1/images/generations"
+        self.headers = {
+            "Accept": "application/json",
+            "Accept-Language": "en-US,en;q=0.9",
+            "Content-Type": "application/json",
+            "User-Agent": LitAgent().random(),  # Using our fire random agent! 🔥
+            "Origin": "https://freeaichatplayground.com",
+            "Referer": "https://freeaichatplayground.com/",
+        }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        self.session.proxies.update(proxies)
+        self.timeout = timeout
+        self.model = model
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "png"
+        self.logging = logging
+        if self.logging:
+            logger.info("FreeAIPlayground initialized! Ready to create some fire art! 🚀")
+
+    def generate(
+        self, prompt: str, amount: int = 1, additives: bool = True,
+        size: str = "1024x1024", quality: str = "standard",
+        style: str = "vivid"
+    ) -> List[bytes]:
+        """Generate some fire images from your prompt! 🎨
+
+        Args:
+            prompt (str): Your creative prompt
+            amount (int): How many images to generate
+            additives (bool): Add random characters to make prompts unique
+            size (str): Image size (1024x1024, 1024x1792, 1792x1024)
+            quality (str): Image quality (standard, hd)
+            style (str): Image style (vivid, natural)
+
+        Returns:
+            List[bytes]: Your 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)
+        )
+
+        if self.logging:
+            logger.info(f"Generating {amount} images... 🎨")
+
+        self.prompt = prompt
+        response = []
+        for _ in range(amount):
+            payload = {
+                "model": self.model,
+                "prompt": prompt + ads(),
+                "n": 1,
+                "size": size,
+                "quality": quality,
+                "style": style
+            }
+            try:
+                resp = self.session.post(
+                    url=self.image_gen_endpoint,
+                    json=payload,
+                    timeout=self.timeout
+                )
+                resp.raise_for_status()
+                image_url = resp.json()['data'][0]['url']
+                # Get the image data from the URL
+                img_resp = self.session.get(image_url, timeout=self.timeout)
+                img_resp.raise_for_status()
+                response.append(img_resp.content)
+                if self.logging:
+                    logger.success(f"Generated image {len(response)}/{amount}! 🎨")
+            except Exception as e:
+                if self.logging:
+                    logger.error(f"Failed to generate image: {e} 😢")
+                raise
+
+        if self.logging:
+            logger.success("All images generated successfully! 🎉")
+        return response
+
+    def save(
+        self,
+        response: List[bytes],
+        name: str = None,
+        dir: str = os.getcwd(),
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire images! 💾
+
+        Args:
+            response (List[bytes]): List of image data
+            name (str, optional): Base name for saved files
+            dir (str, optional): Where to save the images
+            filenames_prefix (str, optional): Prefix for filenames
+
+        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
+
+        if not os.path.exists(dir):
+            os.makedirs(dir)
+            if self.logging:
+                logger.info(f"Created directory: {dir} 📁")
+
+        if self.logging:
+            logger.info(f"Saving {len(response)} images... 💾")
+
+        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)
+            if self.logging:
+                logger.success(f"Saved image to: {absolute_path_to_file} 💾")
+
+        if self.logging:
+            logger.success(f"All images saved successfully! Check {dir} 🎉")
+        return filenames
+
+
+if __name__ == "__main__":
+    bot = FreeAIImager()
+    try:
+        resp = bot.generate("A shiny red sports car speeding down a scenic mountain road", 1)
+        print(bot.save(resp))
+    except Exception as e:
+        if bot.logging:
+            logger.error(f"An error occurred: {e} 😢")

webscout/Provider/TTI/__init__.py

@@ -1,3 +1,4 @@
+from .FreeAIPlayground import *
 from .deepinfra import *
 from .PollinationsAI import *
 from .AiForce import *
@@ -6,4 +7,4 @@ from .Nexra import *
 from .huggingface import *
 from .artbit import *
 from .imgninza import *
-from .talkai import *
+from .talkai import *

webscout/Provider/TextPollinationsAI.py

@@ -13,10 +13,30 @@ class TextPollinationsAI(Provider):
     """
 
     AVAILABLE_MODELS = [
-        "openai", "openai-large", "qwen", "qwen-coder", "llama", "mistral",
-        "unity", "midijourney", "rtist", "searchgpt", "evil", "deepseek",
-        "claude-hybridspace", "deepseek-r1", "llamalight", "llamaguard",
-        "gemini", "gemini-thinking", "hormoz"
+        "openai",              # OpenAI GPT-4o-mini
+        "openai-large",        # OpenAI GPT-4o
+        "openai-reasoning",    # OpenAI o1-mini
+        "qwen-coder",          # Qwen 2.5 Coder 32B
+        "llama",               # Llama 3.3 70B
+        "mistral",             # Mistral Nemo
+        "unity",               # Unity with Mistral Large
+        "midijourney",         # Midijourney musical transformer
+        "rtist",               # Rtist image generator
+        "searchgpt",           # SearchGPT with realtime search
+        "evil",                # Evil Mode - Experimental
+        "deepseek",            # DeepSeek-V3
+        "claude-hybridspace",  # Claude Hybridspace
+        "deepseek-r1",         # DeepSeek-R1 Distill Qwen 32B
+        "deepseek-reasoner",   # DeepSeek R1 - Full
+        "llamalight",          # Llama 3.1 8B Instruct
+        "llamaguard",          # Llamaguard 7B AWQ
+        "gemini",              # Gemini 2.0 Flash
+        "gemini-thinking",     # Gemini 2.0 Flash Thinking
+        "hormoz",              # Hormoz 8b
+        "hypnosis-tracy",      # Hypnosis Tracy
+        "sur",                 # Sur AI Assistant
+        "sur-mistral",         # Sur AI Assistant (Mistral)
+        "llama-scaleway"       # Llama (Scaleway)
     ]
 
     def __init__(
@@ -225,4 +245,5 @@ if __name__ == "__main__":
     ai = TextPollinationsAI(model="deepseek-r1", logging=True)
     response = ai.chat(input(">>> "), stream=True)
     for chunk in response:
-        print(chunk, end="", flush=True)
+        print(chunk, end="", flush=True)
+

webscout/Provider/__init__.py

@@ -69,6 +69,7 @@ from .QwenLM import *
 from .granite import *
 from .WiseCat import *
 from .DeepSeek import *
+from .freeaichat import FreeAIChat
 __all__ = [
     'LLAMA',
     'DeepSeek',
@@ -141,4 +142,5 @@ __all__ = [
     'JadveOpenAI',
     'ChatGLM',
     'NousHermes',
+    'FreeAIChat',
 ]