Update app.py

app.py

@@ -31,7 +31,9 @@ logging.basicConfig(
     ]
 )
 logger = logging.getLogger(__name__)
-
+device = "cuda" if torch.cuda.is_available() else "cpu"
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
+logger.info("TTS model initialized on %s", device)
 # Set up environment
 if os.path.exists("/tmp"):
     OUTPUT_DIR = "/tmp/outputs"  # Use /tmp for Huggingface Spaces
@@ -339,22 +341,6 @@ async def on_generate(api_service, api_key, serpapi_key, title, topic, instructi
         """
         return
 
-    tts = None
-    try:
-        device = "cuda" if torch.cuda.is_available() else "cpu"
-        tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
-        logger.info("TTS model initialized on %s", device)
-    except Exception as e:
-        logger.error("Failed to initialize TTS model: %s", str(e))
-        yield f"""
-        <div style="display: flex; flex-direction: column; justify-content: center; align-items: center; height: 100%; min-height: 700px; padding: 20px; text-align: center; border: 1px solid #ddd; border-radius: 8px;">
-            <h2 style="color: #d9534f;">TTS model initialization failed</h2>
-            <p style="margin-top: 20px;">Error: {str(e)}</p>
-            <p>Please ensure the Coqui TTS model is properly installed and try again.</p>
-        </div>
-        """
-        return
-
     model_client = get_model_client(api_service, api_key)
     
     total_slides = num_slides  # Use exactly the number of slides from input