Update app.py

app.py

@@ -215,35 +215,33 @@ async def startup_event():
 
 @app.post("/generate")
 async def generate_speech(request: TTSRequest):
-    """Generate speech from text with improved error handling"""
+    """Generate speech from text with detailed timing"""
     try:
-        if not service:
-            raise HTTPException(status_code=503, detail="Service not initialized")
+        total_start = time.time()
+        logger.info(f"\nReceived request for text: {request.text[:50]}...")
         
-        logger.info(f"Received request for text: {request.text[:50]}...")
-        
-        start_time = time.time()
+        # Model processing time
+        model_start = time.time()
         wav = service.generate_speech(request.text)
-        processing_time = time.time() - start_time
-        
-        logger.info(f"Processing time: {processing_time:.2f} seconds")
+        model_time = time.time() - model_start
         
-        # Convert to bytes
+        # Audio conversion time
+        conversion_start = time.time()
         buffer = io.BytesIO()
         np.save(buffer, wav.astype(np.float32))
+        audio_base64 = base64.b64encode(buffer.getvalue()).decode()
+        conversion_time = time.time() - conversion_start
         
-        return {
-            "status": "success",
-            "audio": base64.b64encode(buffer.getvalue()).decode(),
-            "processing_time": processing_time
+        # Total processing time
+        total_time = time.time() - total_start
+        
+        timing_info = {
+            "total_processing_time": round(total_time, 2),
+            "model_processing_time": round(model_time, 2),
+            "audio_conversion_time": round(conversion_time, 2)
         }
         
-    except ValueError as e:
-        logger.error(f"Validation error: {str(e)}")
-        raise HTTPException(status_code=400, detail=str(e))
-    except Exception as e:
-        logger.error(f"Error in generate_speech endpoint: {str(e)}")
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.info(f"Timing breakdown: {timing_info}")
 
 @app.get("/health")
 async def health_check():