context and context score added

app.py

@@ -29,6 +29,23 @@ SENTENCES = [
     "Les systèmes vocaux progressent rapidement"
 ]
 
+CONTEXTS = {
+"Train station announcement":
+"Le train 7891 à destination de Paris Nord, prévu à 14h35, partira voie 8.",
+ 
+"Weather forecast":
+"Demain, les températures atteindront 27 degrés avec un ciel dégagé.",
+ 
+"Voice assistant":
+"Je peux vous aider à trouver le restaurant le plus proche.",
+ 
+"Audiobook":
+"Le vieux château se dressait au sommet de la colline depuis plusieurs siècles.",
+ 
+"Customer service":
+"Votre demande a bien été prise en compte et sera traitée sous quarante-huit heures."
+}
+
 # ============================
 # CONFIGURATION
 # ============================
@@ -77,7 +94,9 @@ f5tts = F5TTS()
 
 print("F5-TTS loaded")
 
-def generate_audio(model_name, sentence):
+def generate_audio(model_name, context):
+
+    sentence = CONTEXTS[context]
 
     if model_name == "OmniVoice":
         return generate_omnivoice(sentence)
@@ -146,8 +165,9 @@ def submit_annotation_hf(
         tts_experience, 
         device_type, 
         selected_model,
-        sentence, 
-        score
+        context, 
+        score,
+        context_score
 ):
     annotation = {
         "user_id": user_id,
@@ -158,9 +178,10 @@ def submit_annotation_hf(
         "device_type": device_type,
         
         "model_name": selected_model,
-        "sentence": sentence,
+        "context": context,
 
         "score": score,
+        "context_score": context_score,
         "timestamp": datetime.now(UTC).isoformat()
     }
 
@@ -168,17 +189,17 @@ def submit_annotation_hf(
     short_id = str(uuid.uuid4()) [:8]
 
     file_name = (
-        f"OmniVoice_{user_id}_{timestamp}_{short_id}"
+        f"{selected_model}_{user_id}_{timestamp}_{short_id}"
     )
 
-    local_file = file_name
+    local_file = f"{file_name}.json"
 
     with open(local_file, "w", encoding="utf-8") as f:
         json.dump(annotation, f, indent=2)
 
     api.upload_file(
         path_or_fileobj=local_file,
-        path_in_repo=f"annotations/{file_name}.json",
+        path_in_repo=f"annotations/{local_file}",
         repo_id=HF_DATASET_NAME,
         repo_type="dataset"
     )
@@ -194,10 +215,12 @@ def submit_annotation(
         tts_experience, 
         device_type, 
         selected_model,
-        sentence, 
-        score
+        context, 
+        score,
+        context_score
 ):
-   
+   try:
+
     submit_annotation_hf(
         user_id, 
         age_group, 
@@ -206,11 +229,24 @@ def submit_annotation(
         tts_experience, 
         device_type, 
         selected_model,
-        sentence,
-        score
+        context, 
+        score,
+        context_score
     )
 
-    return f"Annotation saved for {sentence}."
+    return f"Annotation saved for {context}."
+
+   except Exception as e:
+       print("Submit error")
+       print(type(e).__name__)
+       print(e)
+
+       return f"ERROR: {e}"
+
+def update_sentence(context):
+
+        return CONTEXTS[context]
+ 
 
 # ============================
 # SUBMIT BUTTON CONTROL
@@ -272,12 +308,12 @@ with gr.Blocks() as demo:
     with gr.Row():
         age_group = gr.Dropdown(
             choices=[
-                "10-20",
-                "20-30",
-                "30-40",
-                "40-50",
-                "50-60",
-                "60-70",
+                "10-19",
+                "20-29",
+                "30-39",
+                "40-49",
+                "50-59",
+                "60-69",
                 "70+"
             ],
             label="Age Group"
@@ -309,10 +345,15 @@ with gr.Blocks() as demo:
         )
 
     with gr.Row():
-        sentence_selector = gr.Dropdown(
-            choices=SENTENCES,
-            value=SENTENCES[0],
-            label="Sentence to synthesize"
+        context_selector = gr.Dropdown(
+            choices=list(CONTEXTS.keys()),
+            value="Train station announcement",
+            label="Context"
+        )
+
+        sentence_display = gr.Textbox(
+            label = "Sentence",
+            interactive = False
         )
 
         model_selector = gr.Dropdown(
@@ -321,6 +362,12 @@ with gr.Blocks() as demo:
             label = "Model"
         )
 
+        context_selector.change(
+            update_sentence,
+            inputs=context_selector,
+            outputs=sentence_display
+        )
+
         generate_btn = gr.Button("Generate Audio")
 
     audio_player = gr.Audio(label="Generated audio")
@@ -329,13 +376,14 @@ with gr.Blocks() as demo:
         generate_audio,
         inputs=[
             model_selector,
-            sentence_selector
+            context_selector
         ],
         outputs=[audio_player, audio_played]
     )
 
     with gr.Row():
         score = gr.Dropdown(choices=["None","1", "2", "3", "4", "5"], value="None", label="MOS Score (1–5)")
+        score_context = gr.Dropdown(choices=["None","1", "2", "3", "4", "5"], value="None", label="Voice-context adequacy")
         submit_btn = gr.Button("Submit Score", interactive=False)
 
     status = gr.Textbox(label="Status", interactive=False)
@@ -375,7 +423,8 @@ with gr.Blocks() as demo:
             native_language, 
             tts_experience, 
             device_type, 
-            sentence_selector, 
+            model_selector,
+            context_selector, 
             score
         ],
         outputs=status