Update app.py

app.py

@@ -8,6 +8,21 @@ import json
 import asyncio
 import logging
 
+# ===== MODEL CONFIGURATION =====
+# Einfach zwischen den deutschen Modellen wechseln:
+USE_KARTOFFEL_MODEL = True  # True = Kartoffel, False = Canopy-Deutsch
+
+if USE_KARTOFFEL_MODEL:
+    MODEL_NAME = "SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1"
+    TOKENIZER_NAME = "SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1"
+    DEFAULT_VOICE = "Jakob"
+    print("🥔 Using Kartoffel German Model")
+else:
+    MODEL_NAME = "canopylabs/3b-de-ft-research_release"
+    TOKENIZER_NAME = "canopylabs/3b-de-ft-research_release"
+    DEFAULT_VOICE = "thomas"
+    print("🇩🇪 Using Canopy German Model")
+
 # Add the orpheus-tts module to the path
 sys.path.append(os.path.join(os.path.dirname(__file__), 'orpheus-tts'))
 
@@ -39,10 +54,10 @@ async def startup_event():
     global engine
     try:
         engine = OrpheusModel(
-            model_name="SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1",
-            tokenizer="SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1"
+            model_name=MODEL_NAME,
+            tokenizer=TOKENIZER_NAME
         )
-        logger.info("Orpheus model loaded successfully")
+        logger.info(f"Orpheus model loaded successfully: {MODEL_NAME}")
     except Exception as e:
         logger.error(f"Error loading Orpheus model: {e}")
         raise e
@@ -79,7 +94,7 @@ async def health_check():
 @app.get("/tts")
 async def tts_stream(
     prompt: str = Query(..., description="Text to synthesize"),
-    voice: str = Query("Jakob", description="Voice to use"),
+    voice: str = Query(DEFAULT_VOICE, description="Voice to use"),
     temperature: float = Query(0.4, description="Temperature for generation"),
     top_p: float = Query(0.9, description="Top-p for generation"),
     max_tokens: int = Query(2000, description="Maximum tokens"),
@@ -139,7 +154,7 @@ async def websocket_tts(websocket: WebSocket):
             request = json.loads(data)
             
             prompt = request.get("prompt", "")
-            voice = request.get("voice", "Jakob")
+            voice = request.get("voice", DEFAULT_VOICE)
             temperature = request.get("temperature", 0.4)
             top_p = request.get("top_p", 0.9)
             max_tokens = request.get("max_tokens", 2000)
@@ -211,7 +226,7 @@ async def root():
         "message": "Orpheus TTS Server",
         "endpoints": {
             "health": "/health",
-            "tts_http": "/tts?prompt=your_text&voice=Jakob",
+            "tts_http": f"/tts?prompt=your_text&voice={DEFAULT_VOICE}",
             "tts_websocket": "/ws/tts",
             "voices": "/voices"
         },