|
|
|
@ -1,62 +1,3 @@ |
|
|
|
""" |
|
|
|
|
|
|
|
API endpoints для управления аудиофайлами (регистрация и пакетная обработка) |
|
|
|
|
|
|
|
Используется Calls_WEB_Client_main для оркестрации процесса распознавания |
|
|
|
|
|
|
|
""" |
|
|
|
|
|
|
|
from fastapi import APIRouter, HTTPException, Depends, BackgroundTasks |
|
|
|
|
|
|
|
from sqlalchemy.orm import Session |
|
|
|
|
|
|
|
from pydantic import BaseModel |
|
|
|
|
|
|
|
from typing import Optional, List |
|
|
|
|
|
|
|
import os |
|
|
|
|
|
|
|
import logging |
|
|
|
|
|
|
|
from datetime import datetime |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from apiApp.database import get_db |
|
|
|
|
|
|
|
from apiApp.database.Audio import Audio |
|
|
|
|
|
|
|
from apiApp.config import AUDIOFILES_PATH |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
|
|
audio_management_router = APIRouter() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AudioRegisterRequest(BaseModel): |
|
|
|
|
|
|
|
"""Запрос на регистрацию аудиофайла""" |
|
|
|
|
|
|
|
filename: str |
|
|
|
|
|
|
|
file_path: str # Полный путь к файлу в общей папке audiofiles |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AudioProcessAllRequest(BaseModel): |
|
|
|
|
|
|
|
"""Запрос на пакетное распознавание""" |
|
|
|
|
|
|
|
limit: int = 100 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AudioRegisterResponse(BaseModel): |
|
|
|
|
|
|
|
"""Ответ на регистрацию аудиофайла""" |
|
|
|
|
|
|
|
id: str |
|
|
|
|
|
|
|
filename: str |
|
|
|
|
|
|
|
file_size: int |
|
|
|
|
|
|
|
created_at: datetime |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@audio_management_router.post("/audio/register", response_model=AudioRegisterResponse, status_code=201) |
|
|
|
|
|
|
|
async def register_audio_file( |
|
|
|
|
|
|
|
request: AudioRegisterRequest, |
|
|
|
|
|
|
|
db: Session = Depends(get_db) |
|
|
|
|
|
|
|
): |
|
|
|
|
|
|
|
""" |
|
|
|
|
|
|
|
Регистрация аудиофайла в БД (без копирования файла) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Создаёт запись в таблице Audio для файла, который уже находится |
|
|
|
|
|
|
|
в общей папке audiofiles. НЕ копирует файл, только создаёт запись в БД. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Args: |
|
|
|
|
|
|
|
request: {filename: "in-xxx.wav", file_path: "/app/audiofiles/in-xxx.wav"} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Returns: |
|
|
|
|
|
|
|
201 Created + информация о созданной записи |
|
|
|
|
|
|
|
400 Bad Request если файл уже зарегистрирован |
|
|
|
|
|
|
|
404 Not Found если файл не существует на диске |
|
|
|
|
|
|
|
""" |
|
|
|
|
|
|
|
try: |
|
|
|
|
|
|
|
filename = request.filename |
|
|
|
filename = request.filename |
|
|
|
file_path = request.file_path |
|
|
|
file_path = request.file_path |
|
|
|
|
|
|
|
|
|
|
|
@ -170,16 +111,24 @@ async def process_all_pending_audio( |
|
|
|
} |
|
|
|
} |
|
|
|
""" |
|
|
|
""" |
|
|
|
try: |
|
|
|
try: |
|
|
|
|
|
|
|
from sqlalchemy import exists |
|
|
|
|
|
|
|
|
|
|
|
limit = request.limit |
|
|
|
limit = request.limit |
|
|
|
|
|
|
|
|
|
|
|
logger.info(f"🚀 Поиск Audio без AiConclusion (limit={limit})") |
|
|
|
logger.info(f"🚀 Поиск Audio без AiConclusion (limit={limit})") |
|
|
|
|
|
|
|
|
|
|
|
# Находим все Audio без AiConclusion |
|
|
|
# Находим все Audio без AiConclusion через подзапрос |
|
|
|
|
|
|
|
subquery = db.query(AiConclusion.audio_id).filter( |
|
|
|
|
|
|
|
AiConclusion.audio_id == Audio.id |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
pending_audio = db.query(Audio).filter( |
|
|
|
pending_audio = db.query(Audio).filter( |
|
|
|
Audio.AiConclusion == None |
|
|
|
~exists().where(subquery.exists()) |
|
|
|
).order_by(Audio.index_date.asc()).limit(limit).all() |
|
|
|
).order_by(Audio.index_date.asc()).limit(limit).all() |
|
|
|
|
|
|
|
|
|
|
|
total_pending = db.query(Audio).filter(Audio.AiConclusion == None).count() |
|
|
|
total_pending = db.query(Audio).filter( |
|
|
|
|
|
|
|
~exists().where(subquery.exists()) |
|
|
|
|
|
|
|
).count() |
|
|
|
|
|
|
|
|
|
|
|
if not pending_audio: |
|
|
|
if not pending_audio: |
|
|
|
logger.info("ℹ️ Нет файлов для распознавания") |
|
|
|
logger.info("ℹ️ Нет файлов для распознавания") |
|
|
|
@ -262,8 +211,14 @@ async def get_pending_audio( |
|
|
|
Список файлов, ожидающих распознавания |
|
|
|
Список файлов, ожидающих распознавания |
|
|
|
""" |
|
|
|
""" |
|
|
|
try: |
|
|
|
try: |
|
|
|
|
|
|
|
from sqlalchemy import exists |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
subquery = db.query(AiConclusion.audio_id).filter( |
|
|
|
|
|
|
|
AiConclusion.audio_id == Audio.id |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
pending_audio = db.query(Audio).filter( |
|
|
|
pending_audio = db.query(Audio).filter( |
|
|
|
Audio.AiConclusion == None |
|
|
|
~exists().where(subquery.exists()) |
|
|
|
).order_by(Audio.index_date.asc()).limit(limit).all() |
|
|
|
).order_by(Audio.index_date.asc()).limit(limit).all() |
|
|
|
|
|
|
|
|
|
|
|
files_info = [] |
|
|
|
files_info = [] |
|
|
|
@ -279,7 +234,9 @@ async def get_pending_audio( |
|
|
|
"exists_on_disk": exists |
|
|
|
"exists_on_disk": exists |
|
|
|
}) |
|
|
|
}) |
|
|
|
|
|
|
|
|
|
|
|
total_pending = db.query(Audio).filter(Audio.AiConclusion == None).count() |
|
|
|
total_pending = db.query(Audio).filter( |
|
|
|
|
|
|
|
~exists().where(subquery.exists()) |
|
|
|
|
|
|
|
).count() |
|
|
|
|
|
|
|
|
|
|
|
return { |
|
|
|
return { |
|
|
|
"total_pending": total_pending, |
|
|
|
"total_pending": total_pending, |
|
|
|
@ -304,9 +261,19 @@ async def get_audio_stats(db: Session = Depends(get_db)): |
|
|
|
Статистика по Audio записям |
|
|
|
Статистика по Audio записям |
|
|
|
""" |
|
|
|
""" |
|
|
|
try: |
|
|
|
try: |
|
|
|
|
|
|
|
from sqlalchemy import exists |
|
|
|
|
|
|
|
|
|
|
|
total_audio = db.query(Audio).count() |
|
|
|
total_audio = db.query(Audio).count() |
|
|
|
with_conclusion = db.query(Audio).filter(Audio.AiConclusion != None).count() |
|
|
|
|
|
|
|
without_conclusion = db.query(Audio).filter(Audio.AiConclusion == None).count() |
|
|
|
subquery = db.query(AiConclusion.audio_id).filter( |
|
|
|
|
|
|
|
AiConclusion.audio_id == Audio.id |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
with_conclusion = db.query(Audio).filter( |
|
|
|
|
|
|
|
exists().where(subquery.exists()) |
|
|
|
|
|
|
|
).count() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
without_conclusion = total_audio - with_conclusion |
|
|
|
|
|
|
|
|
|
|
|
# Проверяем существование файлов на диске |
|
|
|
# Проверяем существование файлов на диске |
|
|
|
all_audio = db.query(Audio).all() |
|
|
|
all_audio = db.query(Audio).all() |
|
|
|
|