Update app.py

app.py

@@ -1,69 +1,79 @@
 import os
 import io
+import traceback
+import numpy as np
 import soundfile as sf
-from flask import Flask, request, jsonify, send_file
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import Response
+from pydantic import BaseModel
 from supertonic import TTS
+import uvicorn
 
-app = Flask(__name__)
+app = FastAPI(title="Supertonic TTS API")
 
-# 1. Загружаем модель ГЛОБАЛЬНО при старте приложения
+# Модели для валидации запросов
+class TTSRequest(BaseModel):
+    text: str
+    lang: str = "ru"
+    voice: str = "M2"
+
+# Глобальная загрузка модели
 print("Загрузка модели Supertonic TTS...")
 tts = TTS(auto_download=True)
-# Предзагружаем дефолтный стиль, чтобы экономить время на запросах
 default_style = tts.get_voice_style(voice_name="M2")
 print("Модель успешно загружена и готова к работе!")
 
-@app.route('/')
-def index():
-    return jsonify({
+@app.get("/")
+async def root():
+    return {
         "status": "ok",
         "message": "Supertonic TTS API is running",
-        "usage": "Отправьте POST запрос на /api/tts с JSON: {'text': 'ваш текст', 'lang': 'ru', 'voice': 'M2'}"
-    })
-
-@app.route('/api/tts', methods=['POST'])
-def synthesize():
-    data = request.get_json()
-    
-    if not data or 'text' not in data:
-        return jsonify({"error": "Нужно передать JSON с полем 'text'"}), 400
-
-    text = data.get('text')
-    # В вашем примере текст на русском, а lang="en". Я вынес это в параметры.
-    lang = data.get('lang', 'ru') 
-    voice_name = data.get('voice', 'M2')
+        "docs": "/docs",
+        "usage": "POST /api/tts с JSON: {'text': 'ваш текст', 'lang': 'ru', 'voice': 'M2'}"
+    }
 
+@app.post("/api/tts")
+async def synthesize(request: TTSRequest):
     try:
         # Получаем стиль голоса
-        if voice_name == "M2":
+        if request.voice == "M2":
             style = default_style
         else:
-            style = tts.get_voice_style(voice_name=voice_name)
+            style = tts.get_voice_style(voice_name=request.voice)
 
         # Синтез
-        wav, duration = tts.synthesize(text, voice_style=style, lang=lang)
+        wav, duration = tts.synthesize(request.text, voice_style=style, lang=request.lang)
 
-        # 2. Конвертируем numpy array в WAV прямо в оперативной памяти
+        # Конвертация тензоров в numpy если нужно
+        if hasattr(wav, 'cpu'):
+            wav = wav.cpu().numpy()
+        elif hasattr(wav, 'numpy'):
+            wav = wav.numpy()
+            
+        wav = np.asarray(wav, dtype=np.float32)
+
+        # Получаем sample rate
+        sample_rate = getattr(tts, 'sample_rate', 24000)
+
+        # Записываем в память
         out = io.BytesIO()
-        # Узнаем sample_rate модели (обычно 24000 или 44100). 
-        # Если звук идет "мультяшным" голосом или слишком медленный, поменяйте 24000 на нужный.
-        sample_rate = getattr(tts, 'sample_rate', 24000) 
-        
-        sf.write(out, wav, samplerate=sample_rate, format='WAV')
-        out.seek(0)
+        sf.write(out, wav, samplerate=sample_rate, format='WAV', subtype='PCM_16')
+        audio_bytes = out.getvalue()
 
-        # Отдаем аудиофайл как HTTP-ответ
-        return send_file(
-            out,
-            mimetype='audio/wav',
-            as_attachment=False,
-            download_name='speech.wav'
+        # Возвращаем аудио
+        return Response(
+            content=audio_bytes,
+            media_type='audio/wav',
+            headers={
+                "Content-Disposition": f"attachment; filename=speech.wav",
+                "X-Audio-Duration": str(round(duration, 2))
+            }
         )
         
     except Exception as e:
-        return jsonify({"error": f"Ошибка генерации: {str(e)}"}), 500
+        traceback.print_exc()
+        raise HTTPException(status_code=500, detail=f"Ошибка генерации: {str(e)}")
 
 if __name__ == '__main__':
-    # Hugging Face автоматически передает порт через переменную окружения PORT
     port = int(os.environ.get('PORT', 7860))
-    app.run(host='0.0.0.0', port=port)
+    uvicorn.run(app, host='0.0.0.0', port=port)