Update engines/orpheus_engine.py

engines/orpheus_engine.py

@@ -22,24 +22,11 @@ START_TOKEN_ID = 128259
 END_TOKEN_IDS = [128009, 128260, 128261, 128257]
 CUSTOM_TOKEN_PREFIX = "<custom_token_"
 
-
 class OrpheusVoice:
-    """
-    Represents the configuration for an Orpheus voice.
-    
-    Attributes:
-        name (str): The name of the voice. Must be one of the AVAILABLE_VOICES.
-    
-    Raises:
-        ValueError: If the voice name provided is not in AVAILABLE_VOICES.
-    """
-    def __init__(self, name: str):
-        # if name not in AVAILABLE_VOICES:
-        #     raise ValueError(f"Invalid voice '{name}'. Available voices: {AVAILABLE_VOICES}")
-        self.name = name
+    def __init__(self, name: str, gender: str | None = None):
+        self.name   = name
+        self.gender = gender    # optional, falls du es anzeigen willst
 
-    def __repr__(self):
-        return f"OrpheusVoice(name='{self.name}')"
 
 
 class OrpheusEngine(BaseEngine):
@@ -319,34 +306,43 @@ class OrpheusEngine(BaseEngine):
         logging.info("Returning None after failed conversion.")
         return None
 
-    def get_voices(self):
-        """
-        Retrieve the list of available voices.
 
-        Returns:
-            list: A list of OrpheusVoice instances for each available voice.
-        """
-        return [OrpheusVoice(name) for name in AVAILABLE_VOICES]
-
-    def set_voice(self, voice: Union[str, OrpheusVoice]):
-        """
-        Set the current voice for synthesis.
-
-        Args:
-            voice (Union[str, OrpheusVoice]): The voice name or an OrpheusVoice instance.
-
-        Raises:
-            ValueError: If the provided voice name is invalid.
-            TypeError: If the voice argument is neither a string nor an OrpheusVoice instance.
-        """
-        if isinstance(voice, str):
-            # if voice not in AVAILABLE_VOICES:
-            #     raise ValueError(f"Invalid voice '{voice}'")
-            self.voice = OrpheusVoice(voice)
-        elif isinstance(voice, OrpheusVoice):
-            self.voice = voice
-        else:
-            raise TypeError("Voice must be a string or an OrpheusVoice instance.")
+    _SPEAKERS = [
+        # männlich
+        OrpheusVoice("Jakob",       "m"),
+        OrpheusVoice("Anton",       "m"),
+        OrpheusVoice("Julian",      "m"),
+        OrpheusVoice("Jan",         "m"),
+        OrpheusVoice("Alexander",   "m"),
+        OrpheusVoice("Emil",        "m"),
+        OrpheusVoice("Ben",         "m"),
+        OrpheusVoice("Elias",       "m"),
+        OrpheusVoice("Felix",       "m"),
+        OrpheusVoice("Jonas",       "m"),
+        OrpheusVoice("Noah",        "m"),
+        OrpheusVoice("Maximilian",  "m"),
+        # weiblich
+        OrpheusVoice("Sophie",      "f"),
+        OrpheusVoice("Marie",       "f"),
+        OrpheusVoice("Mia",         "f"),
+        OrpheusVoice("Maria",       "f"),
+        OrpheusVoice("Sophia",      "f"),
+        OrpheusVoice("Lina",        "f"),
+        OrpheusVoice("Lea",         "f"),
+    ]
+
+    # -----------------------------------------------------------
+    # Public API, damit das FastAPI-Backend sie abfragen kann
+    # -----------------------------------------------------------
+    def get_voices(self) -> list[OrpheusVoice]:
+        """Return a list of available voices for the Kartoffel checkpoint."""
+        return self._SPEAKERS
+
+    # In set_voice() akzeptieren wir einfach den Namen:
+    def set_voice(self, voice_name: str) -> None:
+        if voice_name not in [v.name for v in self._SPEAKERS]:
+            raise ValueError(f"Unknown Orpheus speaker '{voice_name}'")
+        self.current_voice = voice_name
 
     def set_voice_parameters(self, **kwargs):
         """