lollms-client 1.6.6__py3-none-any.whl → 1.6.10__py3-none-any.whl

lollms_client/lollms_discussion.py

@@ -1208,14 +1208,27 @@ class LollmsDiscussion:
             prompt_for_agent = self.export("markdown", branch_tip_id if branch_tip_id else self.active_branch_id)
             if debug:
                 ASCIIColors.cyan("\n" + "="*50 + "\n--- DEBUG: AGENTIC TURN TRIGGERED ---\n" + f"--- PROMPT FOR AGENT (from discussion history) ---\n{prompt_for_agent}\n" + "="*50 + "\n")
-
+            
+            
+            # Combine system prompt and data zones
+            system_prompt_part = (self._system_prompt or "").strip()
+            data_zone_part = self.get_full_data_zone() # This now returns a clean, multi-part block or an empty string
+            full_system_prompt = ""
+
+            # Combine them intelligently
+            if system_prompt_part and data_zone_part:
+                full_system_prompt = f"{system_prompt_part}\n\n{data_zone_part}"
+            elif system_prompt_part:
+                full_system_prompt = system_prompt_part
+            else:
+                full_system_prompt = data_zone_part
             agent_result = self.lollmsClient.generate_with_mcp_rag(
                 prompt=prompt_for_agent,
                 use_mcps=effective_use_mcps,
                 use_data_store=use_data_store,
                 max_reasoning_steps=max_reasoning_steps,
                 images=images,
-                system_prompt = self._system_prompt,
+                system_prompt = full_system_prompt,
                 debug=debug,
                 **kwargs
             )

lollms_client/lollms_stt_binding.py

@@ -4,7 +4,7 @@ import importlib
 from pathlib import Path
 from typing import Optional, List, Dict, Any, Union
 from ascii_colors import trace_exception
-
+import yaml
 class LollmsSTTBinding(ABC):
     """Abstract base class for all LOLLMS Speech-to-Text bindings."""
 
@@ -122,4 +122,58 @@ class LollmsSTTBindingManager:
             list[str]: List of binding names.
         """
         return [binding_dir.name for binding_dir in self.stt_bindings_dir.iterdir()
-                if binding_dir.is_dir() and (binding_dir / "__init__.py").exists()]
+                if binding_dir.is_dir() and (binding_dir / "__init__.py").exists()]
+
+
+    @staticmethod
+    def _get_fallback_description(binding_name: str) -> Dict:
+        return {
+            "binding_name": binding_name,
+            "title": binding_name.replace("_", " ").title(),
+            "author": "Unknown",
+            "version": "N/A",
+            "description": f"A binding for {binding_name}. No description.yaml file was found.",
+            "input_parameters": [
+                 {
+                    "name": "model_name",
+                    "type": "str",
+                    "description": "The model name or ID to be used.",
+                    "mandatory": False,
+                    "default": ""
+                }
+            ],
+            "generate_audio_parameters": []
+        }
+    @staticmethod
+    def get_bindings_list(stt_bindings_dir: Union[str, Path]) -> List[Dict]:
+        bindings_dir = Path(stt_bindings_dir)
+        if not bindings_dir.is_dir():
+            return []
+
+        bindings_list = []
+        for binding_folder in bindings_dir.iterdir():
+            if binding_folder.is_dir() and (binding_folder / "__init__.py").exists():
+                binding_name = binding_folder.name
+                description_file = binding_folder / "description.yaml"
+                
+                binding_info = {}
+                if description_file.exists():
+                    try:
+                        with open(description_file, 'r', encoding='utf-8') as f:
+                            binding_info = yaml.safe_load(f)
+                        binding_info['binding_name'] = binding_name
+                    except Exception as e:
+                        print(f"Error loading description.yaml for {binding_name}: {e}")
+                        binding_info = LollmsSTTBindingManager._get_fallback_description(binding_name)
+                else:
+                    binding_info = LollmsSTTBindingManager._get_fallback_description(binding_name)
+                
+                bindings_list.append(binding_info)
+
+        return sorted(bindings_list, key=lambda b: b.get('title', b['binding_name']))
+    
+
+def get_available_bindings(stt_bindings_dir: Union[str, Path] = None) -> List[Dict]:
+    if stt_bindings_dir is None:
+        stt_bindings_dir = Path(__file__).resolve().parent / "stt_bindings"
+    return LollmsSTTBindingManager.get_bindings_list(stt_bindings_dir)

lollms_client/stt_bindings/whisper/__init__.py

@@ -82,6 +82,7 @@ class WhisperSTTBinding(LollmsSTTBinding):
                                     If None, `torch` will attempt to auto-detect. Defaults to None.
         """
         super().__init__(binding_name="whisper") # Not applicable
