webscout 7.5__py3-none-any.whl → 7.6__py3-none-any.whl

webscout/Provider/typegpt.py

@@ -15,43 +15,64 @@ class TypeGPT(Provider):
     A class to interact with the TypeGPT.net API. Improved to match webscout standards.
     """
     url = "https://chat.typegpt.net"
-    working = True
-    supports_message_history = True
 
-    models = [
+    AVAILABLE_MODELS = [
         # OpenAI Models
         "gpt-3.5-turbo",
-        "chatgpt-4o-latest",
         "gpt-3.5-turbo-202201",
         "gpt-4o",
         "gpt-4o-2024-05-13",
+        "gpt-4o-2024-11-20",
+        "gpt-4o-mini",
+        "gpt-4o-mini-2024-07-18",
+        # "gpt-4o-mini-ddg", >>>> NOT WORKING
+        "o1",
+        # "o1-mini-2024-09-12", >>>> NOT WORKING
         "o1-preview",
+        "o3-mini",
+        "chatgpt-4o-latest",
         
         # Claude Models
-        "claude",
+        # "claude", >>>> NOT WORKING
         "claude-3-5-sonnet",
-        "claude-sonnet-3.5",
         "claude-3-5-sonnet-20240620",
+        "claude-3-5-sonnet-x",
+        # "claude-3-haiku-ddg", >>>> NOT WORKING
+        "claude-hybridspace",
+        "claude-sonnet-3.5",
+        "Claude-sonnet-3.7",
+        "anthropic/claude-3.5-sonnet",
+        "anthropic/claude-3.7-sonnet",
         
         # Meta/LLaMA Models
         "@cf/meta/llama-2-7b-chat-fp16",
         "@cf/meta/llama-2-7b-chat-int8",
         "@cf/meta/llama-3-8b-instruct",
         "@cf/meta/llama-3.1-8b-instruct",
-        "@cf/meta-llama/llama-2-7b-chat-hf-lora",
+        "@cf/meta/llama-3.3-70b-instruct-fp8-fast",
+        # "@cf/meta-llama/llama-2-7b-chat-hf-lora", >>>> NOT WORKING
         "llama-3.1-405b",
         "llama-3.1-70b",
+        # "llama-3.1-70b-ddg", >>>> NOT WORKING
         "llama-3.1-8b",
-        "meta-llama/Llama-2-7b-chat-hf",
-        "meta-llama/Llama-3.1-70B-Instruct",
-        "meta-llama/Llama-3.1-8B-Instruct",
+        # "llama-scaleway", >>>> NOT WORKING
+        "llama3.1-8b", # >>>> NOT WORKING
+        "llama3.3-70b",
+        # "llamalight", >>>> NOT WORKING
+        "Meta-Llama-3.1-405B-Instruct-Turbo",
+        "Meta-Llama-3.3-70B-Instruct-Turbo",
+        # "meta-llama/Llama-2-7b-chat-hf", >>>> NOT WORKING
+        # "meta-llama/Llama-3.1-70B-Instruct", >>>> NOT WORKING
+        # "meta-llama/Llama-3.1-8B-Instruct", >>>> NOT WORKING
         "meta-llama/Llama-3.2-11B-Vision-Instruct",
-        "meta-llama/Llama-3.2-1B-Instruct",
-        "meta-llama/Llama-3.2-3B-Instruct",
+        # "meta-llama/Llama-3.2-1B-Instruct", >>>> NOT WORKING
+        # "meta-llama/Llama-3.2-3B-Instruct", >>>> NOT WORKING
         "meta-llama/Llama-3.2-90B-Vision-Instruct",
-        "meta-llama/Llama-Guard-3-8B",
-        "meta-llama/Meta-Llama-3-70B-Instruct",
-        "meta-llama/Meta-Llama-3-8B-Instruct",
+        "meta-llama/Llama-3.3-70B-Instruct",
+        "meta-llama/Llama-3.3-70B-Instruct-Turbo",
+        # "meta-llama/Llama-Guard-3-8B", >>>> NOT WORKING
+        # "meta-llama/Meta-Llama-3-70B-Instruct", >>>> NOT WORKING
+        # "meta-llama/Meta-Llama-3-8B-Instruct", >>>> NOT WORKING
         "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
         "meta-llama/Meta-Llama-3.1-8B-Instruct",
         "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
@@ -60,25 +81,34 @@ class TypeGPT(Provider):
         "mistral",
         "mistral-large",
         "@cf/mistral/mistral-7b-instruct-v0.1",
-        "@cf/mistral/mistral-7b-instruct-v0.2-lora",
+        # "@cf/mistral/mistral-7b-instruct-v0.2-lora", >>>> NOT WORKING
         "@hf/mistralai/mistral-7b-instruct-v0.2",
         "mistralai/Mistral-7B-Instruct-v0.2",
         "mistralai/Mistral-7B-Instruct-v0.3",
         "mistralai/Mixtral-8x22B-Instruct-v0.1",
         "mistralai/Mixtral-8x7B-Instruct-v0.1",
+        # "mixtral-8x7b-ddg", >>>> NOT WORKING
+        "Mistral-7B-Instruct-v0.2",
         
         # Qwen Models
         "@cf/qwen/qwen1.5-0.5b-chat",
         "@cf/qwen/qwen1.5-1.8b-chat",
-        "@cf/qwen/qwen1.5-7b-chat-awq",
         "@cf/qwen/qwen1.5-14b-chat-awq",
+        "@cf/qwen/qwen1.5-7b-chat-awq",
         "Qwen/Qwen2.5-3B-Instruct",
         "Qwen/Qwen2.5-72B-Instruct",
         "Qwen/Qwen2.5-Coder-32B-Instruct",
+        "Qwen/Qwen2-72B-Instruct",
+        "Qwen/QwQ-32B",
+        "Qwen/QwQ-32B-Preview",
+        "Qwen2.5-72B-Instruct",
+        "qwen",
+        "qwen-coder",
+        # "Qwen-QwQ-32B-Preview", >>>> NOT WORKING
         
         # Google/Gemini Models
-        "@cf/google/gemma-2b-it-lora",
-        "@cf/google/gemma-7b-it-lora",
+        # "@cf/google/gemma-2b-it-lora", >>>> NOT WORKING
+        # "@cf/google/gemma-7b-it-lora", >>>> NOT WORKING
         "@hf/google/gemma-7b-it",
         "google/gemma-1.1-2b-it",
         "google/gemma-1.1-7b-it",
@@ -86,22 +116,8 @@ class TypeGPT(Provider):
         "gemini-1.5-pro",
         "gemini-1.5-pro-latest",
         "gemini-1.5-flash",
-        
-        # Cohere Models
-        "c4ai-aya-23-35b",
-        "c4ai-aya-23-8b",
-        "command",
-        "command-light",
-        "command-light-nightly",
-        "command-nightly",
-        "command-r",
-        "command-r-08-2024",
-        "command-r-plus",
-        "command-r-plus-08-2024",
-        "rerank-english-v2.0",
-        "rerank-english-v3.0",
-        "rerank-multilingual-v2.0",
-        "rerank-multilingual-v3.0",
+        "gemini-flash-2.0",
+        "gemini-thinking",
         
         # Microsoft Models
         "@cf/microsoft/phi-2",
@@ -109,29 +125,45 @@ class TypeGPT(Provider):
         "microsoft/Phi-3-medium-4k-instruct",
         "microsoft/Phi-3-mini-4k-instruct",
         "microsoft/Phi-3.5-mini-instruct",
+        "microsoft/phi-4",
         "microsoft/WizardLM-2-8x22B",
         
         # Yi Models
         "01-ai/Yi-1.5-34B-Chat",
-        "01-ai/Yi-34B-Chat",
+        # "01-ai/Yi-34B-Chat", >>>> NOT WORKING
         
-        # Specialized Models and Tools
+        # DeepSeek Models
         "@cf/deepseek-ai/deepseek-math-7b-base",
         "@cf/deepseek-ai/deepseek-math-7b-instruct",
+        "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",
+        "deepseek",
+        "deepseek-ai/DeepSeek-R1",
+        "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
+        # "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", >>>> NOT WORKING
+        # "deepseek-ai/DeepSeek-V2.5", >>>> NOT WORKING
+        "deepseek-llm-67b-chat",
+        "deepseek-r1",
+        "deepseek-r1-distill-llama-70b",
+        # "deepseek-reasoner", >>>> NOT WORKING
+        "deepseek-v3",
+        
+        # Specialized Models and Tools
         "@cf/defog/sqlcoder-7b-2",
-        "@cf/openchat/openchat-3.5-0106",
         "@cf/thebloke/discolm-german-7b-v1-awq",
         "@cf/tiiuae/falcon-7b-instruct",
-        "@cf/tinyllama/tinyllama-1.1b-chat-v1.0",
-        "@hf/nexusflow/starling-lm-7b-beta",
-        "@hf/nousresearch/hermes-2-pro-mistral-7b",
-        "@hf/thebloke/deepseek-coder-6.7b-base-awq",
-        "@hf/thebloke/deepseek-coder-6.7b-instruct-awq",
-        "@hf/thebloke/llama-2-13b-chat-awq",
-        "@hf/thebloke/llamaguard-7b-awq",
-        "@hf/thebloke/neural-chat-7b-v3-1-awq",
-        "@hf/thebloke/openhermes-2.5-mistral-7b-awq",
-        "@hf/thebloke/zephyr-7b-beta-awq",
+        # "@cf/tinyllama/tinyllama-1.1b-chat-v1.0", >>>> NOT WORKING
+        # "@hf/nexusflow/starling-lm-7b-beta", >>>> NOT WORKING
+        # "@hf/nousresearch/hermes-2-pro-mistral-7b", >>>> NOT WORKING
+        # "@hf/thebloke/deepseek-coder-6.7b-base-awq", >>>> NOT WORKING
+        # "@hf/thebloke/deepseek-coder-6.7b-instruct-awq", >>>> NOT WORKING
+        # "@hf/thebloke/llama-2-13b-chat-awq", >>>> NOT WORKING
+        # "@hf/thebloke/llamaguard-7b-awq", >>>> NOT WORKING
+        # "@hf/thebloke/mistral-7b-instruct-v0.1-awq", >>>> NOT WORKING
+        # "@hf/thebloke/neural-chat-7b-v3-1-awq", >>>> NOT WORKING
+        # "@hf/thebloke/openhermes-2.5-mistral-7b-awq", >>>> NOT WORKING
+        # "@hf/thebloke/zephyr-7b-beta-awq", >>>> NOT WORKING
+        
+        # Development Agents
         "AndroidDeveloper",
         "AngularJSAgent",
         "AzureAgent",
@@ -150,8 +182,6 @@ class TypeGPT(Provider):
         "GodotAgent",
         "GoogleCloudAgent",
         "HTMLAgent",
-        "HerokuAgent",
-        "ImageGeneration",
         "JavaAgent",
         "JavaScriptAgent",
         "MongoDBAgent",
@@ -162,18 +192,54 @@ class TypeGPT(Provider):
         "RepoMap",
         "SwiftDeveloper",
         "XcodeAgent",
-        "YoutubeAgent",
+        # "YoutubeAgent", >>>> NOT WORKING
+        
+        # Other Models
         "blackboxai",
         "blackboxai-pro",
         "builderAgent",
-        "dify",
+        # "Cipher-20b", >>>> NOT WORKING
+        # "dify", >>>> NOT WORKING
         "flux",
-        "openchat/openchat-3.6-8b",
-        "rtist",
-        "searchgpt",
-        "sur",
-        "sur-mistral",
-        "unity"
+        # "flux-1-schnell", >>>> NOT WORKING
+        # "HelpingAI-15B", >>>> NOT WORKING
+        # "HelpingAI2-3b", >>>> NOT WORKING
+        # "HelpingAI2-6B", >>>> NOT WORKING
+        # "HelpingAI2-9B", >>>> NOT WORKING
+        # "HelpingAI2.5-10B", >>>> NOT WORKING
+        # "Helpingai2.5-10b-1m", >>>> NOT WORKING
+        # "HelpingAI2.5-2B", >>>> NOT WORKING
+        # "HELVETE", >>>> NOT WORKING
+        # "HELVETE-X", >>>> NOT WORKING
+        # "evil", >>>> NOT WORKING
+        # "Image-Generator", >>>> NOT WORKING
+        # "Image-Generator-NSFW", >>>> NOT WORKING
+        # "midijourney", >>>> NOT WORKING
+        # "Niansuh", >>>> NOT WORKING
+        # "niansuh-t1", >>>> NOT WORKING
+        # "Nous-Hermes-2-Mixtral-8x7B-DPO", >>>> NOT WORKING
+        # "NousResearch/Hermes-3-Llama-3.1-8B", >>>> NOT WORKING
+        # "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO", >>>> NOT WORKING
+        # "nvidia/Llama-3.1-Nemotron-70B-Instruct", >>>> NOT WORKING
+        # "openai", >>>> NOT WORKING
+        # "openai-audio", >>>> NOT WORKING
+        # "openai-large", >>>> NOT WORKING
+        # "openai-reasoning", >>>> NOT WORKING
+        # "openai/whisper-large-v3", >>>> NOT WORKING
+        # "openai/whisper-large-v3-turbo", >>>> NOT WORKING
+        # "openbmb/MiniCPM-Llama3-V-2_5", >>>> NOT WORKING
+        # "openchat/openchat-3.6-8b", >>>> NOT WORKING
+        # "p1", >>>> NOT WORKING
+        # "phi", >>>> NOT WORKING
+        # "Phi-4-multilmodal-instruct", >>>> NOT WORKING
+        # "Priya-3B", >>>> NOT WORKING
+        # "rtist", >>>> NOT WORKING
+        # "searchgpt", >>>> NOT WORKING
+        # "sur", >>>> NOT WORKING
+        # "sur-mistral", >>>> NOT WORKING
+        # "tiiuae/falcon-7b-instruct", >>>> NOT WORKING
+        # "TirexAi", >>>> NOT WORKING
+        # "unity", >>>> NOT WORKING
     ]
 
     def __init__(
@@ -195,8 +261,8 @@ class TypeGPT(Provider):
         top_p: float = 1,
     ):
         """Initializes the TypeGPT API client."""
-        if model not in self.models:
-            raise ValueError(f"Invalid model: {model}. Choose from: {', '.join(self.models)}")
+        if model not in self.AVAILABLE_MODELS:
+            raise ValueError(f"Invalid model: {model}. Choose from: {', '.join(self.AVAILABLE_MODELS)}")
 
         self.session = requests.Session()
         self.is_conversation = is_conversation
@@ -354,7 +420,31 @@ class TypeGPT(Provider):
             raise TypeError("Invalid response type. Expected str or dict.")
 
 if __name__ == "__main__":
-    ai = TypeGPT(model="chatgpt-4o-latest")
-    response = ai.chat("hi", stream=True)
-    for chunks in response:
-        print(chunks, end="", flush=True)
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
+    
+    # Test all available models
+    working = 0
+    total = len(TypeGPT.AVAILABLE_MODELS)
+    
+    for model in TypeGPT.AVAILABLE_MODELS:
+        try:
+            test_ai = TypeGPT(model=model, timeout=60)
+            response = test_ai.chat("Say 'Hello' in one word", stream=True)
+            response_text = ""
+            for chunk in response:
+                response_text += chunk
+                print(f"\r{model:<50} {'Testing...':<10}", end="", flush=True)
+            
+            if response_text and len(response_text.strip()) > 0:
+                status = "✓"
+                # Truncate response if too long
+                display_text = response_text.strip()[:50] + "..." if len(response_text.strip()) > 50 else response_text.strip()
+            else:
+                status = "✗"
+                display_text = "Empty or invalid response"
+            print(f"\r{model:<50} {status:<10} {display_text}")
+        except Exception as e:
+            print(f"\r{model:<50} {'✗':<10} {str(e)}")
+    

webscout/Provider/x0gpt.py

@@ -24,6 +24,7 @@ class X0GPT(Provider):
         >>> print(response)
         'The weather today is sunny with a high of 75°F.'
     """
