110 lines
4.9 KiB
Python
110 lines
4.9 KiB
Python
# app/api/dashboard.py
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from pydantic import BaseModel
|
|
from typing import List, Dict, Any, Optional
|
|
from app.core.security import require_auth
|
|
from app.core.vad import get_vad_instance, get_stt_instance
|
|
from app.core.status import status_tracker
|
|
from app.core.system_monitor import get_system_load
|
|
|
|
router = APIRouter(prefix="/api/dashboard", tags=["dashboard"])
|
|
|
|
class SpeechData(BaseModel):
|
|
state: str
|
|
speech_prob: float
|
|
speech_threshold: float # Добавлено: порог обнаружения речи
|
|
energy: float
|
|
energy_threshold: float
|
|
speech_detected: bool # Добавлено: признак обнаружения речи
|
|
continuous_frames: int # Добавлено: количество непрерывных кадров речи
|
|
gain: float # Добавлено: текущий коэффициент усиления
|
|
noise_floor: float # Добавлено: уровень шума
|
|
|
|
class RecognizedPhrase(BaseModel):
|
|
text: str
|
|
timestamp: str
|
|
action: str = None
|
|
|
|
class SystemLoad(BaseModel): # Добавлено: модель для системной нагрузки
|
|
cpu_percent: float
|
|
memory_percent: float
|
|
temperature: float
|
|
|
|
@router.get("/speech-data", response_model=SpeechData)
|
|
async def get_speech_data(request: Request, token: str = Depends(require_auth)):
|
|
"""Получение текущих данных о речи и энергии, включая расширенные метрики VAD и STT"""
|
|
try:
|
|
vad = get_vad_instance(request)
|
|
|
|
if not vad:
|
|
return SpeechData(
|
|
state=status_tracker.get_status(),
|
|
speech_prob=0.0,
|
|
speech_threshold=0.0,
|
|
energy=0,
|
|
energy_threshold=0,
|
|
speech_detected=False,
|
|
continuous_frames=0,
|
|
gain=1.0,
|
|
noise_floor=0.0,
|
|
)
|
|
|
|
# Получаем данные из last_result VAD
|
|
last_result = vad.last_result
|
|
# Дополнительные атрибуты VAD (на основе класса VADPolling)
|
|
gain = vad.current_gain
|
|
noise_floor = vad.noise_floor
|
|
continuous_frames = last_result.get('continuous_frames', vad.continuous_speech_frames) # Если есть в last_result или атрибуте
|
|
|
|
|
|
return SpeechData(
|
|
state=status_tracker.get_status(),
|
|
speech_prob=float(last_result.get('speech_ratio', 0.0)),
|
|
speech_threshold=float(last_result.get('speech_threshold', 0.0)),
|
|
energy=float(last_result.get('energy', 0)),
|
|
energy_threshold=float(last_result.get('energy_threshold', 0)),
|
|
speech_detected=bool(last_result.get('speech_detected', False)),
|
|
continuous_frames=int(continuous_frames),
|
|
gain=float(gain),
|
|
noise_floor=float(noise_floor),
|
|
)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
@router.get("/sys-monitor", response_model=SystemLoad) # Добавлено: новый эндпоинт
|
|
async def get_sys_monitor(token: str = Depends(require_auth)):
|
|
"""Получение данных о нагрузке системы (CPU, память, температура)"""
|
|
try:
|
|
load_data = get_system_load()
|
|
return SystemLoad(**load_data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
@router.get("/recent-phrases", response_model=List[RecognizedPhrase])
|
|
async def get_recent_phrases(request: Request, limit: int = 10, token: str = Depends(require_auth)):
|
|
"""Получение последних распознанных фраз"""
|
|
try:
|
|
stt = get_stt_instance(request)
|
|
if not stt or not hasattr(stt, 'get_history'):
|
|
from app.core.history import get_recognition_history
|
|
phrases = get_recognition_history(limit)
|
|
else:
|
|
phrases = stt.get_history(limit)
|
|
return phrases
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
@router.get("/stats", response_model=Dict[str, Any])
|
|
async def get_stats(request: Request, token: str = Depends(require_auth)):
|
|
"""Получение статистики работы системы"""
|
|
try:
|
|
# Здесь будет логика получения статистики работы системы
|
|
# Например, количество распознанных команд, время работы и т.д.
|
|
return {
|
|
"uptime": "0:00:00", # Будет заменено на реальное время работы
|
|
"recognized_phrases_count": 0,
|
|
"commands_executed_count": 0,
|
|
"system_load": 0.0
|
|
}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e)) |