Update audio_generator.py

audio_generator.py

@@ -1,54 +1,34 @@
 import os
 import edge_tts
 import asyncio
-from typing import Optional
+import logging
+from datetime import datetime
 
-# Available free voices (natural-sounding options)
-VOICES = {
-    "female": {
-        "en-US": "en-US-AriaNeural",  # Very natural
-        "en-GB": "en-GB-LibbyNeural",  # British English
-        "es-ES": "es-ES-ElviraNeural"  # Spanish
-    },
-    "male": {
-        "en-US": "en-US-GuyNeural",
-        "en-GB": "en-GB-RyanNeural",
-        "es-ES": "es-ES-AlvaroNeural"
-    }
-}
-
-async def generate_speech(
-    text: str,
-    output_file: str = "output.mp3",
-    voice: Optional[str] = None,
-    rate: str = "+0%",  # Speed adjustment
-    volume: str = "+0%"  # Volume adjustment
-) -> str:
-    """
-    Generate speech using free EdgeTTS
-    Args:
-        text: Input text (max 3000 chars)
-        output_file: Output path
-        voice: Voice code (e.g. 'en-US-AriaNeural')
-        rate: Speaking rate adjustment (-50% to +100%)
-        volume: Volume adjustment (-50% to +50%)
-    """
-    if not voice:
-        voice = VOICES["female"]["en-US"]  # Default to natural female voice
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
+async def _generate_speech(text: str, output_path: str) -> str:
     try:
         communicate = edge_tts.Communicate(
             text=text,
-            voice=voice,
-            rate=rate,
-            volume=volume
+            voice="en-US-AriaNeural",
+            rate="+0%",
+            volume="+0%"
         )
-        await communicate.save(output_file)
-        return output_file
+        await communicate.save(output_path)
+        return output_path
     except Exception as e:
-        raise RuntimeError(f"EdgeTTS error: {str(e)}")
+        logger.error(f"Generation failed: {str(e)}")
+        raise RuntimeError(f"Audio generation error: {str(e)}")
 
-def generate_audio(text: str, filename: str = "output.mp3") -> str:
-    """Synchronous wrapper for async TTS"""
-    os.makedirs(os.path.dirname(filename) or ".", exist_ok=True)
-    return asyncio.run(generate_speech(text, filename))
+def generate_audio(text: str) -> str:
+    """Main entry point for audio generation"""
+    try:
+        os.makedirs("tts_outputs", exist_ok=True)
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_path = os.path.join("tts_outputs", f"tts_{timestamp}.mp3")
+        
+        return asyncio.run(_generate_speech(text, output_path))
+    except Exception as e:
+        logger.error(f"Audio generation failed: {str(e)}")
+        raise