+    AVAILABLE_MODELS = ["UNKNOWN"]
 
     def __init__(
         self,
@@ -36,7 +37,8 @@ class X0GPT(Provider):
         proxies: dict = {},
         history_offset: int = 10250,
         act: str = None,
-        system_prompt: str = "You are a helpful assistant."
+        system_prompt: str = "You are a helpful assistant.",
+        model: str = "UNKNOWN"
     ):
         """
         Initializes the X0GPT API with given parameters.

webscout/Provider/yep.py

@@ -38,7 +38,8 @@ class YEPCHAT(Provider):
         act: str = None,
         model: str = "DeepSeek-R1-Distill-Qwen-32B",
         temperature: float = 0.6,
-        top_p: float = 0.7
+        top_p: float = 0.7,
+        browser: str = "chrome"
     ):
         """
         Initializes the YEPCHAT provider with the specified parameters.
@@ -69,20 +70,25 @@ class YEPCHAT(Provider):
 
         # Initialize LitAgent for user agent generation
         self.agent = LitAgent()
+        # Use fingerprinting to create a consistent browser identity
+        self.fingerprint = self.agent.generate_fingerprint(browser)
 
+        # Use the fingerprint for headers
         self.headers = {
-            "Accept": "*/*",
+            "Accept": self.fingerprint["accept"],
             "Accept-Encoding": "gzip, deflate, br, zstd",
-            "Accept-Language": "en-US,en;q=0.9,en-IN;q=0.8",
+            "Accept-Language": self.fingerprint["accept_language"],
             "Content-Type": "application/json; charset=utf-8",
             "DNT": "1",
             "Origin": "https://yep.com",
             "Referer": "https://yep.com/",
-            "Sec-CH-UA": '"Not)A;Brand";v="99", "Microsoft Edge";v="127", "Chromium";v="127"',
+            "Sec-CH-UA": self.fingerprint["sec_ch_ua"] or '"Not)A;Brand";v="99", "Microsoft Edge";v="127", "Chromium";v="127"',
             "Sec-CH-UA-Mobile": "?0",
-            "Sec-CH-UA-Platform": '"Windows"',
-            "User-Agent": self.agent.random(),  # Use LitAgent to generate a random user agent
+            "Sec-CH-UA-Platform": f'"{self.fingerprint["platform"]}"',
+            "User-Agent": self.fingerprint["user_agent"],
         }