+        self.default_model_name = kwargs.get("model_name", "base")
 
         if not _whisper_installed:
             raise ImportError(f"Whisper STT binding dependencies not met. Please ensure 'openai-whisper' and 'torch' are installed. Error: {_whisper_installation_error}")
@@ -104,7 +105,7 @@ class WhisperSTTBinding(LollmsSTTBinding):
 
     def _load_whisper_model(self, model_name_to_load: str):
         """Loads or reloads the Whisper model."""
-        if model_name_to_load not in self.WHISPER_MODEL_SIZES:
+        if model_name_to_load not in whisper.available_models():
             ASCIIColors.warning(f"'{model_name_to_load}' is not a standard Whisper model size. Attempting to load anyway. Known sizes: {self.WHISPER_MODEL_SIZES}")
 
         if self.model is not None and self.loaded_model_name == model_name_to_load:
@@ -202,7 +203,7 @@ class WhisperSTTBinding(LollmsSTTBinding):
         Returns:
             List[str]: A list of available Whisper model size identifiers.
         """
-        return self.WHISPER_MODEL_SIZES.copy() # Return a copy
+        return whisper.available_models() # Return a copy
 
     def __del__(self):
         """Clean up: Unload the model to free resources."""

lollms_client/stt_bindings/whispercpp/__init__.py

@@ -31,6 +31,8 @@ class WhisperCppSTTBinding(LollmsSTTBinding):
         n_threads = kwargs.get("n_threads", 4)
         extra_whisper_args = kwargs.get("extra_whisper_args", [])  # e.g. ["--no-timestamps"]
 
+        self.default_model_name = "base"
+
         # --- Validate FFMPEG ---
         self.ffmpeg_exe = None
         if ffmpeg_path:
@@ -376,4 +378,8 @@ if __name__ == '__main__':
                     TEST_MODELS_SEARCH_DIR.rmdir()
             except OSError: pass # Ignore if not empty or other issues
 
-    ASCIIColors.yellow("\n--- WhisperCppSTTBinding Test Finished ---")
+    ASCIIColors.yellow("\n--- WhisperCppSTTBinding Test Finished ---")
+
+    def list_models(self) -> List[Dict[str, Any]]:
+        return ["base" , "small", "medium", "large"]
+

lollms_client/tti_bindings/diffusers/__init__.py

@@ -45,7 +45,7 @@ class DiffusersBinding(LollmsTTIBinding):
 
         self.config = kwargs
         self.host = kwargs.get("host", "localhost")
-        self.port = kwargs.get("port", 9630)
+        self.port = kwargs.get("port", 9632)
         self.auto_start_server = kwargs.get("auto_start_server", True)
         self.server_process = None
         self.base_url = f"http://{self.host}:{self.port}"
@@ -61,7 +61,7 @@ class DiffusersBinding(LollmsTTIBinding):
     def is_server_running(self) -> bool:
         """Checks if the server is already running and responsive."""
         try:
-            response = requests.get(f"{self.base_url}/status", timeout=2)
+            response = requests.get(f"{self.base_url}/status", timeout=4)
             if response.status_code == 200 and response.json().get("status") == "running":
                 return True
         except requests.exceptions.RequestException:
@@ -90,7 +90,7 @@ class DiffusersBinding(LollmsTTIBinding):
         try:
             # Try to acquire the lock with a timeout. If another process is starting
             # the server, this will wait until it's finished.
-            with lock.acquire(timeout=60):
+            with lock.acquire(timeout=3):
                 # After acquiring the lock, we MUST re-check if the server is running.
                 # Another process might have started it and released the lock while we were waiting.
                 if not self.is_server_running():
@@ -105,7 +105,7 @@ class DiffusersBinding(LollmsTTIBinding):
             # This happens if the process holding the lock takes more than 60 seconds to start the server.
             # We don't try to start another one. We just wait for the existing one to be ready.
             ASCIIColors.yellow("Could not acquire lock, another process is taking a long time to start the server. Waiting...")
-            self._wait_for_server(timeout=300) # Give it a longer timeout here just in case.
+            self._wait_for_server(timeout=60) # Give it a longer timeout here just in case.
 
         # A final verification to ensure we are connected.
         if not self.is_server_running():
@@ -214,7 +214,7 @@ class DiffusersBinding(LollmsTTIBinding):
         self.server_process = subprocess.Popen(command, creationflags=creationflags)
         ASCIIColors.info("Diffusers server process launched in the background.")
 
-    def _wait_for_server(self, timeout=300):
+    def _wait_for_server(self, timeout=30):
         """Waits for the server to become responsive."""
         ASCIIColors.info("Waiting for Diffusers server to become available...")
         start_time = time.time()
@@ -374,4 +374,4 @@ class DiffusersBinding(LollmsTTIBinding):
     def __del__(self):
         # The client destructor does not stop the server,
         # as it is a shared resource for all worker processes.
-        pass
+        pass

lollms_client/tti_bindings/diffusers/server/main.py

@@ -60,66 +60,14 @@ MODELS_PATH = Path("./models")
 
 # --- START: Core Logic (Complete and Unabridged) ---
 CIVITAI_MODELS = {
-    "realistic-vision-v6": {
-        "display_name": "Realistic Vision V6.0", "url": "https://civitai.com/api/download/models/501240?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "realisticVisionV60_v60B1.safensors", "description": "Photorealistic SD1.5 checkpoint.", "owned_by": "civitai"
-    },
-    "absolute-reality": {
-        "display_name": "Absolute Reality", "url": "https://civitai.com/api/download/models/132760?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "absolutereality_v181.safetensors", "description": "General realistic SD1.5.", "owned_by": "civitai"
-    },
-    "dreamshaper-8": {
+    "DreamShaper-8": {
         "display_name": "DreamShaper 8", "url": "https://civitai.com/api/download/models/128713",
         "filename": "dreamshaper_8.safetensors", "description": "Versatile SD1.5 style model.", "owned_by": "civitai"
     },
-    "juggernaut-xl": {
+    "Juggernaut-xl": {
         "display_name": "Juggernaut XL", "url": "https://civitai.com/api/download/models/133005",
         "filename": "juggernautXL_version6Rundiffusion.safetensors", "description": "Artistic SDXL.", "owned_by": "civitai"
     },
-    "lyriel-v1.6": {
-        "display_name": "Lyriel v1.6", "url": "https://civitai.com/api/download/models/72396?type=Model&format=SafeTensor&size=full&fp=fp16",
-        "filename": "lyriel_v16.safetensors", "description": "Fantasy/stylized SD1.5.", "owned_by": "civitai"
-    },
-    "ui_icons": {
-        "display_name": "UI Icons", "url": "https://civitai.com/api/download/models/367044?type=Model&format=SafeTensor&size=full&fp=fp16",
-        "filename": "uiIcons_v10.safetensors", "description": "A model for generating UI icons.", "owned_by": "civitai"
-    },
-    "meinamix": {
-        "display_name": "MeinaMix", "url": "https://civitai.com/api/download/models/948574?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "meinamix_meinaV11.safetensors", "description": "Anime/illustration SD1.5.", "owned_by": "civitai"
-    },
-    "rpg-v5": {
-        "display_name": "RPG v5", "url": "https://civitai.com/api/download/models/124626?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "rpg_v5.safetensors", "description": "RPG assets SD1.5.", "owned_by": "civitai"
-    },
-    "pixel-art-xl": {
-        "display_name": "Pixel Art XL", "url": "https://civitai.com/api/download/models/135931?type=Model&format=SafeTensor",
-        "filename": "pixelartxl_v11.safetensors", "description": "Pixel art SDXL.", "owned_by": "civitai"
-    },
-    "lowpoly-world": {
-        "display_name": "Lowpoly World", "url": "https://civitai.com/api/download/models/146502?type=Model&format=SafeTensor",
-        "filename": "LowpolySDXL.safetensors", "description": "Lowpoly style SD1.5.", "owned_by": "civitai"
-    },
-    "toonyou": {
-        "display_name": "ToonYou", "url": "https://civitai.com/api/download/models/125771?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "toonyou_beta6.safetensors", "description": "Cartoon/Disney SD1.5.", "owned_by": "civitai"
-    },
-    "papercut": {
-        "display_name": "Papercut", "url": "https://civitai.com/api/download/models/133503?type=Model&format=SafeTensor",
-        "filename": "papercut.safetensors", "description": "Paper cutout SD1.5.", "owned_by": "civitai"
-    },
-    "fantassifiedIcons": {
-        "display_name": "Fantassified Icons", "url": "https://civitai.com/api/download/models/67584?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "fantassifiedIcons_fantassifiedIconsV20.safetensors", "description": "Flat, modern Icons.", "owned_by": "civitai"
-    },
-    "game_icon_institute": {
-        "display_name": "Game icon institute", "url": "https://civitai.com/api/download/models/158776?type=Model&format=SafeTensor&size=full&fp=fp16",
-        "filename": "gameIconInstituteV10_v10.safetensors", "description": "Flat, modern game Icons.", "owned_by": "civitai"
-    },
-    "M4RV3LS_DUNGEONS": {
-        "display_name": "M4RV3LS & DUNGEONS", "url": "https://civitai.com/api/download/models/139417?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "filename": "M4RV3LSDUNGEONSNEWV40COMICS_mD40.safetensors", "description": "comics.", "owned_by": "civitai"
-    },
 }
 
 HF_PUBLIC_MODELS = {
@@ -145,8 +93,11 @@ HF_PUBLIC_MODELS = {
     ],
     "Image Editing Tools": [
         {"model_name": "stabilityai/stable-diffusion-xl-refiner-1.0", "display_name": "SDXL Refiner 1.0", "desc": "A dedicated refiner model to improve details in SDXL generations."},
-        {"model_name": "Qwen/Qwen-Image-Edit", "display_name": "Qwen Image Edit", "desc": "An instruction-based model for various image editing tasks."},
-        {"model_name": "Qwen/Qwen-Image-Edit-2509", "display_name": "Qwen Image Edit Plus", "desc": "Advanced multi-image editing, fusion, and pose transfer."},
+        {"model_name": "timbrooks/instruct-pix2pix", "display_name": "Instruct-Pix2Pix", "desc": "The original instruction-based image editing model (SD 1.5)."},
+        {"model_name": "kandinsky-community/kandinsky-2-2-instruct-pix2pix", "display_name": "Kandinsky 2.2 Instruct", "desc": "An instruction-based model with strong prompt adherence, based on Kandinsky 2.2."},
+        {"model_name": "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", "display_name": "SDXL Inpainting", "desc": "A dedicated inpainting model based on SDXL 1.0 for filling in masked areas."},
+        {"model_name": "Qwen/Qwen-Image-Edit", "display_name": "Qwen Image Edit", "desc": "An instruction-based model for various image editing tasks. (Review License)."},
+        {"model_name": "Qwen/Qwen-Image-Edit-2509", "display_name": "Qwen Image Edit Plus", "desc": "Advanced multi-image editing and fusion. (Review License)."},
     ],
     "Legacy & Base Models": [
         {"model_name": "runwayml/stable-diffusion-v1-5", "display_name": "Stable Diffusion 1.5", "desc": "The classic and versatile SD1.5 base model."},
@@ -180,6 +131,7 @@ SCHEDULER_MAPPING = {
     "dpm2_karras": "KDPM2DiscreteScheduler", "dpm2_a": "KDPM2AncestralDiscreteScheduler", "dpm2_a_karras": "KDPM2AncestralDiscreteScheduler",
     "euler": "EulerDiscreteScheduler", "euler_a": "EulerAncestralDiscreteScheduler", "heun": "HeunDiscreteScheduler", "lms": "LMSDiscreteScheduler"
 }
+
 SCHEDULER_USES_KARRAS_SIGMAS = [
     "dpm_multistep_karras","dpm++_2m_karras","dpm++_2s_ancestral_karras", "dpm++_sde_karras","heun_karras","lms_karras",
     "dpm++_2m_sde_karras","dpm2_karras","dpm2_a_karras"
@@ -601,7 +553,7 @@ class ServerState:
         return {
             "model_name": "", "device": "auto", "torch_dtype_str": "auto", "use_safetensors": True,
             "scheduler_name": "default", "safety_checker_on": True, "num_inference_steps": 25,
-            "guidance_scale": 7.0, "width": 512, "height": 512, "seed": -1,
+            "guidance_scale": 7.0, "width": 1024, "height": 1024, "seed": -1,
             "enable_cpu_offload": False, "enable_sequential_cpu_offload": False, "enable_xformers": False,
             "hf_variant": None, "hf_token": None, "hf_cache_path": None, "local_files_only": False,
             "unload_inactive_model_after": 0
@@ -728,36 +680,33 @@ async def generate_image(request: T2IRequest):
     manager = None
     temp_config = None
     try:
-        params = request.params
-        
         # Determine which model manager to use for this specific request
-        if "model_name" in params and params["model_name"]:
+        if "model_name" in request.params and request.params["model_name"]:
             temp_config = state.config.copy()
-            temp_config["model_name"] = params.pop("model_name") # Remove from params to avoid being passed to pipeline
+            temp_config["model_name"] = request.params.pop("model_name") # Remove from params to avoid being passed to pipeline
             manager = state.registry.get_manager(temp_config, state.models_path)
             ASCIIColors.info(f"Using per-request model: {temp_config['model_name']}")
         else:
             manager = state.get_active_manager()
             ASCIIColors.info(f"Using session-configured model: {manager.config.get('model_name')}")
 
-        seed = int(params.get("seed", manager.config.get("seed", -1)))
-        generator = None
+        # Start with the manager's config (base settings)
+        pipeline_args = manager.config.copy()
+        # Override with per-request parameters
+        pipeline_args.update(request.params)
+
+        # Add prompts and ensure types for specific args
+        pipeline_args["prompt"] = request.prompt
+        pipeline_args["negative_prompt"] = request.negative_prompt
+        pipeline_args["width"] = int(pipeline_args.get("width", 1024))
+        pipeline_args["height"] = int(pipeline_args.get("height", 1024))
+        pipeline_args["num_inference_steps"] = int(pipeline_args.get("num_inference_steps", 25))
+        pipeline_args["guidance_scale"] = float(pipeline_args.get("guidance_scale", 7.0))
+
+        seed = int(pipeline_args.get("seed", -1))
+        pipeline_args["generator"] = None
         if seed != -1:
-            generator = torch.Generator(device=manager.config["device"]).manual_seed(seed)
-        
-        width = int(params.get("width", manager.config.get("width", 512)))
-        height = int(params.get("height", manager.config.get("height", 512)))
-        
-        pipeline_args = {
-            "prompt": request.prompt,
-            "negative_prompt": request.negative_prompt,
-            "width": width,
-            "height": height,
-            "num_inference_steps": int(params.get("num_inference_steps", manager.config.get("num_inference_steps", 25))),
-            "guidance_scale": float(params.get("guidance_scale", manager.config.get("guidance_scale", 7.0))),
-            "generator": generator
-        }
-        pipeline_args.update(params)
+            pipeline_args["generator"] = torch.Generator(device=manager.config["device"]).manual_seed(seed)
         
         model_name = manager.config.get("model_name", "")
         task = "text2image"
@@ -765,12 +714,24 @@ async def generate_image(request: T2IRequest):
         if "Qwen-Image-Edit" in model_name:
             rng_seed = seed if seed != -1 else None
             rng = np.random.default_rng(seed=rng_seed)
-            random_pixels = rng.integers(0, 256, size=(height, width, 3), dtype=np.uint8)
+            random_pixels = rng.integers(0, 256, size=(pipeline_args["height"], pipeline_args["width"], 3), dtype=np.uint8)
             placeholder_image = Image.fromarray(random_pixels, 'RGB')
             pipeline_args["image"] = placeholder_image
-            pipeline_args["strength"] = float(params.get("strength", 1.0))
+            pipeline_args["strength"] = float(pipeline_args.get("strength", 1.0))
             task = "image2image" 
         
+        log_args = {k: v for k, v in pipeline_args.items() if k not in ['generator', 'image']}
+        if pipeline_args.get("generator"): log_args['generator'] = f"<torch.Generator(seed={seed})>"
+        if pipeline_args.get("image"): log_args['image'] = "<PIL Image object>"
+        
+        ASCIIColors.cyan("--- Generating Image with Settings ---")
+        try:
+            print(json.dumps(log_args, indent=2, default=str))
+        except Exception as e:
+            ASCIIColors.warning(f"Could not print all settings: {e}")
+            print(log_args)
+        ASCIIColors.cyan("------------------------------------")
+        
         future = Future()
         manager.queue.put((future, task, pipeline_args))
         result_bytes = future.result()
@@ -789,17 +750,20 @@ async def edit_image(request: EditRequestJSON):
     manager = None
     temp_config = None
     try:
-        params = request.params
-
-        if "model_name" in params and params["model_name"]:
+        if "model_name" in request.params and request.params["model_name"]:
             temp_config = state.config.copy()
-            temp_config["model_name"] = params.pop("model_name")
+            temp_config["model_name"] = request.params.pop("model_name")
             manager = state.registry.get_manager(temp_config, state.models_path)
             ASCIIColors.info(f"Using per-request model: {temp_config['model_name']}")
         else:
             manager = state.get_active_manager()
             ASCIIColors.info(f"Using session-configured model: {manager.config.get('model_name')}")
 
+        # Start with manager's config, then override with request params
+        pipeline_args = manager.config.copy()
+        pipeline_args.update(request.params)
+        
+        pipeline_args["prompt"] = request.prompt
         model_name = manager.config.get("model_name", "")
 
         pil_images = []
@@ -810,27 +774,38 @@ async def edit_image(request: EditRequestJSON):
 
         if not pil_images: raise HTTPException(status_code=400, detail="No valid images provided.")
 
-        pipeline_args = {"prompt": request.prompt}
-        seed = int(params.get("seed", -1))
+        seed = int(pipeline_args.get("seed", -1))
+        pipeline_args["generator"] = None
         if seed != -1: pipeline_args["generator"] = torch.Generator(device=manager.config["device"]).manual_seed(seed)
         
-        if "mask_image" in params and params["mask_image"]:
-            b64_mask = params["mask_image"]
+        if "mask_image" in pipeline_args and pipeline_args["mask_image"]:
+            b64_mask = pipeline_args["mask_image"]
             b64_data = b64_mask.split(";base64,")[1] if ";base64," in b64_mask else b64_mask
             mask_bytes = base64.b64decode(b64_data)
             pipeline_args["mask_image"] = Image.open(BytesIO(mask_bytes)).convert("L")
         
-        task = "inpainting" if "mask_image" in pipeline_args else "image2image"
+        task = "inpainting" if "mask_image" in pipeline_args and pipeline_args["mask_image"] else "image2image"
         
         if "Qwen-Image-Edit-2509" in model_name:
             task = "image2image"
             pipeline_args.update({"true_cfg_scale": 4.0, "guidance_scale": 1.0, "num_inference_steps": 40, "negative_prompt": " "})
-            edit_mode = params.get("edit_mode", "fusion")
+            edit_mode = pipeline_args.get("edit_mode", "fusion")
             if edit_mode == "fusion": pipeline_args["image"] = pil_images
         else:
             pipeline_args.update({"image": pil_images[0], "strength": 0.8, "guidance_scale": 7.5, "num_inference_steps": 25})
 
-        pipeline_args.update(params)
+        log_args = {k: v for k, v in pipeline_args.items() if k not in ['generator', 'image', 'mask_image']}
+        if pipeline_args.get("generator"): log_args['generator'] = f"<torch.Generator(seed={seed})>"
+        if 'image' in pipeline_args: log_args['image'] = f"[<{len(pil_images)} PIL Image(s)>]"
+        if 'mask_image' in pipeline_args and pipeline_args['mask_image']: log_args['mask_image'] = "<PIL Mask Image>"
+
+        ASCIIColors.cyan("--- Editing Image with Settings ---")
+        try:
+            print(json.dumps(log_args, indent=2, default=str))
+        except Exception as e:
+            ASCIIColors.warning(f"Could not print all settings: {e}")
+            print(log_args)
+        ASCIIColors.cyan("---------------------------------")
         
         future = Future(); manager.queue.put((future, task, pipeline_args))
         return Response(content=future.result(), media_type="image/png")
@@ -963,4 +938,4 @@ if __name__ == "__main__":
     else:
         ASCIIColors.info(f"Detected device: {state.config['device']}")
 
-    uvicorn.run(app, host=args.host, port=args.port, reload=False)
+    uvicorn.run(app, host=args.host, port=args.port, reload=False)

lollms_client/tts_bindings/xtts/__init__.py

@@ -6,13 +6,6 @@ import time
 from pathlib import Path
 from typing import Optional, List
 
-# Ensure pipmaster is available.
-try:
-    import pipmaster as pm
-except ImportError:
-    print("FATAL: pipmaster is not installed. Please install it using: pip install pipmaster")
-    sys.exit(1)
-
 # Ensure filelock is available for process-safe server startup.
 try:
     from filelock import FileLock, Timeout
@@ -39,7 +32,7 @@ class XTTSClientBinding(LollmsTTSBinding):
 
         self.config = kwargs
         self.host = kwargs.get("host", "localhost")
-        self.port = kwargs.get("port", 8081)
+        self.port = kwargs.get("port", 9633)
         self.auto_start_server = kwargs.get("auto_start_server", True)
         self.server_process = None
         self.base_url = f"http://{self.host}:{self.port}"
@@ -76,7 +69,7 @@ class XTTSClientBinding(LollmsTTSBinding):
             return
 
         try:
-            with lock.acquire(timeout=60):
+            with lock.acquire(timeout=10):
                 if not self.is_server_running():
                     ASCIIColors.yellow("Lock acquired. Starting dedicated XTTS server...")
                     self.start_server()
@@ -85,7 +78,7 @@ class XTTSClientBinding(LollmsTTSBinding):
                     ASCIIColors.green("Server was started by another process while we waited. Connected successfully.")
         except Timeout:
             ASCIIColors.yellow("Could not acquire lock, another process is starting the server. Waiting...")
-            self._wait_for_server(timeout=180)
+            self._wait_for_server(timeout=60)
 
         if not self.is_server_running():
             raise RuntimeError("Failed to start or connect to the XTTS server after all attempts.")
@@ -97,6 +90,12 @@ class XTTSClientBinding(LollmsTTSBinding):
         using pipmaster, which handles complex packages like PyTorch.
         """
         ASCIIColors.info(f"Setting up virtual environment in: {self.venv_dir}")
