Added counter with the number of recordings left

app.py

@@ -44,14 +44,14 @@ def get_unique_name():
     return ''.join([random.choice(string.ascii_letters
             + string.digits) for n in range(32)])
 
-def get_prompts(domain, size, language_code):
+def get_prompts(domain, size, language_code,prompts_left_info):
     print(f"Retrieving prompts for domain {domain} with method: {type} for language_code {language_code} of size {size}")
     size = int(size)
     promptset = dict_promptset[domain][0:size]
-
-    return(promptset, promptset[0])
+    prompts_left_info = size
+    return(promptset, promptset[0],prompts_left_info)
     
-def save_recording_and_meta(project_name, recording, prompt_text, language_code, spk_name, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number):
+def save_recording_and_meta(project_name, recording, prompt_text, language_code, spk_name, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number, prompts_left_info):
     #, name, age, gender):
     # TODO save user data in the next version
 
@@ -141,7 +141,9 @@ def save_recording_and_meta(project_name, recording, prompt_text, language_code,
         # get next prompt
         prompt = promptset[prompt_number]
 
-    return [prompt, prompt_number, None]
+    prompts_left_info = prompts_left_info - 1
+
+    return [prompt, prompt_number, None, prompts_left_info]
 
 
 def whisper_model_change(radio_whisper_model):
@@ -236,6 +238,7 @@ with block:
     spk_city = gr.State("unknown")
     spk_gender = gr.State("unknown")
     spk_nativity = gr.State("unknown")
+    spk_name = gr.State("unknown")
     cities = sorted(dict_origin["Poland"]["cities"])
     
 
@@ -302,6 +305,10 @@ with block:
         whisper_model = init_whisper_model(whisper_model_type)
         return [whisper_model_type, whisper_model]
 
+    def change_prompts_left(prompts_left, current_prompt, promptset_size):
+        prompts_left = promptset_size - current_prompt
+        return [prompts_left]
+    
     gr.Markdown(markdown)
 
     with gr.Tabs():
@@ -380,19 +387,21 @@ with block:
             with gr.Accordion(label="Promptset settings"):
                 radio_prompts_domain = gr.Dropdown(["bridge"], label="Select promptset domain", info="")
                 #radio_promptset_type = gr.Radio(["New promptset generation", "Existing promptset use"], label="Language", value ="Existing promptset use", info="New promptset is generated using. Requires providing open AI key in general settings tab")
-                var_promptset_size = gr.Textbox(label="Specify number of prompts (min 10, max 200)") 
-                button_get_prompts = gr.Button("Save settings and get first prompt to record")
+                var_promptset_size = gr.Textbox(label="How many recordings you want to make? (max 200)") 
+                button_get_prompts = gr.Button("Save settings and get a first prompt")
 
-            prompt_text = gr.Textbox(placeholder='Prompt to be recorded',label="Prompt to be read during recording")
-            speech_recording = gr.Audio(source="microphone",label="Select 'record from microphone' and read prompt displayed above", type="filepath")
+            prompts_left_info = gr.Number(placeholder='',label="Recordings left",lines=1, max_lines=1, show_label=True, interactive=False)
+            prompt_text = gr.Textbox(placeholder='Prompt to read during recording',label="Prompt to be read during recording")
+            speech_recording = gr.Audio(source="microphone",label="Select 'record from microphone' and read the prompt displayed above", type="filepath")
             
             radio_prompts_domain.change(fn=change_domain, inputs=radio_prompts_domain, outputs=domain)
             #radio_promptset_type.change(fn=change_prompts_type, inputs=radio_promptset_type, outputs=prompts_type)
+            #prompts_left.change(change_prompts_left, inputs = [prompts_left, current_prompt, var_promptset_size], outputs = [prompts_left])
 
-            button_save_and_next = gr.Button("Save audio recording and move to the next prompt")
+            button_save_and_next = gr.Button("Save recording and get the next prompt")
             # TODO - add option to generate new promptset on the fly for new domains
-            button_get_prompts.click(get_prompts, inputs=[radio_prompts_domain, var_promptset_size, language_code], outputs = [promptset, prompt_text])
+            button_get_prompts.click(get_prompts, inputs=[radio_prompts_domain, var_promptset_size, language_code, prompts_left_info], outputs = [promptset, prompt_text, prompts_left_info])
             
-            button_save_and_next.click(save_recording_and_meta, inputs=[project_name, speech_recording, prompt_text, language_code, spk_name, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number], outputs=[prompt_text, prompt_number, speech_recording])
+            button_save_and_next.click(save_recording_and_meta, inputs=[project_name, speech_recording, prompt_text, language_code, spk_name, spk_age, spk_accent, spk_city, spk_gender, spk_nativity, promptset, prompt_number, prompts_left_info], outputs=[prompt_text, prompt_number, speech_recording,prompts_left_info])
                 
 block.launch()