+        
+        # Create session cookies with unique identifiers
         self.cookies = {"__Host-session": uuid.uuid4().hex, '__cf_bm': uuid.uuid4().hex}
 
         self.__available_optimizers = (
@@ -101,8 +107,38 @@ class YEPCHAT(Provider):
         )
         self.conversation.history_offset = history_offset
         self.session.proxies = proxies
+        
+        # Set consistent headers for the scraper session
+        for header, value in self.headers.items():
+            self.session.headers[header] = value
 
-        self.knowledge_cutoff = "December 2023"
+    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": self.fingerprint["accept"],
+            "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
+        
+        # Generate new cookies
+        self.cookies = {"__Host-session": uuid.uuid4().hex, '__cf_bm': uuid.uuid4().hex}
+        
+        return self.fingerprint
 
     def ask(
         self,
@@ -147,9 +183,20 @@ class YEPCHAT(Provider):
             try:
                 with self.session.post(self.chat_endpoint, headers=self.headers, cookies=self.cookies, json=data, stream=True, timeout=self.timeout) as response:
                     if not response.ok:
-                        raise exceptions.FailedToGenerateResponseError(
-                            f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
-                        )
+                        # If we get a non-200 response, try refreshing our identity once
+                        if response.status_code in [403, 429]:
+                            self.refresh_identity()
+                            # Retry with new identity
+                            with self.session.post(self.chat_endpoint, headers=self.headers, cookies=self.cookies, json=data, stream=True, timeout=self.timeout) as retry_response:
+                                if not retry_response.ok:
+                                    raise exceptions.FailedToGenerateResponseError(
+                                        f"Failed to generate response after identity refresh - ({retry_response.status_code}, {retry_response.reason}) - {retry_response.text}"
+                                    )
+                                response = retry_response
+                        else:
+                            raise exceptions.FailedToGenerateResponseError(
+                                f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
+                            )
 
                     streaming_text = ""
                     for line in response.iter_lines(decode_unicode=True):
@@ -177,9 +224,30 @@ class YEPCHAT(Provider):
                 raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
 
         def for_non_stream():
-            for _ in for_stream():
-                pass
-            return self.last_response
+            try:
+                response = self.session.post(self.chat_endpoint, headers=self.headers, cookies=self.cookies, json=data, timeout=self.timeout)
+                if not response.ok:
+                    if response.status_code in [403, 429]:
+                        self.refresh_identity()
+                        response = self.session.post(self.chat_endpoint, headers=self.headers, cookies=self.cookies, json=data, timeout=self.timeout)
+                        if not response.ok:
+                            raise exceptions.FailedToGenerateResponseError(
+                                f"Failed to generate response after identity refresh - ({response.status_code}, {response.reason}) - {response.text}"
+                            )
+                    else:
+                        raise exceptions.FailedToGenerateResponseError(
+                            f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
+                        )
+
+                response_data = response.json()
+                if 'choices' in response_data and len(response_data['choices']) > 0:
+                    content = response_data['choices'][0].get('message', {}).get('content', '')
+                    self.conversation.update_chat_history(prompt, content)
+                    return {"text": content}
+                else:
+                    raise exceptions.FailedToGenerateResponseError("No response content found")
+            except Exception as e:
+                raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
 
         return for_stream() if stream else for_non_stream()
 
@@ -189,7 +257,7 @@ class YEPCHAT(Provider):
         stream: bool = False,
         optimizer: str = None,
         conversationally: bool = False,
-    ) -> str:
+    ) -> Union[str, Generator[str, None, None]]:
         """
         Initiates a chat with the Yep API using the provided prompt.
 
@@ -234,9 +302,23 @@ class YEPCHAT(Provider):
 
 
 if __name__ == "__main__":
-    from rich import print
-
-    ai = YEPCHAT(model="DeepSeek-R1-Distill-Qwen-32B")
-    response = ai.chat("how many r in 'strawberry'", stream=True)
-    for chunk in response:
-        print(chunk, end="", flush=True)
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
+
+    for model in YEPCHAT.AVAILABLE_MODELS:
+        try:
+            test_ai = YEPCHAT(model=model, timeout=60)
+            response = test_ai.chat("Say 'Hello' in one word")
+            response_text = response
+            
+            if response_text and len(response_text.strip()) > 0:
+                status = "✓"
+                # Truncate response if too long
+                display_text = response_text.strip()[:50] + "..." if len(response_text.strip()) > 50 else response_text.strip()
+            else:
+                status = "✗"
+                display_text = "Empty or invalid response"
+            print(f"{model:<50} {status:<10} {display_text}")
+        except Exception as e:
+            print(f"{model:<50} {'✗':<10} {str(e)}")

webscout/__init__.py

@@ -31,3 +31,6 @@ except Exception:
 
 import logging
 logging.getLogger("webscout").addHandler(logging.NullHandler())
+
+# Import models for easy access
+from webscout.models import model

webscout/cli.py

@@ -1,34 +1,12 @@
-import os
 import sys
-from concurrent.futures import ThreadPoolExecutor, as_completed
-from datetime import datetime
-from urllib.parse import unquote
-from pathlib import Path
-from .swiftcli import CLI, option, argument, group
-from curl_cffi import requests
-import pyreqwest_impersonate as pri
+from .swiftcli import CLI, option
 from .webscout_search import WEBS
-from .utils import json_dumps, json_loads
 from .version import __version__
-from .Litlogger import LitLogger, LogFormat, ColorScheme
 from rich.console import Console
 from rich.panel import Panel
-from rich.markdown import Markdown
 from rich.table import Table
-from rich.style import Style
 from rich.text import Text
-from rich.align import Align
-from rich.progress import track, Progress
-from rich.prompt import Prompt, Confirm
-from rich.columns import Columns
-from pyfiglet import figlet_format
 
-logger = LitLogger(
-    name="webscout",
-    format=LogFormat.MODERN_EMOJI,
-    color_scheme=ColorScheme.CYBERPUNK,
-    console_output=True
-)
 
 COLORS = {
     0: "black",
@@ -89,7 +67,7 @@ def chat(proxy: str = None, model: str = "gpt-4o-mini"):
     console = Console()
     
     # Display header
-    console.print(f"[bold blue]{figlet_format('Webscout Chat')}[/]\n", justify="center")
+    # console.print(f"[bold blue]{figlet_format('Webscout Chat')}[/]\n", justify="center")
     console.print(f"[bold green]Using model:[/] {model}\n")
     console.print("[cyan]Type your message and press Enter. Press Ctrl+C or type 'exit' to quit.[/]\n")
     
@@ -97,23 +75,18 @@ def chat(proxy: str = None, model: str = "gpt-4o-mini"):
     try:
         while True:
             try:
-                user_input = input("You: ").strip()
+                user_input = input(">>> ").strip()
                 if not user_input or user_input.lower() in ['exit', 'quit']:
                     break
                     
-                logger.info(f"Sending message to {model}")
                 response = webs.chat(keywords=user_input, model=model)
                 console.print(f"\nAI: {response}\n")
-                logger.success("Received response from AI")
                 
             except Exception as e:
-                logger.error(f"Error in chat: {str(e)}")
                 console.print(f"[bold red]Error:[/] {str(e)}\n")
                 
     except KeyboardInterrupt:
-        logger.info("Chat session terminated by user")
-        
-    console.print("\n[cyan]Chat session ended. Goodbye![/]")
+        console.print("\n[bold red]Chat session interrupted. Exiting...[/]")
 
 @app.command()
 @option("--keywords", "-k", help="Search keywords", required=True)
@@ -130,7 +103,6 @@ def text(keywords: str, region: str, safesearch: str, timelimit: str, backend: s
         results = webs.text(keywords, region, safesearch, timelimit, backend, max_results)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in text search: {e}")
         raise e
 
 @app.command()
@@ -143,7 +115,6 @@ def answers(keywords: str, proxy: str = None):
         results = webs.answers(keywords)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in answers search: {e}")
         raise e
 
 @app.command()
@@ -177,7 +148,6 @@ def images(
         results = webs.images(keywords, region, safesearch, timelimit, size, color, type, layout, license, max_results)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in images search: {e}")
         raise e
 
 @app.command()
@@ -207,7 +177,6 @@ def videos(
         results = webs.videos(keywords, region, safesearch, timelimit, resolution, duration, license, max_results)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in videos search: {e}")
         raise e
 
 @app.command()
@@ -224,7 +193,6 @@ def news(keywords: str, region: str, safesearch: str, timelimit: str, max_result
         results = webs.news(keywords, region, safesearch, timelimit, max_results)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in news search: {e}")
         raise e
 
 @app.command()
@@ -275,7 +243,6 @@ def maps(
         )
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in maps search: {e}")
         raise e
 
 @app.command()
@@ -290,7 +257,6 @@ def translate(keywords: str, from_: str, to: str, proxy: str = None):
         results = webs.translate(keywords, from_, to)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in translation: {e}")
         raise e
 
 @app.command()
@@ -304,7 +270,6 @@ def suggestions(keywords: str, region: str, proxy: str = None):
         results = webs.suggestions(keywords, region)
         _print_data(results)
     except Exception as e:
-        logger.error(f"Error in suggestions search: {e}")
         raise e
 
 def main():
@@ -312,7 +277,6 @@ def main():
     try:
         app.run()
     except Exception as e:
-        logger.error(f"Error: {e}")
         sys.exit(1)
 
 if __name__ == "__main__":