webscout 8.2.9__py3-none-any.whl → 2026.1.19__py3-none-any.whl

webscout/Provider/StandardInput.py

@@ -1,290 +0,0 @@
-from curl_cffi.requests import Session 
-import uuid
-import re
-from typing import Any, Dict, Optional, Union
-from webscout.AIutel import Optimizers
-from webscout.AIutel import Conversation
-from webscout.AIutel import AwesomePrompts, sanitize_stream # Import sanitize_stream
-from webscout.AIbase import Provider
-from webscout import exceptions
-from webscout.litagent import LitAgent
-
-class StandardInputAI(Provider):
-    """
-    A class to interact with the Standard Input chat API.
-    """
-
-    AVAILABLE_MODELS = {
-        "standard-quick": "quick",
-        "standard-reasoning": "quick",  # Same model but with reasoning enabled
-    }
-
-    def __init__(
-        self,
-        is_conversation: bool = True,
-        max_tokens: int = 2049,
-        timeout: int = 30,
-        intro: str = None,
-        filepath: str = None,
-        update_file: bool = True,
-        proxies: dict = {},
-        history_offset: int = 10250,
-        act: str = None,
-        model: str = "standard-quick",
-        chat_id: str = None,
-        user_id: str = None,
-        browser: str = "chrome",
-        system_prompt: str = "You are a helpful assistant.",
-        enable_reasoning: bool = False,
-    ):
-        """
-        Initializes the Standard Input API client.
-
-        Args:
-            is_conversation (bool): Whether to maintain conversation history.
-            max_tokens (int): Maximum number of tokens to generate.
-            timeout (int): Request timeout in seconds.
-            intro (str): Introduction text for the conversation.
-            filepath (str): Path to save conversation history.
-            update_file (bool): Whether to update the conversation history file.
-            proxies (dict): Proxy configuration for requests.
-            history_offset (int): Maximum history length in characters.
-            act (str): Persona for the AI to adopt.
-            model (str): Model to use, must be one of AVAILABLE_MODELS.
-            chat_id (str): Unique identifier for the chat session.
-            user_id (str): Unique identifier for the user.
-            browser (str): Browser to emulate in requests.
-            system_prompt (str): System prompt for the AI.
-            enable_reasoning (bool): Whether to enable reasoning feature.
-        """
-        if model not in self.AVAILABLE_MODELS:
-            raise ValueError(f"Invalid model: {model}. Choose from: {self.AVAILABLE_MODELS}")
-
-        self.url = "https://chat.standard-input.com/api/chat"
-
-        # Initialize LitAgent for user agent generation
-        self.agent = LitAgent()
-        # Use fingerprinting to create a consistent browser identity
-        self.fingerprint = self.agent.generate_fingerprint(browser)
-        self.system_prompt = system_prompt
-
-        # Use the fingerprint for headers
-        self.headers = {
-            "accept": "*/*",
-            "accept-encoding": "gzip, deflate, br, zstd",
-            "accept-language": self.fingerprint["accept_language"],
-            "content-type": "application/json",
-            "dnt": "1",
-            "origin": "https://chat.standard-input.com",
-            "referer": "https://chat.standard-input.com/",
-            "sec-ch-ua": self.fingerprint["sec_ch_ua"] or '"Microsoft Edge";v="135", "Not-A.Brand";v="8", "Chromium";v="135"',
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": f'"{self.fingerprint["platform"]}"',
-            "sec-fetch-dest": "empty",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-site": "same-origin",
-            "sec-gpc": "1",
-            "user-agent": self.fingerprint["user_agent"],
-        }
-
-        # Default cookies - these should be updated for production use
-        self.cookies = {
-            "auth-chat": '''%7B%22user%22%3A%7B%22id%22%3A%2243a26ebd-7691-4a5a-8321-12aff017af86%22%2C%22email%22%3A%22iu511inmev%40illubd.com%22%2C%22accountId%22%3A%22057d78c9-06db-48eb-aeaa-0efdbaeb9446%22%2C%22provider%22%3A%22password%22%7D%2C%22tokens%22%3A%7B%22access%22%3A%22eyJhbGciOiJFUzI1NiIsImtpZCI6Ijg1NDhmZWY1LTk5MjYtNDk2Yi1hMjI2LTQ5OTExYjllYzU2NSIsInR5cCI6IkpXVCJ9.eyJtb2RlIjoiYWNjZXNzIiwidHlwZSI6InVzZXIiLCJwcm9wZXJ0aWVzIjp7ImlkIjoiNDNhMjZlYmQtNzY5MS00YTVhLTgzMzEtMTJhZmYwMTdhZjg2IiwiZW1haWwiOiJpdTUxMWlubWV2QGlsbHViZC5jb20iLCJhY2NvdW50SWQiOiIwNTdkNzhjOS0wNmRiLTQ4ZWItYWVhYS0wZWZkYmFlYjk0NDYiLCJwcm92aWRlciI6InBhc3N3b3JkIn0sImF1ZCI6InN0YW5kYXJkLWlucHV0LWlvcyIsImlzcyI6Imh0dHBzOi8vYXV0aC5zdGFuZGFyZC1pbnB1dC5jb20iLCJzdWIiOiJ1c2VyOjRmYWMzMTllZjA4MDRiZmMiLCJleHAiOjE3NDU0MDU5MDN9.d3VsEq-UCNsQWkiPlTVw7caS0wTXfCYe6yeFLeb4Ce6ZYTIFFn685SF-aKvLOxaYaq7Pyk4D2qr24riPVhxUWQ%22%2C%22refresh%22%3A%22user%3A4fac319ef0804bfc%3A3a757177-5507-4a36-9356-492f5ed06105%22%7D%7D''',
-            "auth": '''%7B%22user%22%3A%7B%22id%22%3A%22c51e291f-8f44-439d-a38b-9ea147581a13%22%2C%22email%22%3A%22r6cigexlsb%40mrotzis.com%22%2C%22accountId%22%3A%22599fd4ce-04a2-40f6-a78f-d33d0059b77f%22%2C%22provider%22%3A%22password%22%7D%2C%22tokens%22%3A%7B%22access%22%3A%22eyJhbGciOiJFUzI1NiIsImtpZCI6Ijg1NDhmZWY1LTk5MjYtNDk2Yi1hMjI2LTQ5OTExYjllYzU2NSIsInR5cCI6IkpXVCJ9.eyJtb2RlIjoiYWNjZXNzIiwidHlwZSI6InVzZXIiLCJwcm9wZXJ0aWVzIjp7ImlkIjoiYzUxZTI5MWYtOGY0NC00MzlkLWEzOGItOWVhMTQ3NTgxYTEzIiwiZW1haWwiOiJyNmNpZ2V4bHNiQG1yb3R6aXMuY29tIiwiYWNjb3VudElkIjoiNTk5ZmQ0Y2UtMDRhMi00MGY2LWE3OGYtZDMzZDAwNTliNzdmIiwicHJvdmlkZXIiOiJwYXNzd29yZCJ9LCJhdWQiOiJzdGFuZGFyZC1pbnB1dC1pb3MiLCJpc3MiOiJodHRwczovL2F1dGguc3RhbmRhcmQtaW5wdXQuY29tIiwic3ViIjoidXNlcjo4Y2FmMjRkYzUxNDc4MmNkIiwiZXhwIjoxNzQ2NzI0MTU3fQ.a3970nBJkd8JoU-khRA2JlRMuYeJ7378QS4ZL446kOkDi35uTwuC4qGrWH9efk9GkFaVcWPtYeOJjRb7f2SeJA%22%2C%22refresh%22%3A%22user%3A8caf24dc514782cd%3A14e24386-8443-4df0-ae25-234ad59218ef%22%7D%7D''',
-            "sidebar:state": "true",
-            "ph_phc_f3wUUyCfmKlKtkc2pfT7OsdcW2mBEVGN2A87yEYbG3c_posthog": '''%7B%22distinct_id%22%3A%220195c7cc-ac8f-79ff-b901-e14a78fc2a67%22%2C%22%24sesid%22%3A%5B1744688627860%2C%220196377f-9f12-77e6-a9ea-0e9669423803%22%2C1744687832850%5D%2C%22%24initial_person_info%22%3A%7B%22r%22%3A%22%24direct%22%2C%22u%22%3A%22https%3A%2F%2Fstandard-input.com%2F%22%7D%7D'''
-        }
-
-        self.session = Session() # Use curl_cffi Session
-        self.session.headers.update(self.headers)
-        self.session.proxies = proxies # Assign proxies directly
-
-        self.is_conversation = is_conversation
-        self.max_tokens_to_sample = max_tokens
-        self.timeout = timeout
-        self.last_response = {}
-        self.model = model
-        self.chat_id = chat_id or str(uuid.uuid4())
-        self.user_id = user_id or f"user_{str(uuid.uuid4())[:8].upper()}"
-        self.enable_reasoning = enable_reasoning
-
-        self.__available_optimizers = (
-            method
-            for method in dir(Optimizers)
-            if callable(getattr(Optimizers, method)) and not method.startswith("__")
-        )
-        Conversation.intro = (
-            AwesomePrompts().get_act(
-                act, raise_not_found=True, default=None, case_insensitive=True
-            )
-            if act
-            else intro or Conversation.intro
-        )
-
-        self.conversation = Conversation(
-            is_conversation, self.max_tokens_to_sample, filepath, update_file
-        )
-        self.conversation.history_offset = history_offset
-
-    def refresh_identity(self, browser: str = None):
-        """
-        Refreshes the browser identity fingerprint.
-
-        Args:
-            browser: Specific browser to use for the new fingerprint
-        """
-        browser = browser or self.fingerprint.get("browser_type", "chrome")
-        self.fingerprint = self.agent.generate_fingerprint(browser)
-
-        # Update headers with new fingerprint
-        self.headers.update({
-            "Accept-Language": self.fingerprint["accept_language"],
-            "Sec-CH-UA": self.fingerprint["sec_ch_ua"] or self.headers["sec-ch-ua"],
-            "Sec-CH-UA-Platform": f'"{self.fingerprint["platform"]}"',
-            "User-Agent": self.fingerprint["user_agent"],
-        })
-
-        # Update session headers
-        for header, value in self.headers.items():
-            self.session.headers[header] = value
-
-        return self.fingerprint
-
-    @staticmethod
-    def _standardinput_extractor(chunk: Union[str, Dict[str, Any]]) -> Optional[str]:
-        """Extracts content from the StandardInput stream format '0:"..."'."""
-        if isinstance(chunk, str):
-            match = re.search(r'0:"(.*?)"(?=,|$)', chunk) # Look for 0:"...", possibly followed by comma or end of string
-            if match:
-                # Decode potential unicode escapes like \u00e9 and handle escaped quotes/backslashes
-                content = match.group(1).encode().decode('unicode_escape')
-                return content.replace('\\\\', '\\').replace('\\"', '"')
-        return None
-
-    def ask(
-        self,
-        prompt: str,
-        optimizer: str = None,
-        conversationally: bool = False,
-    ) -> Dict[str, Any]:
-        conversation_prompt = self.conversation.gen_complete_prompt(prompt)
-        if optimizer:
-            if optimizer in self.__available_optimizers:
-                conversation_prompt = getattr(Optimizers, optimizer)(
-                    conversation_prompt if conversationally else prompt
-                )
-            else:
-                raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
-
-        # Prepare the messages
-        messages = [
-            {"role": "system", "content": self.system_prompt},
-            {"role": "user", "content": conversation_prompt, "parts": [{"type": "text", "text": conversation_prompt}]}
-        ]
-
-        # Prepare the request payload
-        payload = {
-            "id": self.chat_id,
-            "messages": messages,
-            "modelId": self.AVAILABLE_MODELS[self.model],
-            "enabledFeatures": ["reasoning"] if self.enable_reasoning or self.model == "standard-reasoning" else []
-        }
-
-        try:
-            # Use curl_cffi post with impersonate
-            response = self.session.post(
-                self.url,
-                cookies=self.cookies,
-                json=payload,
-                stream=True,
-                timeout=self.timeout,
-                impersonate="chrome120" # Add impersonate
-            )
-
-            if response.status_code != 200:
-                try:
-                    error_content = response.text
-                except:
-                    error_content = "<could not read response content>"
-
-                if response.status_code in [403, 429]:
-                    self.refresh_identity()
-                    response = self.session.post(
-                        self.url, cookies=self.cookies, json=payload, stream=True,
-                        timeout=self.timeout, impersonate="chrome120"
-                    )
-                    if not response.ok:
-                        raise exceptions.FailedToGenerateResponseError(
-                            f"Failed to generate response after identity refresh - ({response.status_code}, {response.reason}) - {error_content}"
-                        )
-                else:
-                    raise exceptions.FailedToGenerateResponseError(
-                        f"Request failed with status code {response.status_code}. Response: {error_content}"
-                    )
-
-            full_response = ""
-            # Use sanitize_stream
-            processed_stream = sanitize_stream(
-                data=response.iter_content(chunk_size=None), # Pass byte iterator
-                intro_value=None, # No simple prefix
-                to_json=False,    # Content is not JSON
-                content_extractor=self._standardinput_extractor # Use the specific extractor
-            )
-            for content_chunk in processed_stream:
-                if content_chunk and isinstance(content_chunk, str):
-                    full_response += content_chunk
-
-            self.last_response = {"text": full_response}
-            self.conversation.update_chat_history(prompt, full_response)
-            return {"text": full_response}
-        except Exception as e:
-            raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
-
-    def chat(
-        self,
-        prompt: str,
-        optimizer: str = None,
-        conversationally: bool = False,
-    ) -> str:
-        return self.get_message(
-            self.ask(
-                prompt, optimizer=optimizer, conversationally=conversationally
-            )
-        )
-
-    def get_message(self, response: dict) -> str:
-        assert isinstance(response, dict), "Response should be of dict data-type only"
-        # Extractor handles formatting
-        return response.get("text", "").replace('\\n', '\n').replace('\\n\\n', '\n\n')
-
-if __name__ == "__main__":
-    print("-" * 100)
-    print(f"{'Model':<50} {'Status':<10} {'Response'}")
-    print("-" * 100)
-
-    test_prompt = "Say 'Hello' in one word"
-
-    # Test each model
-    for model in StandardInputAI.AVAILABLE_MODELS:
-        print(f"\rTesting {model}...", end="")
-
-        try:
-            test_ai = StandardInputAI(model=model, timeout=120)  # Increased timeout
-            response = test_ai.chat(test_prompt)
-
-            if response and len(response.strip()) > 0:
-                status = "✓"
-                # Clean and truncate response
-                clean_text = response.strip().encode('utf-8', errors='ignore').decode('utf-8')
-                display_text = clean_text[:50] + "..." if len(clean_text) > 50 else clean_text
-            else:
-                status = "✗"
-                display_text = "Empty or invalid response"
-
-            print(f"\r{model:<50} {status:<10} {display_text}")
-        except Exception as e:
-            error_msg = str(e)
-            # Truncate very long error messages
-            if len(error_msg) > 100:
-                error_msg = error_msg[:97] + "..."
-            print(f"\r{model:<50} {'✗':<10} Error: {error_msg}")

