Update genesis/narration.py

genesis/narration.py

@@ -1,40 +1,47 @@
 # genesis/narration.py
+"""
+Narration Module for GENESIS-AI
+Generates audio narration from text using ElevenLabs, with gTTS fallback.
+"""
+
 import os
 import requests
-from typing import Optional
+from gtts import gTTS
 
 ELEVEN_LABS_API_KEY = os.getenv("ELEVEN_LABS_API_KEY")
-VOICE_ID = "EXAVITQu4vr4xnSDxMaL"  # Default voice; change in ElevenLabs dashboard if needed
-
-def narrate_text(text: str) -> Optional[str]:
-    """Convert text to speech using ElevenLabs API and return audio URL."""
-    if not ELEVEN_LABS_API_KEY:
-        print("[Narration] No ElevenLabs API key provided.")
-        return None
+ELEVEN_TTS_URL = "https://api.elevenlabs.io/v1/text-to-speech"
 
+def narrate_text_elevenlabs(text, voice="Rachel"):
+    """Generate narration using ElevenLabs API."""
+    headers = {
+        "xi-api-key": ELEVEN_LABS_API_KEY,
+        "Content-Type": "application/json"
+    }
+    payload = {"text": text, "voice_settings": {"stability": 0.75, "similarity_boost": 0.75}}
     try:
-        url = f"https://api.elevenlabs.io/v1/text-to-speech/{VOICE_ID}"
-        headers = {
-            "xi-api-key": ELEVEN_LABS_API_KEY,
-            "Content-Type": "application/json"
-        }
-        payload = {
-            "text": text,
-            "voice_settings": {
-                "stability": 0.4,
-                "similarity_boost": 0.8
-            }
-        }
-        r = requests.post(url, headers=headers, json=payload, timeout=30)
+        r = requests.post(f"{ELEVEN_TTS_URL}/{voice}", json=payload, headers=headers)
         r.raise_for_status()
-
-        # Save audio locally
-        audio_file = "narration.mp3"
-        with open(audio_file, "wb") as f:
+        output_path = "narration.mp3"
+        with open(output_path, "wb") as f:
             f.write(r.content)
+        return output_path
+    except Exception as e:
+        return {"error": str(e)}
 
-        # In Spaces, you can serve local files directly
-        return audio_file
+def narrate_text_gtts(text, lang="en"):
+    """Fallback narration with gTTS."""
+    try:
+        tts = gTTS(text=text, lang=lang)
+        output_path = "narration_fallback.mp3"
+        tts.save(output_path)
+        return output_path
     except Exception as e:
-        print(f"[Narration] Failed: {e}")
-        return None
+        return {"error": str(e)}
+
+def narrate_text(text):
+    """Try ElevenLabs first, else fallback."""
+    if ELEVEN_LABS_API_KEY:
+        result = narrate_text_elevenlabs(text)
+        if isinstance(result, str):
+            return result
+    return narrate_text_gtts(text)