Merge pull request #572 from ShaerWare/server/phase-4.5-admin-endpoints

shikhalev · web-flow · commit edb43de7c32f · 2026-03-13T16:42:43.000+05:00
Phase 4.5: Extract all remaining admin endpoints from orchestrator
diff --git a/modules/llm/router_models.py b/modules/llm/router_models.py
@@ -0,0 +1,113 @@
+"""HuggingFace model management endpoints.
+
+List, scan, download, delete, and search models.
+GPU-only — not registered in cloud deployment mode.
+"""
+
+import logging
+
+from fastapi import APIRouter, HTTPException, Request
+
+from model_manager import get_model_manager
+
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/admin/models", tags=["models"])
+
+
+@router.get("/list")
+async def admin_list_models():
+    """Список всех локальных моделей"""
+    manager = get_model_manager()
+    return {"models": manager.get_cached_models()}
+
+
+@router.post("/scan")
+async def admin_scan_models(request: Request):
+    """Запустить сканирование моделей"""
+    data = await request.json() if request.headers.get("content-type") == "application/json" else {}
+    include_system = data.get("include_system", False)
+
+    manager = get_model_manager()
+    if manager.scan_all_models(include_system=include_system):
+        return {"status": "ok", "message": "Scan started"}
+    else:
+        return {"status": "error", "message": "Scan already in progress"}
+
+
+@router.post("/scan/cancel")
+async def admin_cancel_scan():
+    """Отменить сканирование"""
+    manager = get_model_manager()
+    manager.cancel_scan()
+    return {"status": "ok", "message": "Scan cancelled"}
+
+
+@router.get("/scan/status")
+async def admin_scan_status():
+    """Статус сканирования"""
+    manager = get_model_manager()
+    return {"status": manager.get_scan_progress()}
+
+
+@router.post("/download")
+async def admin_download_model(request: Request):
+    """Скачать модель с HuggingFace"""
+    data = await request.json()
+    repo_id = data.get("repo_id")
+    revision = data.get("revision", "main")
+
+    if not repo_id:
+        raise HTTPException(status_code=400, detail="repo_id required")
+
+    manager = get_model_manager()
+    if manager.download_model(repo_id, revision):
+        return {"status": "ok", "message": f"Download started: {repo_id}"}
+    else:
+        return {"status": "error", "message": "Download already in progress"}
+
+
+@router.post("/download/cancel")
+async def admin_cancel_download():
+    """Отменить загрузку"""
+    manager = get_model_manager()
+    manager.cancel_download()
+    return {"status": "ok", "message": "Download cancelled"}
+
+
+@router.get("/download/status")
+async def admin_download_status():
+    """Статус загрузки"""
+    manager = get_model_manager()
+    return {"status": manager.get_download_progress()}
+
+
+@router.delete("/delete")
+async def admin_delete_model(path: str):
+    """Удалить модель"""
+    manager = get_model_manager()
+    result = manager.delete_model(path)
+    if result["status"] == "ok":
+        return result
+    else:
+        raise HTTPException(status_code=400, detail=result.get("error", "Delete failed"))
+
+
+@router.get("/search")
+async def admin_search_huggingface(query: str, limit: int = 20):
+    """Поиск моделей на HuggingFace"""
+    manager = get_model_manager()
+    results = manager.search_huggingface(query, limit)
+    return {"results": results}
+
+
+@router.get("/details/{repo_id:path}")
+async def admin_get_model_details(repo_id: str):
+    """Получить детали модели с HuggingFace"""
+    manager = get_model_manager()
+    details = manager.get_model_details(repo_id)
+    if details:
+        return {"details": details}
+    else:
+        raise HTTPException(status_code=404, detail="Model not found")
diff --git a/modules/monitoring/router_logs.py b/modules/monitoring/router_logs.py
@@ -0,0 +1,53 @@
+"""Log viewing endpoints.
+
+List, read, and stream application logs.
+"""
+
+import logging
+from typing import Optional
+
+from fastapi import APIRouter, Depends
+from fastapi.responses import StreamingResponse
+
+from auth_manager import User, require_permission
+from service_manager import get_service_manager
+
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/admin/logs", tags=["logs"])
+
+
+@router.get("")
+async def admin_list_logs():
+    """Список доступных логов"""
+    manager = get_service_manager()
+    return {"logs": manager.get_available_logs()}
+
+
+@router.get("/{logfile}")
+async def admin_read_log(
+    logfile: str, lines: int = 100, offset: int = 0, search: Optional[str] = None
+):
+    """Прочитать лог файл"""
+    manager = get_service_manager()
+    return manager.read_log(logfile, lines=lines, offset=offset, search=search)
+
+
+@router.get("/stream/{logfile}")
+async def admin_stream_log(
+    logfile: str,
+    user: User = Depends(require_permission("system", "view")),
+):
+    """SSE streaming логов"""
+    manager = get_service_manager()
+
+    async def generate():
+        async for data in manager.stream_log(logfile):
+            yield f"data: {data}\n\n"
+
+    return StreamingResponse(
+        generate(),
+        media_type="text/event-stream",
+        headers={"Cache-Control": "no-cache", "Connection": "keep-alive"},
+    )
diff --git a/modules/speech/router_voices.py b/modules/speech/router_voices.py
@@ -0,0 +1,205 @@
+"""Voice selection and testing endpoints.
+
+Manages active voice configuration (XTTS/OpenVoice/Piper).
+GPU-only — not registered in cloud deployment mode.
+"""
+
+import logging
+import time
+from pathlib import Path
+
+from fastapi import APIRouter, HTTPException
+from fastapi.responses import FileResponse
+from pydantic import BaseModel
+
+from app.dependencies import get_container
+
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(tags=["voices"])
+
+TEMP_DIR = Path("./temp")
+TEMP_DIR.mkdir(exist_ok=True)
+
+
+class VoiceRequest(BaseModel):
+    voice: str  # anna / marina / marina_openvoice / dmitri / irina
+
+
+@router.get("/admin/voices")
+async def admin_get_voices():
+    """Получить список всех доступных голосов"""
+    container = get_container()
+    voices = []
+
+    # XTTS голос (Анна) - требует GPU CC >= 7.0 (по умолчанию)
+    if container.anna_voice_service:
+        voices.append(
+            {
+                "id": "anna",
+                "name": "Анна (XTTS)",
+                "engine": "xtts",
+                "description": "Клонированный голос Гули (XTTS v2, GPU CC >= 7.0)",
+                "available": True,
+                "samples_count": len(container.anna_voice_service.voice_samples),
+                "default": True,
+            }
+        )
+
+    # XTTS голос (Марина) - требует GPU CC >= 7.0
+    if container.voice_service:
+        voices.append(
+            {
+                "id": "marina",
+                "name": "Марина (XTTS)",
+                "engine": "xtts",
+                "description": "Клонированный голос Лидии (XTTS v2, GPU CC >= 7.0)",
+                "available": True,
+                "samples_count": len(container.voice_service.voice_samples),
+            }
+        )
+
+    # OpenVoice голос (Марина) - работает на GPU CC 6.1+
+    if container.openvoice_service:
+        voices.append(
+            {
+                "id": "marina_openvoice",
+                "name": "Марина (OpenVoice)",
+                "engine": "openvoice",
+                "description": "Клонированный голос (OpenVoice v2, GPU CC 6.1+)",
+                "available": True,
+                "samples_count": len(container.openvoice_service.voice_samples)
+                if container.openvoice_service.voice_samples
+                else 0,
+            }
+        )
+
+    # Piper голоса (CPU)
+    if container.piper_service:
+        piper_voices = container.piper_service.get_available_voices()
+        for voice_id, info in piper_voices.items():
+            voices.append(
+                {
+                    "id": voice_id,
+                    "name": info["name"],
+                    "engine": "piper",
+                    "description": info["description"],
+                    "available": info["available"],
+                }
+            )
+
+    return {
+        "voices": voices,
+        "current": container.current_voice_config,
+    }
+
+
+@router.get("/admin/voice")
+async def admin_get_current_voice():
+    """Получить текущий выбранный голос"""
+    container = get_container()
+    return container.current_voice_config
+
+
+@router.post("/admin/voice")
+async def admin_set_voice(request: VoiceRequest):
+    """Установить активный голос"""
+    container = get_container()
+    voice_id = request.voice.lower()
+
+    if voice_id == "anna":
+        if not container.anna_voice_service:
+            raise HTTPException(
+                status_code=503, detail="XTTS service (Анна) not available (requires GPU CC >= 7.0)"
+            )
+        new_config = {"engine": "xtts", "voice": "anna"}
+        logger.info("🎤 Голос изменён на: Анна (XTTS)")
+
+    elif voice_id == "marina":
+        if not container.voice_service:
+            raise HTTPException(
+                status_code=503,
+                detail="XTTS service (Марина) not available (requires GPU CC >= 7.0)",
+            )
+        new_config = {"engine": "xtts", "voice": "marina"}
+        logger.info("🎤 Голос изменён на: Марина (XTTS)")
+
+    elif voice_id == "marina_openvoice":
+        if not container.openvoice_service:
+            raise HTTPException(status_code=503, detail="OpenVoice service not available")
+        new_config = {"engine": "openvoice", "voice": "marina_openvoice"}
+        logger.info("🎤 Голос изменён на: Марина (OpenVoice)")
+
+    elif voice_id in ["dmitri", "irina"]:
+        if not container.piper_service:
+            raise HTTPException(status_code=503, detail="Piper TTS service not available")
+        piper_voices = container.piper_service.get_available_voices()
+        if voice_id not in piper_voices or not piper_voices[voice_id]["available"]:
+            raise HTTPException(status_code=400, detail=f"Voice model not found: {voice_id}")
+        new_config = {"engine": "piper", "voice": voice_id}
+        logger.info(f"🎤 Голос изменён на: {piper_voices[voice_id]['name']} (Piper)")
+
+    else:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Unknown voice: {voice_id}. Available: anna, marina, marina_openvoice, dmitri, irina",
+        )
+
+    container.current_voice_config = new_config
+    return {"status": "ok", **new_config}
+
+
+@router.post("/admin/voice/test")
+async def admin_test_voice(request: VoiceRequest):
+    """Тестовый синтез выбранным голосом"""
+    container = get_container()
+    voice_id = request.voice.lower()
+    test_text = "Здравствуйте! Это тестовое сообщение для проверки голоса."
+
+    output_path = TEMP_DIR / f"voice_test_{voice_id}_{int(time.time())}.wav"
+
+    try:
+        if voice_id == "anna":
+            if not container.anna_voice_service:
+                raise HTTPException(
+                    status_code=503, detail="XTTS (Анна) not available (requires GPU CC >= 7.0)"
+                )
+            container.anna_voice_service.synthesize_to_file(
+                test_text, str(output_path), preset="natural"
+            )
+
+        elif voice_id == "marina":
+            if not container.voice_service:
+                raise HTTPException(
+                    status_code=503, detail="XTTS (Марина) not available (requires GPU CC >= 7.0)"
+                )
+            container.voice_service.synthesize_to_file(
+                test_text, str(output_path), preset="natural"
+            )
+
+        elif voice_id == "marina_openvoice":
+            if not container.openvoice_service:
+                raise HTTPException(status_code=503, detail="OpenVoice not available")
+            container.openvoice_service.synthesize_to_file(
+                test_text, str(output_path), language="ru"
+            )
+
+        elif voice_id in ["dmitri", "irina"]:
+            if not container.piper_service:
+                raise HTTPException(status_code=503, detail="Piper not available")
+            container.piper_service.synthesize_to_file(test_text, str(output_path), voice=voice_id)
+
+        else:
+            raise HTTPException(
+                status_code=400,
+                detail=f"Unknown voice: {voice_id}. Available: anna, marina, marina_openvoice, dmitri, irina",
+            )
+
+        return FileResponse(output_path, media_type="audio/wav", filename=f"test_{voice_id}.wav")
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"❌ Ошибка тестового синтеза: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
diff --git a/orchestrator.py b/orchestrator.py
