lollms-client 1.6.7__py3-none-any.whl → 1.7.0__py3-none-any.whl

lollms_client/lollms_stt_binding.py

@@ -4,12 +4,13 @@ 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."""
 
     def __init__(self,
-                 binding_name:str="unknown"):
+                 binding_name:str="unknown",
+                 config={}):
         """
         Initialize the LollmsSTTBinding base class.
 
@@ -17,6 +18,7 @@ class LollmsSTTBinding(ABC):
             binding_name (Optional[str]): The binding name
         """
         self.binding_name = binding_name
+        self.config = config
 
     @abstractmethod
     def transcribe_audio(self, audio_path: Union[str, Path], model: Optional[str] = None, **kwargs) -> str:
@@ -122,4 +124,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/lollms_tti_binding.py

@@ -9,7 +9,8 @@ class LollmsTTIBinding(ABC):
     """Abstract base class for all LOLLMS Text-to-Image bindings."""
 
     def __init__(self,
-                 binding_name:str="unknown"):
+                 binding_name:str="unknown",
+                 config={}):
         """
         Initialize the LollmsTTIBinding base class.
 
@@ -17,6 +18,7 @@ class LollmsTTIBinding(ABC):
             binding_name (Optional[str]): The binding name
         """
         self.binding_name = binding_name
+        self.config = config
 
     @abstractmethod
     def generate_image(self,

lollms_client/lollms_ttm_binding.py

@@ -9,7 +9,8 @@ class LollmsTTMBinding(ABC):
     """Abstract base class for all LOLLMS Text-to-Music bindings."""
 
     def __init__(self,
-                 binding_name:str="unknown"):
+                 binding_name:str="unknown",
+                 config={}):
         """
         Initialize the LollmsTTMBinding base class.
 
@@ -17,6 +18,7 @@ class LollmsTTMBinding(ABC):
             binding_name (Optional[str]): The binding name
         """
         self.binding_name = binding_name
+        self.config = config
 
 
     @abstractmethod

lollms_client/lollms_tts_binding.py

@@ -8,9 +8,9 @@ from ascii_colors import trace_exception
 class LollmsTTSBinding(ABC):
     def __init__(self,
                  binding_name: str = "unknown",
-                 **kwargs):
+                 config={}):
         self.binding_name = binding_name
-        self.settings = kwargs
+        self.config = config
 
     @abstractmethod
     def generate_audio(self,

lollms_client/lollms_ttv_binding.py

@@ -9,7 +9,8 @@ class LollmsTTVBinding(ABC):
     """Abstract base class for all LOLLMS Text-to-Video bindings."""
 
     def __init__(self,
-                 binding_name:str="unknown"):
+                 binding_name:str="unknown",
+                 config={}):
         """
         Initialize the LollmsTTVBinding base class.
 
@@ -17,6 +18,7 @@ class LollmsTTVBinding(ABC):
             binding_name (Optional[str]): The binding name
         """
         self.binding_name = binding_name
+        self.config = config
 
     @abstractmethod
     def generate_video(self, prompt: str, **kwargs) -> bytes:

lollms_client/stt_bindings/whisper/__init__.py

@@ -67,7 +67,7 @@ class WhisperSTTBinding(LollmsSTTBinding):
     """
 
     # Standard Whisper model sizes
-    WHISPER_MODEL_SIZES = ["tiny", "tiny.en", "base", "base.en", "small", "small.en", "medium", "medium.en", "large", "large-v1", "large-v2", "large-v3"]
+    WHISPER_MODEL_SIZES = ["turbo"]
 
     def __init__(self,
                  **kwargs # To catch any other LollmsSTTBinding standard args
@@ -76,12 +76,13 @@ class WhisperSTTBinding(LollmsSTTBinding):
         Initialize the Whisper STT binding.
 
         Args:
-            model_name (str): The Whisper model size to use (e.g., "tiny", "base", "small", "medium", "large", "large-v2", "large-v3").
-                              Defaults to "base".
+            model_name (str): The Whisper model size to use (e.g., "turbo").
+                              Defaults to "turno".
             device (Optional[str]): The device to run the model on ("cpu", "cuda", "mps").
                                     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", "turno")
 
         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}")
@@ -99,12 +100,15 @@ class WhisperSTTBinding(LollmsSTTBinding):
         
         self.loaded_model_name = None
         self.model = None
-        self._load_whisper_model(kwargs.get("model_name", "base")) # Default to "base" if not specified
+        try:
+            self._load_whisper_model(kwargs.get("model_name", "turbo")) # Default to "turno" if not specified
+        except Exception as e:
+            pass
 
 
     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:
@@ -149,6 +153,10 @@ class WhisperSTTBinding(LollmsSTTBinding):
             RuntimeError: If the Whisper model is not loaded or transcription fails.
             Exception: For other errors during transcription.
         """
+        if not self.model:
+            self._load_whisper_model(self.default_model_name)
+
+            
         audio_file = Path(audio_path)
         if not audio_file.exists():
             raise FileNotFoundError(f"Audio file not found at: {audio_path}")
@@ -191,8 +199,8 @@ class WhisperSTTBinding(LollmsSTTBinding):
             trace_exception(e)
             raise Exception(f"Whisper transcription error: {e}") from e
 
-
-    def list_models(self, **kwargs) -> List[str]:
+    @staticmethod
+    def list_models(**kwargs) -> List[str]:
         """
         Lists the available standard Whisper model sizes.
 
