From fa743e2e9b3e367b19bd7d349f8e4055757b3361 Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Sat, 11 Feb 2023 13:00:39 +0000
Subject: [PATCH] store generation time per generation rather than per entire
 request

---
 webui.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/webui.py b/webui.py
index 8eaa131..73390b7 100755
--- a/webui.py
+++ b/webui.py
@@ -121,7 +121,7 @@ def generate(
     else:
         texts = split_and_recombine_text(text)
  
-    start_time = time.time()
+    full_start_time = time.time()
  
     outdir = f"./results/{voice}/"
     os.makedirs(outdir, exist_ok=True)
@@ -166,8 +166,10 @@ def generate(
 
         print(f"[{str(line+1)}/{str(len(texts))}] Generating line: {cut_text}")
 
+        start_time = time.time()
         gen, additionals = tts.tts(cut_text, **settings )
         seed = additionals[0]
+        run_time = time.time()-start_time
  
         if isinstance(gen, list):
             for j, g in enumerate(gen):
@@ -175,12 +177,14 @@ def generate(
                 audio_cache[name] = {
                     'audio': g,
                     'text': cut_text,
+                    'time': run_time
                 }
         else:
             name = get_name(line=line)
             audio_cache[name] = {
                 'audio': gen,
                 'text': cut_text,
+                'time': run_time,
             }
 
     for k in audio_cache:
@@ -215,6 +219,7 @@ def generate(
             audio_cache[name] = {
                 'audio': audio,
                 'text': text,
+                'time': time.time()-full_start_time
             }
 
             output_voices.append(f'{outdir}/{voice}_{name}.wav')
@@ -252,7 +257,7 @@ def generate(
         'repetition_penalty': repetition_penalty,
         'cond_free_k': cond_free_k,
         'experimentals': experimental_checkboxes,
-        'time': time.time()-start_time,
+        'time': time.time()-full_start_time,
     }
     
     with open(f'{outdir}/input_{idx}.json', 'w', encoding="utf-8") as f:
@@ -265,6 +270,7 @@ def generate(
     if args.embed_output_metadata:
         for path in audio_cache:
             info['text'] = audio_cache[path]['text']
+            info['time'] = audio_cache[path]['time']
 
             metadata = music_tag.load_file(f"{outdir}/{voice}_{path}.wav")
             metadata['lyrics'] = json.dumps(info)