Update admin_routes.py

admin_routes.py

@@ -139,6 +139,22 @@ class APIUpdate(BaseModel):
 class TestRequest(BaseModel):
     test_type: str  # "all", "ui", "backend", "integration", "spark"
 
+class EnvironmentUpdate(BaseModel):
+    work_mode: str
+    cloud_token: Optional[str] = None
+    spark_endpoint: str
+    internal_prompt: Optional[str] = None
+    tts_engine: str = "no_tts"
+    tts_engine_api_key: Optional[str] = None
+    stt_engine: str = "no_stt"
+    stt_engine_api_key: Optional[str] = None
+
+class TTSRequest(BaseModel):
+    text: str
+    voice_id: Optional[str] = None
+    model_id: Optional[str] = None
+    output_format: Optional[str] = "mp3_44100_128"
+
 # ===================== Helpers =====================
 def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)) -> str:
     """Verify JWT token and return username"""
@@ -408,9 +424,14 @@ async def get_environment(username: str = Depends(verify_token)):
         "work_mode": env_config.get("work_mode", "on-premise"),
         "cloud_token": env_config.get("cloud_token", ""),
         "spark_endpoint": env_config.get("spark_endpoint", "http://localhost:7861"),
-        "internal_prompt": env_config.get("internal_prompt", "") 
+        "internal_prompt": env_config.get("internal_prompt", ""),
+        "tts_engine": env_config.get("tts_engine", "no_tts"),
+        "tts_engine_api_key": env_config.get("tts_engine_api_key", ""),
+        "stt_engine": env_config.get("stt_engine", "no_stt"),
+        "stt_engine_api_key": env_config.get("stt_engine_api_key", "")
     }
 
+
 @router.put("/environment")
 async def update_environment(
     update: EnvironmentUpdate,
@@ -423,35 +444,113 @@ async def update_environment(
     if update.work_mode in ("gpt4o", "gpt4o-mini"):
         if not update.cloud_token:
             raise HTTPException(status_code=400, detail="OpenAI API key is required for GPT modes")
-        # Basic format check for OpenAI key
         if not update.cloud_token.startswith("sk-") and not update.cloud_token.startswith("enc:"):
             raise HTTPException(status_code=400, detail="Invalid OpenAI API key format")
     elif update.work_mode in ("hfcloud", "cloud"):
         if not update.cloud_token:
             raise HTTPException(status_code=400, detail="Cloud token is required for cloud modes")
     
+    # TTS/STT validation
+    if update.tts_engine not in ("no_tts", "elevenlabs", "blaze"):
+        raise HTTPException(status_code=400, detail="Invalid TTS engine")
+    
+    if update.stt_engine not in ("no_stt", "elevenlabs", "flicker"):
+        raise HTTPException(status_code=400, detail="Invalid STT engine")
+    
+    if update.tts_engine != "no_tts" and not update.tts_engine_api_key:
+        raise HTTPException(status_code=400, detail=f"{update.tts_engine} API key is required")
+    
+    if update.stt_engine != "no_stt" and not update.stt_engine_api_key:
+        raise HTTPException(status_code=400, detail=f"{update.stt_engine} API key is required")
+    
     # Spark endpoint validation
     if update.work_mode not in ("gpt4o", "gpt4o-mini") and not update.spark_endpoint:
         raise HTTPException(status_code=400, detail="Spark endpoint is required for non-GPT modes")
     
+    # Encrypt API keys if needed
+    from encryption_utils import encrypt
+    
     # Update config
     config["config"]["work_mode"] = update.work_mode
     config["config"]["cloud_token"] = update.cloud_token or ""
     config["config"]["spark_endpoint"] = update.spark_endpoint
     config["config"]["internal_prompt"] = update.internal_prompt or ""
+    config["config"]["tts_engine"] = update.tts_engine
+    config["config"]["tts_engine_api_key"] = encrypt(update.tts_engine_api_key) if update.tts_engine_api_key else ""
+    config["config"]["stt_engine"] = update.stt_engine
+    config["config"]["stt_engine_api_key"] = encrypt(update.stt_engine_api_key) if update.stt_engine_api_key else ""
     config["config"]["last_update_date"] = get_timestamp()
     config["config"]["last_update_user"] = username
     
     # Add activity log
     add_activity_log(config, username, "UPDATE_ENVIRONMENT", "config", "environment", 
-                    "environment", f"Changed to {update.work_mode}")
+                    "environment", f"Changed to {update.work_mode}, TTS: {update.tts_engine}, STT: {update.stt_engine}")
     
     # Save
     save_config(config)
     
-    log(f"✅ Environment updated to {update.work_mode} by {username}")
+    log(f"✅ Environment updated to {update.work_mode} with TTS: {update.tts_engine} by {username}")
     return {"success": True}
 
+# ===================== TTS Endpoints =====================
+@router.post("/tts/generate")
+async def generate_tts(
+    request: TTSRequest,
+    username: str = Depends(verify_token)
+):
+    """Generate TTS audio from text"""
+    try:
+        config = load_config()
+        env_config = config.get("config", {})
+        
+        tts_engine = env_config.get("tts_engine", "no_tts")
+        if tts_engine == "no_tts":
+            raise HTTPException(status_code=400, detail="TTS is not configured")
+        
+        # Get TTS provider
+        from config_provider import ConfigProvider
+        cfg = ConfigProvider.get()
+        api_key = cfg.global_config.get_tts_api_key()
+        
+        if not api_key:
+            raise HTTPException(status_code=400, detail="TTS API key not configured")
+        
+        # Import here to avoid circular dependency
+        from tts_interface import create_tts_provider
+        
+        tts_provider = create_tts_provider(tts_engine, api_key)
+        if not tts_provider:
+            raise HTTPException(status_code=500, detail="Failed to create TTS provider")
+        
+        log(f"🎤 Generating TTS for {len(request.text)} characters using {tts_engine}")
+        
+        # Generate audio
+        audio_data = await tts_provider.synthesize(
+            text=request.text,
+            voice_id=request.voice_id,
+            model_id=request.model_id,
+            output_format=request.output_format
+        )
+        
+        # Return audio data
+        from fastapi.responses import Response
+        
+        content_type = "audio/mpeg" if request.output_format.startswith("mp3") else "audio/wav"
+        
+        return Response(
+            content=audio_data,
+            media_type=content_type,
+            headers={
+                "Content-Disposition": f"attachment; filename=tts_output.{request.output_format.split('_')[0]}"
+            }
+        )
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        log(f"❌ TTS generation error: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 # ===================== Project Endpoints =====================
 @router.get("/projects/names")
 def list_enabled_projects():