Update app.py

app.py

@@ -31,9 +31,7 @@ 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
@@ -43,6 +41,15 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 logger.info(f"Using output directory: {OUTPUT_DIR}")
 os.environ["COQUI_TOS_AGREED"] = "1"
 
+# Initialize TTS model at the top
+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))
+    tts = None
+
 # Define Pydantic model for slide data
 class Slide(BaseModel):
     title: str
@@ -341,6 +348,16 @@ async def on_generate(api_service, api_key, serpapi_key, title, topic, instructi
         """
         return
 
+    if tts is None:
+        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;">The TTS model failed to initialize at startup.</p>
+            <p>Please ensure the Coqui TTS model is properly installed and try restarting the application.</p>
+        </div>
+        """
+        return
+
     model_client = get_model_client(api_service, api_key)
     
     total_slides = num_slides  # Use exactly the number of slides from input
@@ -861,7 +878,7 @@ with gr.Blocks(title="Agent Feynman") as demo:
                 generate_btn = gr.Button("Generate Lecture")
         with gr.Column(scale=2):
             default_slide_html = """
-            <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;">
+            <div style="display: flex; flex-direction: column; justify-content: center; align-items: center; height: 100%; minISlide-height: 700px; padding: 20px; text-align: center; border: 1px solid #ddd; border-radius: 8px;">
                 <h2 style="font-style: italic; color: #555;">Waiting for lecture content...</h2>
                 <p style="margin-top: 10px; font-size: 16px;">Please Generate lecture content via the form on the left first before lecture begins</p>
             </div>