87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
"""Auto-casting : attribue une voix distincte a chaque personnage.
|
|
|
|
Strategie deterministe :
|
|
- Narrateur : voix FR native par defaut (ff_siwis), sinon premiere voix.
|
|
- Personnages : voix du meme genre, distinctes tant qu'il en reste ; au-dela on
|
|
recycle en repartissant le plus equitablement possible. Genre inconnu -> pool
|
|
mixte. L'ordre (tri par nom) garantit la reproductibilite.
|
|
L'utilisateur pourra surcharger ces choix dans l'UI.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from collections import Counter
|
|
from typing import Optional
|
|
|
|
from ..models import Cast, Character, Voicebank
|
|
|
|
# Voix narrateur preferee (FR native).
|
|
PREFERRED_NARRATOR = "fr_f_siwis"
|
|
|
|
|
|
def _pick_pool(vb: Voicebank, gender: Optional[str], narrator_id: str) -> list[str]:
|
|
"""Voix candidates : on privilegie STRICTEMENT le genre (quitte a reutiliser).
|
|
|
|
On ne croise le genre que si aucune voix du bon genre n'existe. Le narrateur
|
|
est exclu tant qu'il reste d'autres options, pour le distinguer.
|
|
"""
|
|
same = [e.id for e in vb.by_gender(gender)] if gender in ("male", "female") else []
|
|
pool = same if same else [e.id for e in vb.entries]
|
|
non_narrator = [vid for vid in pool if vid != narrator_id]
|
|
return non_narrator or pool # garde le narrateur seulement s'il est seul
|
|
|
|
|
|
def assign_voices(
|
|
characters: list[Character],
|
|
vb: Voicebank,
|
|
*,
|
|
narrator_voice_id: Optional[str] = None,
|
|
respect_existing: bool = False,
|
|
) -> Cast:
|
|
"""Renvoie un Cast avec narrateur + voix par personnage (mutation des chars).
|
|
|
|
`respect_existing=True` conserve les voix deja attribuees (overrides UI) ;
|
|
sinon tout est re-calcule (auto-casting frais).
|
|
"""
|
|
if not vb.entries:
|
|
return Cast(narrator_voice_id=narrator_voice_id, characters=characters)
|
|
|
|
narrator_id = narrator_voice_id or (
|
|
PREFERRED_NARRATOR if vb.by_id(PREFERRED_NARRATOR) else vb.entries[0].id)
|
|
|
|
usage: Counter[str] = Counter()
|
|
usage[narrator_id] += 1 # le narrateur compte deja
|
|
|
|
for ch in sorted(characters, key=lambda c: c.name.lower()):
|
|
if respect_existing and ch.voice_id and vb.by_id(ch.voice_id):
|
|
usage[ch.voice_id] += 1
|
|
continue # respecte une attribution existante (override utilisateur)
|
|
pool = _pick_pool(vb, ch.gender, narrator_id)
|
|
# Choisit la voix la moins utilisee du pool (donc une voix neuve d'abord).
|
|
best = min(pool, key=lambda vid: (usage[vid], pool.index(vid)))
|
|
ch.voice_id = best
|
|
usage[best] += 1
|
|
|
|
return Cast(narrator_voice_id=narrator_id, characters=characters)
|
|
|
|
|
|
def resolve_speaker_voice(
|
|
speaker: str, cast: Cast, vb: Voicebank
|
|
) -> Optional[str]:
|
|
"""Mappe un nom de locuteur (segment) vers un id de voix.
|
|
|
|
Matche d'abord par nom/alias exact (rapide), puis en dernier recours par
|
|
rapprochement heuristique de tokens (ex: un "Jim" qui n'aurait pas encore
|
|
ete absorbe comme alias de "James Holden").
|
|
"""
|
|
if speaker == "narrateur":
|
|
return cast.narrator_voice_id
|
|
low = speaker.lower()
|
|
for ch in cast.characters:
|
|
if ch.name.lower() == low or low in (a.lower() for a in ch.aliases):
|
|
return ch.voice_id
|
|
from .dedup import heuristic_match
|
|
match = heuristic_match(speaker, cast.characters)
|
|
if isinstance(match, Character):
|
|
return match.voice_id
|
|
return None # inconnu -> le rendu repliera sur le narrateur
|