webscout/Provider/TTI/aiarta.py

@@ -1,365 +0,0 @@
-"""AIArtaImager TTI-Compatible Provider - Generate stunning AI art with AI Arta! 🎨
-
-Examples:
-    >>> from webscout.Provider.TTI.aiarta import AIArta
-    >>> client = AIArta()
-    >>> response = client.images.create(
-    ...     model="flux",
-    ...     prompt="A cool cyberpunk city at night",
-    ...     n=1
-    ... )
-    >>> print(response)
-"""
-
-import requests
-from typing import Optional, List, Dict, Any
-from webscout.Provider.TTI.utils import ImageData, ImageResponse
-from webscout.Provider.TTI.base import TTICompatibleProvider, BaseImages
-from io import BytesIO
-import os
-import tempfile
-from webscout.litagent import LitAgent
-import time
-import json
-
-try:
-    from PIL import Image
-except ImportError:
-    Image = None
-
-class Images(BaseImages):
-    def __init__(self, client: 'AIArta'):
-        self._client = client
-
-    def create(
-        self,
-        *,
-        model: str,
-        prompt: str,
-        n: int = 1,
-        size: str = "1024x1024",
-        response_format: str = "url",
-        user: Optional[str] = None,
-        style: str = "none",
-        aspect_ratio: str = "1:1",
-        timeout: int = 60,
-        image_format: str = "png",
-        **kwargs
-    ) -> ImageResponse:
-        """
-        image_format: "png" or "jpeg"
-        """
-        if Image is None:
-            raise ImportError("Pillow (PIL) is required for image format conversion.")
-
-        images = []
-        urls = []
-        agent = LitAgent()
-
-        def upload_file_with_retry(img_bytes, image_format, max_retries=3):
-            ext = "jpg" if image_format.lower() == "jpeg" else "png"
-            for attempt in range(max_retries):
-                tmp_path = None
-                try:
-                    with tempfile.NamedTemporaryFile(suffix=f".{ext}", delete=False) as tmp:
-                        tmp.write(img_bytes)
-                        tmp.flush()
-                        tmp_path = tmp.name
-                    with open(tmp_path, 'rb') as f:
-                        files = {
-                            'fileToUpload': (f'image.{ext}', f, f'image/{ext}')
-                        }
-                        data = {
-                            'reqtype': 'fileupload',
-                            'json': 'true'
-                        }
-                        headers = {'User-Agent': agent.random()}
-                        if attempt > 0:
-                            headers['Connection'] = 'close'
-                        resp = requests.post("https://catbox.moe/user/api.php", files=files, data=data, headers=headers, timeout=timeout)
-                        if resp.status_code == 200 and resp.text.strip():
-                            text = resp.text.strip()
-                            if text.startswith('http'):
-                                return text
-                            try:
-                                result = resp.json()
-                                if "url" in result:
-                                    return result["url"]
-                            except json.JSONDecodeError:
-                                if 'http' in text:
-                                    return text
-                except Exception:
-                    if attempt < max_retries - 1:
-                        time.sleep(1 * (attempt + 1))
-                finally:
-                    if tmp_path and os.path.isfile(tmp_path):
-                        try:
-                            os.remove(tmp_path)
-                        except Exception:
-                            pass
-            return None
-
-        def upload_file_alternative(img_bytes, image_format):
-            try:
-                ext = "jpg" if image_format.lower() == "jpeg" else "png"
-                with tempfile.NamedTemporaryFile(suffix=f".{ext}", delete=False) as tmp:
-                    tmp.write(img_bytes)
-                    tmp.flush()
-                    tmp_path = tmp.name
-                try:
-                    if not os.path.isfile(tmp_path):
-                        return None
-                    with open(tmp_path, 'rb') as img_file:
-                        files = {'file': img_file}
-                        response = requests.post('https://0x0.st', files=files)
-                        response.raise_for_status()
-                        image_url = response.text.strip()
-                        if not image_url.startswith('http'):
-                            return None
-                        return image_url
-                except Exception:
-                    return None
-                finally:
-                    try:
-                        os.remove(tmp_path)
-                    except Exception:
-                        pass
-            except Exception:
-                return None
-
-        for _ in range(n):
-            # Step 1: Get Authentication Token
-            auth_data = self._client.read_and_refresh_token()
-            gen_headers = {
-                "Authorization": auth_data.get("idToken"),
-            }
-            # Remove content-type header for form data
-            if "content-type" in self._client.session.headers:
-                del self._client.session.headers["content-type"]
-            # get_model now returns the proper style name from model_aliases
-            style_value = self._client.get_model(model)
-            image_payload = {
-                "prompt": str(prompt),
-                "negative_prompt": str(kwargs.get("negative_prompt", "blurry, deformed hands, ugly")),
-                "style": str(style_value),
-                "images_num": str(1),  # Generate one image at a time in the loop
-                "cfg_scale": str(kwargs.get("guidance_scale", 7)),
-                "steps": str(kwargs.get("num_inference_steps", 30)),
-                "aspect_ratio": str(aspect_ratio),
-            }
-            # Step 2: Generate Image (send as form data, not JSON)
-            image_response = self._client.session.post(
-                self._client.image_generation_url,
-                data=image_payload,  # Use form data instead of JSON
-                headers=gen_headers,
-                timeout=timeout
-            )
-            if image_response.status_code != 200:
-                raise RuntimeError(f"AIArta API error {image_response.status_code}: {image_response.text}\nPayload: {image_payload}")
-            image_data = image_response.json()
-            record_id = image_data.get("record_id")
-            if not record_id:
-                raise RuntimeError(f"Failed to initiate image generation: {image_data}")
-            # Step 3: Check Generation Status
-            status_url = self._client.status_check_url.format(record_id=record_id)
-            while True:
-                status_response = self._client.session.get(
-                    status_url,
-                    headers=gen_headers,
-                    timeout=timeout
-                )
-                status_data = status_response.json()
-                status = status_data.get("status")
-                if status == "DONE":
-                    image_urls = [image["url"] for image in status_data.get("response", [])]
-                    if not image_urls:
-                        raise RuntimeError("No image URLs returned from AIArta")
-                    img_resp = self._client.session.get(image_urls[0], timeout=timeout)
-                    img_resp.raise_for_status()
-                    img_bytes = img_resp.content
-                    # Convert to png or jpeg in memory
-                    with BytesIO(img_bytes) as input_io:
-                        with Image.open(input_io) as im:
-                            out_io = BytesIO()
-                            if image_format.lower() == "jpeg":
-                                im = im.convert("RGB")
-                                im.save(out_io, format="JPEG")
-                            else:
-                                im.save(out_io, format="PNG")
-                            img_bytes = out_io.getvalue()
-                    images.append(img_bytes)
-                    if response_format == "url":
-                        uploaded_url = upload_file_with_retry(img_bytes, image_format)
-                        if not uploaded_url:
-                            uploaded_url = upload_file_alternative(img_bytes, image_format)
-                        if uploaded_url:
-                            urls.append(uploaded_url)
-                        else:
-                            raise RuntimeError("Failed to upload image to catbox.moe using all available methods")
-                    break
-                elif status in ("IN_QUEUE", "IN_PROGRESS"):
-                    time.sleep(2)
-                else:
-                    raise RuntimeError(f"Image generation failed with status: {status}")
-
-        result_data = []
-        if response_format == "url":
-            for url in urls:
-                result_data.append(ImageData(url=url))
-        elif response_format == "b64_json":
-            import base64
-            for img in images:
-                b64 = base64.b64encode(img).decode("utf-8")
-                result_data.append(ImageData(b64_json=b64))
-        else:
-            raise ValueError("response_format must be 'url' or 'b64_json'")
-
-        from time import time as _time
-        return ImageResponse(
-            created=int(_time()),
-            data=result_data
-        )
-
-class AIArta(TTICompatibleProvider):
-    # Model aliases mapping from lowercase keys to proper API style names
-    model_aliases = {
-        "flux": "Flux",
-        "medieval": "Medieval",
-        "vincent_van_gogh": "Vincent Van Gogh",
-        "f_dev": "F Dev",
-        "low_poly": "Low Poly",
-        "dreamshaper_xl": "Dreamshaper-xl",
-        "anima_pencil_xl": "Anima-pencil-xl",
-        "biomech": "Biomech",
-        "trash_polka": "Trash Polka",
-        "no_style": "No Style",
-        "cheyenne_xl": "Cheyenne-xl",
-        "chicano": "Chicano",
-        "embroidery_tattoo": "Embroidery tattoo",
-        "red_and_black": "Red and Black",
-        "fantasy_art": "Fantasy Art",
-        "watercolor": "Watercolor",
-        "dotwork": "Dotwork",
-        "old_school_colored": "Old school colored",
-        "realistic_tattoo": "Realistic tattoo",
-        "japanese_2": "Japanese_2",
-        "realistic_stock_xl": "Realistic-stock-xl",
-        "f_pro": "F Pro",
-        "revanimated": "RevAnimated",
-        "katayama_mix_xl": "Katayama-mix-xl",
-        "sdxl_l": "SDXL L",
-        "cor_epica_xl": "Cor-epica-xl",
-        "anime_tattoo": "Anime tattoo",
-        "new_school": "New School",
-        "death_metal": "Death metal",
-        "old_school": "Old School",
-        "juggernaut_xl": "Juggernaut-xl",
-        "photographic": "Photographic",
-        "sdxl_1_0": "SDXL 1.0",
-        "graffiti": "Graffiti",
-        "mini_tattoo": "Mini tattoo",
-        "surrealism": "Surrealism",
-        "neo_traditional": "Neo-traditional",
-        "on_limbs_black": "On limbs black",
-        "yamers_realistic_xl": "Yamers-realistic-xl",
-        "pony_xl": "Pony-xl",
-        "playground_xl": "Playground-xl",
-        "anything_xl": "Anything-xl",
-        "flame_design": "Flame design",
-        "kawaii": "Kawaii",
-        "cinematic_art": "Cinematic Art",
-        "professional": "Professional",
-        "black_ink": "Black Ink"
-    }
-    
-    AVAILABLE_MODELS = list(model_aliases.keys())
-    default_model = "Flux"
-    default_image_model = default_model
-
-    def __init__(self):
-        self.image_generation_url = "https://img-gen-prod.ai-arta.com/api/v1/text2image"
-        self.status_check_url = "https://img-gen-prod.ai-arta.com/api/v1/text2image/{record_id}/status"
-        self.auth_url = "https://www.googleapis.com/identitytoolkit/v3/relyingparty/signupNewUser?key=AIzaSyB3-71wG0fIt0shj0ee4fvx1shcjJHGrrQ"
-        self.token_refresh_url = "https://securetoken.googleapis.com/v1/token?key=AIzaSyB3-71wG0fIt0shj0ee4fvx1shcjJHGrrQ"
-        self.session = requests.Session()
-        self.user_agent = LitAgent().random()
-        self.headers = {
-            "accept": "application/json",
-            "accept-language": "en-US,en;q=0.9",
-            "origin": "https://img-gen-prod.ai-arta.com",
-            "referer": "https://img-gen-prod.ai-arta.com/",
-            "user-agent": self.user_agent,
-        }
-        self.session.headers.update(self.headers)
-        self.images = Images(self)
-
-    def get_auth_file(self) -> str:
-        path = os.path.join(os.path.expanduser("~"), ".ai_arta_cookies")
-        if not os.path.exists(path):
-            os.makedirs(path)
-        filename = f"auth_{self.__class__.__name__}.json"
-        return os.path.join(path, filename)
-
-    def create_token(self, path: str) -> Dict[str, Any]:
-        auth_payload = {"clientType": "CLIENT_TYPE_ANDROID"}
-        proxies = self.session.proxies if self.session.proxies else None
-        auth_response = self.session.post(self.auth_url, json=auth_payload, timeout=60, proxies=proxies)
-        auth_data = auth_response.json()
-        auth_token = auth_data.get("idToken")
-        if not auth_token:
-            raise Exception("Failed to obtain authentication token.")
-        with open(path, 'w') as f:
-            json.dump(auth_data, f)
-        return auth_data
-
-    def refresh_token(self, refresh_token: str) -> tuple[str, str]:
-        payload = {
-            "grant_type": "refresh_token",
-            "refresh_token": refresh_token,
-        }
-        response = self.session.post(self.token_refresh_url, data=payload, timeout=60)
-        response_data = response.json()
-        return response_data.get("id_token"), response_data.get("refresh_token")
-
-    def read_and_refresh_token(self) -> Dict[str, Any]:
-        path = self.get_auth_file()
-        if os.path.isfile(path):
-            with open(path, 'r') as f:
-                auth_data = json.load(f)
-            diff = time.time() - os.path.getmtime(path)
-            expires_in = int(auth_data.get("expiresIn", 3600))
-            if diff < expires_in:
-                if diff > expires_in / 2:
-                    auth_data["idToken"], auth_data["refreshToken"] = self.refresh_token(
-                        auth_data.get("refreshToken")
-                    )
-                    with open(path, 'w') as f:
-                        json.dump(auth_data, f)
-                return auth_data
-        return self.create_token(path)
-
-    def get_model(self, model_name: str) -> str:
-        # Convert to lowercase for lookup
-        model_key = model_name.lower()
-        # Return the proper style name from model_aliases, or the original if not found
-        return self.model_aliases.get(model_key, model_name)
-
-    @property
-    def models(self):
-        class _ModelList:
-            def list(inner_self):
-                return type(self).AVAILABLE_MODELS
-        return _ModelList()
-
-# Example usage:
-if __name__ == "__main__":
-    from rich import print
-    client = AIArta()
-    response = client.images.create(
-        model="flux",
-        prompt="a white siamese cat",
-        response_format="url",
-        n=2,
-        timeout=30,
-    )
-    print(response)