@@ -202,7 +210,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."""
@@ -252,7 +260,7 @@ if __name__ == '__main__':
     else:
         try:
             ASCIIColors.cyan("\n--- Initializing WhisperSTTBinding (model: 'tiny') ---")
-            # Using 'tiny' model for faster testing. Change to 'base' or 'small' for better quality.
+            # Using 'tiny' model for faster testing. Change to 'turno' or 'small' for better quality.
             stt_binding = WhisperSTTBinding(model_name="tiny")
 
             ASCIIColors.cyan("\n--- Listing available Whisper models ---")
@@ -269,9 +277,9 @@ if __name__ == '__main__':
             # print(f"Transcription (tiny, lang='en'): '{transcription_lang_hint}'")
 
             # Test switching model dynamically (optional, will re-download/load if different)
-            # ASCIIColors.cyan(f"\n--- Transcribing '{test_audio_file.name}' by switching to 'base' model ---")
-            # transcription_base = stt_binding.transcribe_audio(test_audio_file, model="base")
-            # print(f"Transcription (base): '{transcription_base}'")
+            # ASCIIColors.cyan(f"\n--- Transcribing '{test_audio_file.name}' by switching to 'turno' model ---")
+            # transcription_base = stt_binding.transcribe_audio(test_audio_file, model="turno")
+            # print(f"Transcription (turno): '{transcription_base}'")
 
 
         except ImportError as e_imp:

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

@@ -36,16 +36,15 @@ class DiffusersBinding(LollmsTTIBinding):
     """
     def __init__(self,
                  **kwargs):
-
-        super().__init__(binding_name=BindingName)
-
         # Prioritize 'model_name' but accept 'model' as an alias from config files.
         if 'model' in kwargs and 'model_name' not in kwargs:
             kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
+
+
 
-        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 +60,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 +89,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 +104,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 +213,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 +373,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.safetensors", "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 = {
@@ -183,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"
@@ -604,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
@@ -749,8 +698,8 @@ async def generate_image(request: T2IRequest):
         # 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", 512))
-        pipeline_args["height"] = int(pipeline_args.get("height", 512))
+        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))
 
@@ -800,6 +749,8 @@ async def generate_image(request: T2IRequest):
 async def edit_image(request: EditRequestJSON):
     manager = None
     temp_config = None
+    ASCIIColors.info(f"Received /edit_image request with {len(request.images_b64)} image(s).")
+    ASCIIColors.info(request.params)
     try:
         if "model_name" in request.params and request.params["model_name"]:
             temp_config = state.config.copy()
@@ -843,7 +794,7 @@ async def edit_image(request: EditRequestJSON):
             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({"image": pil_images[0]})
 
         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})>"
@@ -989,4 +940,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/tti_bindings/gemini/__init__.py

@@ -69,7 +69,10 @@ def _load_image_from_str(image_str: str) -> bytes:
 
 class GeminiTTIBinding_Impl(LollmsTTIBinding):
     def __init__(self, **kwargs):
-        super().__init__(binding_name="gemini")
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
         self.auth_method = kwargs.get("auth_method", "vertex_ai")
         self.client: Optional[Any] = None
         self.available_models = []

lollms_client/tti_bindings/leonardo_ai/__init__.py

@@ -29,8 +29,11 @@ class LeonardoAITTIBinding(LollmsTTIBinding):
     """Leonardo.ai TTI binding for LoLLMS"""
 
     def __init__(self, **kwargs):
-        super().__init__(binding_name=BindingName)
-        self.config = kwargs
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
+
         self.api_key = self.config.get("api_key") or os.environ.get("LEONARDO_API_KEY")
         if not self.api_key:
             raise ValueError("Leonardo.ai API key is required.")

lollms_client/tti_bindings/lollms/__init__.py

