Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
@traderpedroso
traderpedroso committed Oct 17, 2024
1 parent bc487ce commit 4befddd
Showing 1 changed file with 12 additions and 113 deletions.
125 changes: 12 additions & 113 deletions app.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,115 +5,21 @@
from fastapi.responses import StreamingResponse
from io import BytesIO
import uvicorn
from fast_langdetect import detect

app = FastAPI()

# Definir mapeamento de vozes por idioma
# Definir mapeamento de vozes (usando apenas multilíngues)
voice_mapping = {
"en": { # English (British)
"alloy": "en-GB-RyanNeural",
"echo": "en-GB-LibbyNeural",
"fable": "en-GB-SoniaNeural",
"onyx": "en-GB-ElliotNeural",
"nova": "en-GB-MaisieNeural",
"shimmer": "en-GB-AbbieNeural",
},
"pt": { # Portuguese (Brazil)
"alloy": "pt-BR-AntonioNeural",
"echo": "pt-BR-FranciscaNeural",
"fable": "pt-BR-ThalitaNeural",
"onyx": "en-US-AndrewMultilingualNeural",
"nova": "en-US-AvaMultilingualNeural",
"shimmer": "en-US-EmmaMultilingualNeural",
},
"es": { # Spanish (Spain)
"alloy": "es-ES-AlvaroNeural",
"echo": "es-ES-ElviraNeural",
"fable": "es-ES-XimenaNeural", # Substituí LuciaNeural por XimenaNeural (disponível na sua lista)
"onyx": "es-ES-GonzaloNeural",
"nova": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
},
"fr": { # French (France)
"alloy": "fr-FR-HenriNeural", # Substituí DenisNeural por HenriNeural (disponível na sua lista)
"echo": "fr-FR-EloiseNeural",
"fable": "fr-FR-DeniseNeural", # Adicionei DeniseNeural
"onyx": "fr-FR-RemyMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "fr-FR-VivienneMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
},
"it": { # Italian (Italy)
"alloy": "it-IT-DiegoNeural",
"echo": "it-IT-ElsaNeural", # Substituí FabianaNeural por ElsaNeural (disponível na sua lista)
"fable": "it-IT-IsabellaNeural",
"onyx": "it-IT-GiuseppeNeural",
"nova": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
},
"de": { # German (Germany)
"alloy": "de-DE-ConradNeural",
"echo": "de-DE-KatjaNeural",
"fable": "de-DE-AmalaNeural",
"onyx": "de-DE-FlorianMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "de-DE-SeraphinaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
},
"nl": { # Dutch (Netherlands)
"alloy": "nl-NL-ColetteNeural",
"echo": "nl-NL-FennaNeural",
"fable": "nl-NL-MaartenNeural",
"onyx": "en-US-AndrewMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
},
"ru": { # Russian (Russia)
"alloy": "ru-RU-DmitryNeural",
"echo": "ru-RU-SvetlanaNeural",
"fable": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
"onyx": "en-US-AndrewMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-BrianMultilingualNeural", # Adicionei uma voz multilíngue
},
"zh": { # Chinese (Simplified)
"alloy": "zh-CN-YunxiNeural", # Reorganizei as vozes para ter 6 distintas
"echo": "zh-CN-XiaoxiaoNeural",
"fable": "zh-CN-XiaoyiNeural",
"onyx": "zh-CN-YunxiaNeural",
"nova": "zh-CN-YunyangNeural",
"shimmer": "zh-CN-liaoning-XiaobeiNeural",
},
"ja": { # Japanese
"alloy": "ja-JP-KeitaNeural",
"echo": "ja-JP-NanamiNeural",
"fable": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
"onyx": "en-US-AndrewMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-BrianMultilingualNeural", # Adicionei uma voz multilíngue
},
"ko": { # Korean
"alloy": "ko-KR-HyunsuNeural", # Usei as vozes disponíveis na sua lista
"echo": "ko-KR-SunHiNeural",
"fable": "ko-KR-InJoonNeural",
"onyx": "en-US-AndrewMultilingualNeural", # Adicionei uma voz multilíngue
"nova": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
"shimmer": "en-US-EmmaMultilingualNeural", # Adicionei uma voz multilíngue
},
"ar": { # Arabic
"alloy": "ar-SA-HamedNeural",
"echo": "ar-SA-ZariyahNeural",
"fable": "ar-EG-SalmaNeural", # Usei uma voz de outro dialeto árabe
"onyx": "ar-AE-HamdanNeural", # Usei uma voz de outro dialeto árabe
"nova": "ar-KW-NouraNeural", # Usei uma voz de outro dialeto árabe
"shimmer": "en-US-AvaMultilingualNeural", # Adicionei uma voz multilíngue
"default": { # Vozes multilíngues para todos os idiomas
"alloy": "en-US-AndrewMultilingualNeural",
"echo": "en-US-AvaMultilingualNeural",
"fable": "en-US-EmmaMultilingualNeural",
"onyx": "en-US-BrianMultilingualNeural",
"nova": "fr-FR-VivienneMultilingualNeural",
"shimmer": "fr-FR-RemyMultilingualNeural",
}
}

def get_first_line(input_text: str) -> str:
first_line = input_text.split("\n", 1)[0].strip() # Pega a primeira linha e remove espaços
if first_line.startswith("##"):
first_line = first_line.lstrip("#").strip() # Remove os '#' do início
return first_line

# Definir modelo de entrada para TTS
class TTSRequest(BaseModel):
Expand All @@ -136,18 +42,11 @@ async def text_to_speech(input_text, voice_short_name):
async def generate_speech(request: TTSRequest):
if not request.input.strip():
raise HTTPException(status_code=400, detail="Texto não pode estar vazio.")
if request.voice not in voice_mapping["en"]:
if request.voice not in voice_mapping["default"]:
raise HTTPException(status_code=400, detail="A chave de voz não é válida.")

# Detectar idioma
first_line = get_first_line(request.input)
detected_lang = detect(first_line)["lang"]

if detected_lang not in voice_mapping:
raise HTTPException(status_code=400, detail="Idioma não suportado.")

# Selecionar voz com base no idioma
selected_voice = voice_mapping[detected_lang][request.voice]
# Selecionar voz com base no idioma (agora sempre "default")
selected_voice = voice_mapping["default"][request.voice]

# Gerar o arquivo de áudio
audio_path = await text_to_speech(request.input.lstrip("#").strip(), selected_voice)
Expand All @@ -167,4 +66,4 @@ async def generate_speech(request: TTSRequest):
if __name__ == "__main__":
host = "0.0.0.0"
port = 80
uvicorn.run(app, host=host, port=port)
uvicorn.run(app, host=host, port=port)

0 comments on commit 4befddd

Please sign in to comment.