CORRECT services

services/stt_service.py

@@ -1,13 +1,26 @@
-# stt_service.py
+# services/stt_service.py
 import requests
 from config import GROQ_API_KEY, GROQ_STT_MODEL
 
-def speech_to_text(audio_file_path: str) -> str:
-    headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
-    files = {"file": open(audio_file_path, "rb")}
-    response = requests.post(
-        f"https://api.groq.ai/v1/models/{GROQ_STT_MODEL}/predict",
-        headers=headers,
-        files=files
-    )
-    return response.json().get("text", "")
+def speech_to_text(audio_file: str) -> str:
+    """
+    Convertit un fichier audio en texte via l'API Groq STT.
+    audio_file : str -> chemin du fichier audio
+    """
+    if not GROQ_API_KEY:
+        raise RuntimeError("GROQ_API_KEY is not set in config")
+
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}"
+    }
+
+    files = {"file": open(audio_file, "rb")}
+
+    url = f"https://api.groq.ai/v1/models/{GROQ_STT_MODEL}/predict"
+
+    response = requests.post(url, headers=headers, files=files)
+    response.raise_for_status()
+
+    result = response.json()
+    # Supposons que l'API retourne le texte sous 'text'
+    return result.get("text", "")

services/tts_service.py

@@ -1,15 +1,34 @@
-from pathlib import Path
-from config import client
+# services/tts_service.py
+import requests
+from config import GROQ_API_KEY, GROQ_TTS_MODEL, GROQ_TTS_VOICE, GROQ_TTS_FORMAT
 
-def text_to_speech(text: str, voice: str = "Aaliyah-PlayAI", fmt: str = "wav"):
-    speech_path = Path("audio/temp/output." + fmt)
+def text_to_speech(text: str, output_file: str):
+    """
+    Convertit du texte en audio via l'API Groq TTS.
+    text : str -> texte à convertir
+    output_file : str -> chemin du fichier de sortie (ex: 'output.wav')
+    """
+    if not GROQ_API_KEY:
+        raise RuntimeError("GROQ_API_KEY is not set in config")
 
-    response = client.audio.speech.create(
-        model="playai-tts",
-        voice=voice,
-        response_format=fmt,
-        input=text,
-    )
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
 
-    response.stream_to_file(speech_path)
-    return speech_path
+    payload = {
+        "text": text,
+        "voice": GROQ_TTS_VOICE,
+        "format": GROQ_TTS_FORMAT
+    }
+
+    url = f"https://api.groq.ai/v1/models/{GROQ_TTS_MODEL}/predict"
+
+    response = requests.post(url, headers=headers, json=payload)
+    response.raise_for_status()  # Si erreur, lève une exception
+
+    audio_content = response.content
+    with open(output_file, "wb") as f:
+        f.write(audio_content)
+
+    return output_file