+        # Ensure pipmaster is available.
+        try:
+            import pipmaster as pm
+        except ImportError:
+            print("FATAL: pipmaster is not installed. Please install it using: pip install pipmaster")
+            raise Exception("pipmaster not found")
         pm_v = pm.PackageManager(venv_path=str(self.venv_dir))
         
         requirements_file = self.server_dir / "requirements.txt"
@@ -141,7 +140,7 @@ class XTTSClientBinding(LollmsTTSBinding):
         self.server_process = subprocess.Popen(command, creationflags=creationflags)
         ASCIIColors.info("XTTS server process launched in the background.")
 
-    def _wait_for_server(self, timeout=120):
+    def _wait_for_server(self, timeout=10):
         """Waits for the server to become responsive."""
         ASCIIColors.info("Waiting for XTTS server to become available...")
         start_time = time.time()

lollms_client/tts_bindings/xtts/server/main.py

@@ -272,4 +272,4 @@ except Exception as e:
     from ascii_colors import ASCIIColors
     ASCIIColors.red(f"Server: CRITICAL ERROR during startup: {e}")
     import traceback
-    ASCIIColors.red(f"Server: Traceback:\n{traceback.format_exc()}")```
+    ASCIIColors.red(f"Server: Traceback:\n{traceback.format_exc()}")

{lollms_client-1.6.6.dist-info → lollms_client-1.6.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lollms_client
-Version: 1.6.6
+Version: 1.6.10
 Summary: A client library for LoLLMs generate endpoint
 Author-email: ParisNeo <parisneoai@gmail.com>
 License:                                  Apache License

{lollms_client-1.6.6.dist-info → lollms_client-1.6.10.dist-info}/RECORD

@@ -1,15 +1,15 @@
-lollms_client/__init__.py,sha256=51YtCHNJCmroyA9htiIgjui1ZSFfkn_zhhe0USpE8nc,1146
-lollms_client/lollms_agentic.py,sha256=pQiMEuB_XkG29-SW6u4KTaMFPr6eKqacInggcCuCW3k,13914
+lollms_client/__init__.py,sha256=0D-nwmkSe8qYbyPSuKdYU09t7x5P2BYwfpI4mbzEZlU,1147
+lollms_client/lollms_agentic.py,sha256=ljalnmeSU-sbzH19-c9TzrJ-HhEeo4mxXmpJGkXj720,14094
 lollms_client/lollms_config.py,sha256=goEseDwDxYJf3WkYJ4IrLXwg3Tfw73CXV2Avg45M_hE,21876
-lollms_client/lollms_core.py,sha256=Un74iLbnnn2yZYH6HBNRz1mTZ454NEMBEndS4nvh3ZI,244887
-lollms_client/lollms_discussion.py,sha256=LZc9jYbUMRTovehiFJKEp-NXuCl_WnrqUtT3t4Nzayk,123922
+lollms_client/lollms_core.py,sha256=PGHPu_V5rKnO-032EjgMw8M5T0SLpNOWrzSUlWKBBgE,253052
+lollms_client/lollms_discussion.py,sha256=4uzXLqGz72xZcXEtamWGudTOR54cYwuo6k8JY37scqY,124574
 lollms_client/lollms_js_analyzer.py,sha256=01zUvuO2F_lnUe_0NLxe1MF5aHE1hO8RZi48mNPv-aw,8361
 lollms_client/lollms_llm_binding.py,sha256=_6d0q9g9lk8FRZ1oYnLpuqG7Y_WLyBJBn4ANdk-C8gU,25020
 lollms_client/lollms_mcp_binding.py,sha256=psb27A23VFWDfZsR2WUbQXQxiZDW5yfOak6ZtbMfszI,10222
 lollms_client/lollms_mcp_security.py,sha256=FhVTDhSBjksGEZnopVnjFmEF5dv7D8bBTqoaj4BiF0E,3562
 lollms_client/lollms_personality.py,sha256=kGuFwmgA9QDLcQlLQ9sKeceMujdEo0Aw28fN5H8MpjI,8847
 lollms_client/lollms_python_analyzer.py,sha256=7gf1fdYgXCOkPUkBAPNmr6S-66hMH4_KonOMsADASxc,10246
-lollms_client/lollms_stt_binding.py,sha256=jAUhLouEhh2hmm1bK76ianfw_6B59EHfY3FmLv6DU-g,5111
+lollms_client/lollms_stt_binding.py,sha256=WkREwu0uc0UzeCv5Z9ok8AFG42iBq20ZOQJnydSTE0s,7505
 lollms_client/lollms_tti_binding.py,sha256=MhDntyXVNoZeqMH0YpoNtiLijXPL8Y--if2qjQAS0-w,8520
 lollms_client/lollms_ttm_binding.py,sha256=FjVVSNXOZXK1qvcKEfxdiX6l2b4XdGOSNnZ0utAsbDg,4167
 lollms_client/lollms_tts_binding.py,sha256=k13rNq4YmuR50kkAEacwADW7COoDUOMLGAcnm27xjO4,5150
@@ -49,11 +49,11 @@ lollms_client/mcp_bindings/remote_mcp/__init__.py,sha256=YpSclbNJDYVUe2W0H5Xki4g
 lollms_client/mcp_bindings/standard_mcp/__init__.py,sha256=wJQofr4zS5RIS9V5_WuMMFsJxSDJgXDW3PQPX1hlx6g,31519
 lollms_client/stt_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lollms_client/stt_bindings/lollms/__init__.py,sha256=9Vmn1sQQZKLGLe7nZnc-0LnNeSY8r9xw3pYZF-wVtPo,5889
-lollms_client/stt_bindings/whisper/__init__.py,sha256=1Ej67GdRKBy1bba14jMaYDYHiZkxJASkWm5eF07ztDQ,15363
-lollms_client/stt_bindings/whispercpp/__init__.py,sha256=xSAQRjAhljak3vWCpkP0Vmdb6WmwTzPjXyaIB85KLGU,21439
+lollms_client/stt_bindings/whisper/__init__.py,sha256=HVVYRGIPkTTwNw5uhvxvRkSYeAv6nNRZp_geS8SwKZ4,15428
+lollms_client/stt_bindings/whispercpp/__init__.py,sha256=5YQKFy3UaN-S-HGZiFCIcuPGTJTELPgqqct1AcTqz-Q,21595
 lollms_client/tti_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/tti_bindings/diffusers/__init__.py,sha256=esrcyy_z_6HVCFKMVXl1h_qY_pX3kMHwO81M2C8hSIg,17706
-lollms_client/tti_bindings/diffusers/server/main.py,sha256=PQ3WXhkQzEzyT100k7nu1ZHQtkGphvpWNGl7Bcg26eY,49593
+lollms_client/tti_bindings/diffusers/__init__.py,sha256=_Nd3OotY1kBlEmHBuxVsNwIc_vvUy8sBo1Ug9_lUOzI,17705
+lollms_client/tti_bindings/diffusers/server/main.py,sha256=-Eo9vrpsK_LXRFJWkplriSdUX8VnkcL6tfFdxontNnM,48136
 lollms_client/tti_bindings/gemini/__init__.py,sha256=eYGz6gnOxWGdJu2O0H-EwGG-Hg7Yo3Hzsgn4neqx29Q,12963
 lollms_client/tti_bindings/leonardo_ai/__init__.py,sha256=pUbF1rKPZib1x0Kn2Bk1A7sTFWmZzNG02kmW6Iu1j2w,5885
 lollms_client/tti_bindings/lollms/__init__.py,sha256=5Tnsn4b17djvieQkcjtIDBm3qf0pg5ZWWov-4_2wmo0,8762
@@ -76,13 +76,13 @@ lollms_client/tts_bindings/piper_tts/__init__.py,sha256=7LQUuWV8I3IEdacc65NRHmDf
 lollms_client/tts_bindings/piper_tts/server/install_piper.py,sha256=g71Ne2T18wAytOPipfQ9DNeTAOD9PrII5qC-vr9DtLA,3256
 lollms_client/tts_bindings/piper_tts/server/main.py,sha256=DMozfSR1aCbrlmOXltRFjtXhYhXajsGcNKQjsWgRwZk,17402
 lollms_client/tts_bindings/piper_tts/server/setup_voices.py,sha256=UdHaPa5aNcw8dR-aRGkZr2OfSFFejH79lXgfwT0P3ss,1964
-lollms_client/tts_bindings/xtts/__init__.py,sha256=lTlExBPZ97FPaf9DoqxE4ilwwO5y88dPOHeRaR5BCnc,8002
-lollms_client/tts_bindings/xtts/server/main.py,sha256=JYKUzg4qFOGW8O_QDb9ChEdhcPRSccdwOlR3q-kJX7I,12306
+lollms_client/tts_bindings/xtts/__init__.py,sha256=q5xuNUYz4l9ajmZo4yvcTgxPuv9HT6T16u-weh3lzC8,8073
+lollms_client/tts_bindings/xtts/server/main.py,sha256=feTAX4eAo2HY6PpcDTrgRMak5AXocO7UIhKPuGuWpxY,12303
 lollms_client/tts_bindings/xtts/server/setup_voices.py,sha256=UdHaPa5aNcw8dR-aRGkZr2OfSFFejH79lXgfwT0P3ss,1964
 lollms_client/ttv_bindings/__init__.py,sha256=UZ8o2izQOJLQgtZ1D1cXoNST7rzqW22rL2Vufc7ddRc,3141
 lollms_client/ttv_bindings/lollms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client-1.6.6.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-lollms_client-1.6.6.dist-info/METADATA,sha256=i6Gb5wKrXNF6OPUCz41s5YbpBY5HEvLdAD5a6ONZV84,76835
-lollms_client-1.6.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lollms_client-1.6.6.dist-info/top_level.txt,sha256=Bk_kz-ri6Arwsk7YG-T5VsRorV66uVhcHGvb_g2WqgE,14
-lollms_client-1.6.6.dist-info/RECORD,,
+lollms_client-1.6.10.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+lollms_client-1.6.10.dist-info/METADATA,sha256=Mqk7RqLL6F5By0K_XIJemPSNoQhDcaW9JBT5-H9Tewg,76836
+lollms_client-1.6.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lollms_client-1.6.10.dist-info/top_level.txt,sha256=Bk_kz-ri6Arwsk7YG-T5VsRorV66uVhcHGvb_g2WqgE,14
+lollms_client-1.6.10.dist-info/RECORD,,