Update app.py

app.py

@@ -115,6 +115,7 @@ def extract_audio_from_file(input_bytes):
 
 @app.post("/transcribe")
 async def transcribe_audio(request: AudioRequest):
+    start_time = time.time()
     try:
         input_bytes = base64.b64decode(request.audio)
         audio_array, sample_rate = extract_audio_from_file(input_bytes)
@@ -127,20 +128,23 @@ async def transcribe_audio(request: AudioRequest):
             audio_array = librosa.resample(audio_array, orig_sr=sample_rate, target_sr=ASR_SAMPLING_RATE)
 
         result = transcribe(audio_array, request.language)
-        return JSONResponse(content={"transcription": result})
+        processing_time = time.time() - start_time
+        return JSONResponse(content={"transcription": result, "processing_time_seconds": processing_time})
     except Exception as e:
         logger.error(f"Error in transcribe_audio: {str(e)}", exc_info=True)
         error_details = {
             "error": str(e),
             "traceback": traceback.format_exc()
         }
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=500,
-            content={"message": "An error occurred during transcription", "details": error_details}
+            content={"message": "An error occurred during transcription", "details": error_details, "processing_time_seconds": processing_time}
         )
 
 @app.post("/synthesize")
 async def synthesize_speech(request: TTSRequest):
+    start_time = time.time()
     logger.info(f"Synthesize request received: text='{request.text}', language='{request.language}', speed={request.speed}")
     try:
         # Extract the ISO code from the full language name
@@ -204,7 +208,8 @@ async def synthesize_speech(request: TTSRequest):
             url = f"https://s3.{S3_REGION}.amazonaws.com/{S3_BUCKET}/{filename}"
             logger.info(f"Public URL generated: {url}")
             
-            return JSONResponse(content={"audio_url": url})
+            processing_time = time.time() - start_time
+            return JSONResponse(content={"audio_url": url, "processing_time_seconds": processing_time})
         
         except NoCredentialsError:
             logger.error("AWS credentials not available or invalid")
@@ -215,9 +220,10 @@ async def synthesize_speech(request: TTSRequest):
 
     except ValueError as ve:
         logger.error(f"ValueError in synthesize_speech: {str(ve)}", exc_info=True)
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=400,
-            content={"message": "Invalid input", "details": str(ve)}
+            content={"message": "Invalid input", "details": str(ve), "processing_time_seconds": processing_time}
         )
     except Exception as e:
         logger.error(f"Unexpected error in synthesize_speech: {str(e)}", exc_info=True)
@@ -226,57 +232,67 @@ async def synthesize_speech(request: TTSRequest):
             "type": type(e).__name__,
             "traceback": traceback.format_exc()
         }
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=500,
-            content={"message": "An unexpected error occurred during speech synthesis", "details": error_details}
+            content={"message": "An unexpected error occurred during speech synthesis", "details": error_details, "processing_time_seconds": processing_time}
         )
     finally:
         logger.info("Synthesize request completed")
 
 @app.post("/identify")
 async def identify_language(request: AudioRequest):
+    start_time = time.time()
     try:
         input_bytes = base64.b64decode(request.audio)
         audio_array, sample_rate = extract_audio_from_file(input_bytes)
         result = identify(audio_array)
-        return JSONResponse(content={"language_identification": result})
+        processing_time = time.time() - start_time
+        return JSONResponse(content={"language_identification": result, "processing_time_seconds": processing_time})
     except Exception as e:
         logger.error(f"Error in identify_language: {str(e)}", exc_info=True)
         error_details = {
             "error": str(e),
             "traceback": traceback.format_exc()
         }
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=500,
-            content={"message": "An error occurred during language identification", "details": error_details}
+            content={"message": "An error occurred during language identification", "details": error_details, "processing_time_seconds": processing_time}
         )
 
 @app.get("/asr_languages")
 async def get_asr_languages():
+    start_time = time.time()
     try:
-        return JSONResponse(content=ASR_LANGUAGES)
+        processing_time = time.time() - start_time
+        return JSONResponse(content={"languages": ASR_LANGUAGES, "processing_time_seconds": processing_time})
     except Exception as e:
         logger.error(f"Error in get_asr_languages: {str(e)}", exc_info=True)
         error_details = {
             "error": str(e),
             "traceback": traceback.format_exc()
         }
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=500,
-            content={"message": "An error occurred while fetching ASR languages", "details": error_details}
+            content={"message": "An error occurred while fetching ASR languages", "details": error_details, "processing_time_seconds": processing_time}
         )
 
 @app.get("/tts_languages")
 async def get_tts_languages():
+    start_time = time.time()
     try:
-        return JSONResponse(content=TTS_LANGUAGES)
+        processing_time = time.time() - start_time
+        return JSONResponse(content={"languages": TTS_LANGUAGES, "processing_time_seconds": processing_time})
     except Exception as e:
         logger.error(f"Error in get_tts_languages: {str(e)}", exc_info=True)
         error_details = {
             "error": str(e),
             "traceback": traceback.format_exc()
         }
+        processing_time = time.time() - start_time
         return JSONResponse(
             status_code=500,
-            content={"message": "An error occurred while fetching TTS languages", "details": error_details}
+            content={"message": "An error occurred while fetching TTS languages", "details": error_details, "processing_time_seconds": processing_time}
         )