@@ -23,7 +23,10 @@ class LollmsWebuiTTIBinding_Impl(LollmsTTIBinding):
             service_key (Optional[str]): Authentication key (used for client_id verification).
             verify_ssl_certificate (bool): Whether to verify SSL certificates.
         """
-        super().__init__(binding_name="lollms")
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
 
         # Extract parameters from kwargs, providing defaults
         self.host_address = kwargs.get("host_address", "http://localhost:9600")  # Default LOLLMS host

lollms_client/tti_bindings/novita_ai/__init__.py

@@ -27,7 +27,10 @@ class NovitaAITTIBinding(LollmsTTIBinding):
     """Novita.ai TTI binding for LoLLMS"""
 
     def __init__(self, **kwargs):
-        super().__init__(binding_name=BindingName)
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
         self.config = kwargs
         self.api_key = self.config.get("api_key") or os.environ.get("NOVITA_API_KEY")
         if not self.api_key:

lollms_client/tti_bindings/openai/__init__.py

@@ -6,7 +6,7 @@ from io import BytesIO
 from ascii_colors import trace_exception
 from openai import OpenAI
 from lollms_client.lollms_tti_binding import LollmsTTIBinding
-
+import os
 BindingName = "OpenAITTIBinding"
 
 
@@ -50,19 +50,18 @@ class OpenAITTIBinding(LollmsTTIBinding):
 
     def __init__(
         self,
-        model: str = "gpt-image-1",
-        api_key: Optional[str] = None,
-        size: str = "1024x1024",
-        n: int = 1,
-        quality: str = "standard",
         **kwargs,
     ):
-        self.client = OpenAI(api_key=api_key)
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
+        self.client = OpenAI(api_key=kwargs.get("api_key" or os.environ.get("OPENAI_API_KEY")))
         self.global_params = {
-            "model": model,
-            "size": size,
-            "n": n,
-            "quality": quality,
+            "model": kwargs.get("model_name") or "gpt-image-1",
+            "size": kwargs.get("size", "1024x1024"),
+            "n": kwargs.get("n", 1),
+            "quality": kwargs.get("quality", "standard"),
         }
 
     def _resolve_param(self, name: str, kwargs: Dict[str, Any], default: Any) -> Any:

lollms_client/tti_bindings/stability_ai/__init__.py

@@ -33,8 +33,10 @@ class StabilityAITTIBinding(LollmsTTIBinding):
     """Stability AI TTI binding for LoLLMS"""
 
     def __init__(self, **kwargs):
-        super().__init__(binding_name=BindingName)
-        self.config = kwargs
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)
         self.api_key = self.config.get("api_key") or os.environ.get("STABILITY_API_KEY")
         if not self.api_key:
             raise ValueError("Stability AI API key is required. Please set it in the configuration or as STABILITY_API_KEY environment variable.")

lollms_client/ttm_bindings/audiocraft/__init__.py

@@ -75,21 +75,16 @@ DEFAULT_AUDIOCRAFT_MODELS = [
 
 class AudioCraftTTMBinding(LollmsTTMBinding):
     def __init__(self,
-                 model_name: str = "facebook/musicgen-small", # HF ID or local path
-                 device: Optional[str] = None, # "cpu", "cuda", "mps", or None for auto
-                 output_format: str = "wav", # 'wav', 'mp3' (mp3 needs ffmpeg via audiocraft)
-                 # Catch LollmsTTMBinding standard args
-                 host_address: Optional[str] = None, # Not used by local binding
-                 service_key: Optional[str] = None,  # Not used by local binding
-                 verify_ssl_certificate: bool = True,# Not used by local binding
-                 **kwargs): # Catch-all for future compatibility or specific audiocraft params
-
-        super().__init__(binding_name="audiocraft")
+                 **kwargs):
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)    
 
         if not _audiocraft_installed_with_correct_torch:
             raise ImportError(f"AudioCraft TTM binding dependencies not met. Please ensure 'audiocraft', 'torch', 'torchaudio', 'scipy', 'numpy' are installed. Error: {_audiocraft_installation_error}")
 
-        self.device = device
+        self.device = kwargs.get("device")  # "cuda", "mps", "cpu", or None for auto-detect
         if self.device is None: # Auto-detect if not specified by user
             if torch.cuda.is_available():
                 self.device = "cuda"
@@ -117,7 +112,7 @@ class AudioCraftTTMBinding(LollmsTTMBinding):
             ASCIIColors.warning(f"Unsupported output_format '{self.output_format}'. Defaulting to 'wav'.")
             self.output_format = "wav"
         
-        self._load_audiocraft_model(model_name)
+        self._load_audiocraft_model(kwargs.get("model_name") or "facebook/musicgen-small")
 
     def _load_audiocraft_model(self, model_name_to_load: str):
         if self.model is not None and self.loaded_model_name == model_name_to_load:

lollms_client/ttm_bindings/beatoven_ai/__init__.py

@@ -16,9 +16,13 @@ BindingName = "BeatovenAITTMBinding"
 class BeatovenAITTMBinding(LollmsTTMBinding):
     """A Text-to-Music binding for the Beatoven.ai API."""
 
-    def __init__(self, **kwargs):
-        super().__init__(binding_name=BindingName, **kwargs)
-        self.api_key = self.settings.get("api_key") or os.environ.get("BEATOVEN_API_KEY")
+    def __init__(self,
+                 **kwargs):
+        # Prioritize 'model_name' but accept 'model' as an alias from config files.
+        if 'model' in kwargs and 'model_name' not in kwargs:
+            kwargs['model_name'] = kwargs.pop('model')
+        super().__init__(binding_name=BindingName, config=kwargs)    
+        self.api_key = self.config.get("api_key") or os.environ.get("BEATOVEN_API_KEY")
         if not self.api_key:
             raise ValueError("Beatoven.ai API key is required. Please set it in config or as BEATOVEN_API_KEY env var.")
         self.base_url = "https://api.beatoven.ai/api/v1"