Update app.py

app.py

@@ -6,6 +6,7 @@ import shutil
 from datetime import datetime
 from typing import List, Optional
 from pathlib import Path
+from contextlib import asynccontextmanager
 
 import requests
 from fastapi import FastAPI, HTTPException, Form, UploadFile, File
@@ -14,16 +15,47 @@ from pydantic import BaseModel
 import torch
 import numpy as np
 
+# Global state
+tts = None
+model_loaded = False
+current_model = ""
+model_loading = False
+current_voice_style = "default_female"
+voice_cloning_supported = False
+app_startup_time = datetime.now()
+
 # Configure environment
 os.makedirs("/tmp/voices", exist_ok=True)
 os.makedirs("/tmp/output", exist_ok=True)
 
-# Initialize FastAPI app
+# Configuration
+OCI_UPLOAD_API_URL = os.getenv("OCI_UPLOAD_API_URL", "").strip()
+if OCI_UPLOAD_API_URL:
+    OCI_UPLOAD_API_URL = OCI_UPLOAD_API_URL.rstrip('/')
+
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
+    print("=" * 50)
+    print("🚀 TTS API Starting Up...")
+    print(f"✅ Device: {DEVICE}")
+    print(f"🔧 OCI Upload: {OCI_UPLOAD_API_URL or 'Local only'}")
+    print("📝 Models will load on first request (lazy loading)")
+    print("⏰ Startup time:", app_startup_time.isoformat())
+    print("=" * 50)
+    yield
+    # Shutdown
+    print("🛑 TTS API Shutting Down...")
+
+# Initialize FastAPI app with lifespan
 app = FastAPI(
     title="TTS API", 
     description="API for text-to-speech with Coqui TTS",
     docs_url="/",
-    redoc_url=None
+    redoc_url=None,
+    lifespan=lifespan
 )
 
 # Add CORS middleware
@@ -35,25 +67,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Configuration
-OCI_UPLOAD_API_URL = os.getenv("OCI_UPLOAD_API_URL", "").strip()
-if OCI_UPLOAD_API_URL:
-    OCI_UPLOAD_API_URL = OCI_UPLOAD_API_URL.rstrip('/')
-
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-
 print(f"✅ Using device: {DEVICE}")
 print(f"🔧 OCI Upload URL: {OCI_UPLOAD_API_URL or 'Not configured - uploads will be local only'}")
 
-# Global state
-tts = None
-model_loaded = False
-current_model = ""
-model_loading = False
-current_voice_style = "default_female"
-voice_cloning_supported = False
-app_startup_time = datetime.now()
-
 # Pydantic models
 class TTSRequest(BaseModel):
     text: str
@@ -345,6 +361,15 @@ def load_tts_model(voice_style="default_female"):
         model_loading = False
 
 # Health check endpoints - CRITICAL FOR DEPLOYMENT
+@app.get("/")
+async def root():
+    """Root endpoint - redirect to docs"""
+    return {
+        "status": "running", 
+        "service": "TTS API",
+        "message": "Visit /docs for API documentation"
+    }
+
 @app.get("/health")
 async def health_check():
     """Health check endpoint - must respond quickly"""
@@ -680,18 +705,6 @@ async def get_status():
         "uptime": str(datetime.now() - app_startup_time)
     }
 
-# Startup event - NO MODEL LOADING to avoid timeouts
-@app.on_event("startup")
-async def startup_event():
-    """Startup event - no model loading to avoid timeouts"""
-    print("=" * 50)
-    print("🚀 TTS API Starting Up...")
-    print(f"✅ Device: {DEVICE}")
-    print(f"🔧 OCI Upload: {OCI_UPLOAD_API_URL or 'Local only'}")
-    print("📝 Models will load on first request (lazy loading)")
-    print("⏰ Startup time:", app_startup_time.isoformat())
-    print("=" * 50)
-
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000, access_log=False)