Apex / app.py
ernestmindres's picture
Update app.py
7d1c198 verified
# app.py
from flask import Flask, render_template, request, send_file, redirect, url_for, Response # Import de Response
from werkzeug.utils import secure_filename
# Changement: On importe la fonction de streaming
from tts_engine import get_available_languages, stream_text_to_audio
import os
import io
import tempfile
import sys # Import pour les logs
app = Flask(__name__)
# Configuration simple pour l'upload
UPLOAD_FOLDER = 'uploads'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # Limite 16MB
# Fichier temporaire pour l'audio généré
TEMP_AUDIO_FILE = "tts_output.wav"
@app.route('/', methods=['GET'])
def index():
"""Route principale: Affiche le formulaire et la liste des langues."""
# Récupérer les langues pour le menu déroulant
languages = get_available_languages()
# Afficher la template HTML
return render_template('index.html', languages=languages)
@app.route('/read', methods=['POST'])
def read_text():
"""Route de traitement: Reçoit le texte/fichier et la langue, génère l'audio en streaming."""
# 1. Collecte des données (inchangé)
text_input = request.form.get('text_input')
voice_id = request.form.get('voice_id')
# Gérer l'upload de fichier
text_to_speak = ""
# ... (Logique de lecture de fichier/texte - inchangée)
if 'file_upload' in request.files and request.files['file_upload'].filename != '':
uploaded_file = request.files['file_upload']
filename = secure_filename(uploaded_file.filename)
if filename.endswith('.txt'):
try:
text_to_speak = uploaded_file.read().decode('utf-8')
except Exception as e:
return f"Erreur de lecture du fichier: {e}", 400
else:
return "Type de fichier non supporté. Veuillez utiliser un fichier .txt.", 400
elif text_input:
text_to_speak = text_input
if not text_to_speak:
return "Veuillez entrer du texte ou télécharger un fichier.", 400
if not voice_id:
voices = get_available_languages()
# Fallback à la première voix disponible
voice_id = next(iter(voices.values()), None)
# 2. Génération de l'audio en streaming
try:
# La fonction retourne un générateur de morceaux audio (bytes, incluant l'en-tête WAV)
audio_stream_generator = stream_text_to_audio(text_to_speak, voice_id)
# 3. Renvoie la réponse en streaming au client
# Le mimetype doit être 'audio/wav' pour que le navigateur lise le flux
return Response(audio_stream_generator, mimetype='audio/wav')
except Exception as e:
print(f"Erreur de génération audio (eSpeak-NG): {e}", file=sys.stderr)
return f"Erreur de génération audio: {e}", 500