Update app.py

app.py

@@ -21,13 +21,14 @@ sentry_sdk.init(
     dsn="https://26932c88424672dea1c43f8536dd6a0d@o4508269800849408.ingest.us.sentry.io/4508269808254976",
 )
 
-def log_user_request(user_input: str, speaking_rate:float, device:str):
+def log_user_request(user_input: str, speaking_rate:float, device:str, execution_time:str):
     """
     Logs the user's request and the current date in Sentry.
     Args:
         user_input (str): User entered text.
         speaking_rate (float): Rate of speech.
         device (str): Available device.
+        execution_time (str): Generation time in seconds
     """
 
     current_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -38,6 +39,7 @@ def log_user_request(user_input: str, speaking_rate:float, device:str):
         scope.set_extra("speaking_rate", str(speaking_rate))
         scope.set_extra("device", device)
         scope.set_extra("timestamp", current_datetime)
+        scope.set_extra("execution_time", f'{execution_time} sec')
         
         sentry_sdk.capture_message("User request")
 
@@ -132,8 +134,13 @@ vocoder, denoiser = load_vocoder(vocoder_path, device)
 
 def gen_tts(text, speaking_rate):
     try:
-        log_user_request(text, speaking_rate, device)
-        return 22050, get_inference(text = text, length_scale = speaking_rate)
+        start_time = datetime.now()
+        output = 22050, get_inference(text = text, length_scale = speaking_rate)
+        end_time = datetime.now()
+        execution_time = str((end_time - start_time).total_seconds())
+        
+        log_user_request(text, speaking_rate, device, execution_time)
+        return output
     except Exception as e:
         sentry_sdk.capture_exception(e)