ameliorations code & texte

app.py

@@ -11,21 +11,50 @@ login(token=auth_token)
 
 # list all files in the ./audios directory for the dropdown
 AUDIO_FILES = [f for f in os.listdir('./exples_voix') if os.path.isfile(os.path.join('./exples_voix', f))]
-MODELES_TTS = ["ArissBandoss/coqui-tts-moore-V1", "ArissBandoss/mms-tts-mos-male-17-V5"]
-MODELES_ASR = ["ArissBandoss/whisper-small-mos", "openai/whisper-large-v3-turbo"]
+MODELES_TTS = ["ArissBandoss/coqui-tts-moore-V1", "ArissBandoss/mms-tts-mos-V2"]
+MODELES_ASR = ["ArissBandoss/whisper-small-mos", 
+               #"openai/whisper-large-v3-turbo"
+               ]
 LANGUAGES   = ["Automatic Detection"]
 
-DESCRIPTION = """<div style="display: flex; justify-content: space-between; align-items: center; flex-wrap: wrap;">
-                    <div style="flex: 1; min-width: 250px;">
-                        Ce modèle de traduction vers la <b>langue Mooré</b> a été développé from scratch par <b>GO AI CORP</b> et la version disponible en test est celle à 700 millions de paramètres.
-                        <br><br>
-                        Pour les détails techniques sur l'architecture du modèle, prendre attache avec nous via WhatsApp au <b>+226 66 62 83 03</b>.
+DESCRIPTION_TTT = """<div style="display: flex; justify-content: space-between; align-items: center; flex-wrap: wrap;">
+                        <div style="flex: 1; min-width: 250px;">
+                            Cette 1ère interface sert à démontrer les performances du modèle d'IA de traduction de la <b>langue Mooré</b> vers la langue <b>Francaise</b> et vice-versa.
+                            Ce modèle a été concu, développé et maintenu par <b>GO AI CORP</b> et la version disponible en test est celle à 700 millions de paramètres.
+                            <br><br>
+                            Pour plus de détails sur ce projet ou pour toute collaboration, veuillez prendre attache avec nous via WhatsApp au <b>+226 66 62 83 03</b>.
+                        </div>
+                        <div style="flex-shrink: 0; min-width: 150px; text-align: center;">
+                            <img src="https://github.com/ANYANTUDRE/Stage-IA-Selever-GO-AI-Corp/blob/main/img/goaicorp-logo2.jpg?raw=true" width="300px" style="max-width: 100%; height: auto;">
+                        </div>
                     </div>
-                    <div style="flex-shrink: 0; min-width: 150px; text-align: center;">
-                        <img src="https://github.com/ANYANTUDRE/Stage-IA-Selever-GO-AI-Corp/blob/main/img/goaicorp-logo2.jpg?raw=true" width="300px" style="max-width: 100%; height: auto;">
+                """
+
+DESCRIPTION_TTS = """<div style="display: flex; justify-content: space-between; align-items: center; flex-wrap: wrap;">
+                        <div style="flex: 1; min-width: 250px;">
+                            Cette 2ème interface sert à démontrer les performances des modèles de <b>synthèse vocale</b> du <b>Mooré</b> qui ont été concus, développés et maintenus par <b>GO AI CORP</b>.  
+                            Pour l'instant, deux modèles sont mis à disposition pour tests.
+                            <br><br>
+                            Pour plus de détails sur ce projet ou pour toute collaboration, veuillez prendre attache avec nous via WhatsApp au <b>+226 66 62 83 03</b>.
+                        </div>
+                        <div style="flex-shrink: 0; min-width: 150px; text-align: center;">
+                            <img src="https://github.com/ANYANTUDRE/Stage-IA-Selever-GO-AI-Corp/blob/main/img/goaicorp-logo2.jpg?raw=true" width="300px" style="max-width: 100%; height: auto;">
+                        </div>
+                    </div>
+                """
+
+DESCRIPTION_STT = """<div style="display: flex; justify-content: space-between; align-items: center; flex-wrap: wrap;">
+                        <div style="flex: 1; min-width: 250px;">
+                            Cette 3ème interface sert à démontrer les performances des modèles de <b>transcription</b> du <b>Mooré</b> qui ont été concus, développés et maintenus par <b>GO AI CORP</b>.  
+                            Pour l'instant, deux modèles sont mis à disposition pour tests.
+                            <br><br>
+                            Pour plus de détails sur ce projet ou pour toute collaboration, veuillez prendre attache avec nous via WhatsApp au <b>+226 66 62 83 03</b>.
+                        </div>
+                        <div style="flex-shrink: 0; min-width: 150px; text-align: center;">
+                            <img src="https://github.com/ANYANTUDRE/Stage-IA-Selever-GO-AI-Corp/blob/main/img/goaicorp-logo2.jpg?raw=true" width="300px" style="max-width: 100%; height: auto;">
+                        </div>
                     </div>
-                </div>
-            """
+                """
 
 
 demo = gr.Blocks(theme=gr.themes.Soft())
@@ -33,50 +62,19 @@ demo = gr.Blocks(theme=gr.themes.Soft())
 goai_traduction_if = gr.Interface(
     fn=goai_traduction.goai_traduction,
     inputs=[
-        gr.Textbox(label="Texte", placeholder="Yaa sõama"),
+        gr.Textbox(label="Texte à traduire", placeholder="La, pagã sɩd talla raadã n tɩ lebs kãangã pʋgẽ, n na n paam a menga. Rẽ wã yĩnga, sẽn kisa, sɩd kisame. Sẽn ka maande, sɩd ka maand ye."),
         gr.Dropdown(label="Langue source", choices=["fra_Latn", "mos_Latn"], value='fra_Latn'),
         gr.Dropdown(label="Langue cible", choices=["fra_Latn", "mos_Latn"], value='mos_Latn')
     ],
-    outputs=["text"],
+    outputs=["Texte traduit"],
     examples=[["Yʋʋm a wãn la b kẽesd biig lekolle?", "mos_Latn", "fra_Latn"],
               ["Zak-soab la kasma.", "mos_Latn", "fra_Latn"],
               ["Le gouvernement avait pris des mesures louables par rapport à l’augmentation des prix de certaines denrées alimentaires.", "fra_Latn", "mos_Latn"],
               ["Comme lors du match face à la Côte d’Ivoire, c’est sur un coup de pied arrêté que les Etalons encaissent leur but.", "fra_Latn", "mos_Latn"],
     ],
     cache_examples=False,
-    title="Traduction Mooré-Francais",
-    description=DESCRIPTION
-)
-
-goai_stt_if = gr.Interface(
-    fn=goai_stt2.transcribe,
-    inputs=[
-        gr.Audio(sources=["microphone", "upload"], type="filepath"),
-        gr.Dropdown(
-            choices=MODELES_ASR,
-            value="ArissBandoss/whisper-small-mos", 
-            label="Model Name"
-        ),
-        gr.Dropdown(
-            choices=LANGUAGES, 
-            value="Automatic Detection",  # + sorted(get_language_names())
-            label="Language", 
-            interactive = True,
-        ), 
-        gr.Slider(label="Batch Size", minimum=1, maximum=32, value=8, step=1),
-        gr.Slider(label="Chunk Length (s)", minimum=1, maximum=60, value=17.5, step=0.1),
-        gr.Slider(label="Stride Length (s)", minimum=1, maximum=30, value=1, step=0.1),
-    ],
-    outputs=[gr.Textbox(label="Output"), gr.File(label="Download Files")],
-    examples=[["./audios/example1.mp3", "a ye ligdi"], 
-              ["./audios/example2.mp3", "zoe nimbãanega"],
-              ["./audios/example3.mp3", "zãng-zãnga"],
-              ["./audios/example4.mp3", "yõk foto"]
-             ],
-    cache_examples=False,
-    title="Mooré ASR",
-    description=DESCRIPTION,
-    flagging_mode="auto",
+    title="Traduction Mooré-Francais, Francais-Mooré",
+    description=DESCRIPTION_TTT
 )
 
 goai_ttt_tts_pipeline_if = gr.Interface(
@@ -107,14 +105,14 @@ goai_ttt_tts_pipeline_if = gr.Interface(
         gr.Text(label="Texte traduit"),
         gr.Audio(label="Audio généré", format="wav"),
     ],
-    examples=[["Ils vont bien, merci. Mon père travaille dur dans les champs et ma mère est toujours occupée à la maison.", "exple_voix_masculine.wav", None], 
-              ["La finale s’est jouée en présence du Président du Faso, Ibrahim Traoré.", "exple_voix_feminine.wav", None],
-              ["Les enfants apprennent les danses traditionnelles de leurs ancêtres, jouent à des jeux traditionnels dans les rues et aident leurs parents dans les tâches quotidiennes.", "exple_voix_masculine.wav", None],
-              ["Ils achetèrent des troupeaux, firent construire des cases, parcoururent tout le pays pour offrir à leur mère et à leurs femmes les plus beaux bijoux, les plus belles étoffes.", "exple_voix_feminine.wav", None]
+    examples=[["Ils vont bien, merci. Mon père travaille dur dans les champs et ma mère est toujours occupée à la maison.", "exple_voix_masculine.wav", "ArissBandoss/coqui-tts-moore-V1"], 
+              ["La finale s’est jouée en présence du Président du Faso, Ibrahim Traoré.", "exple_voix_feminine.wav", "ArissBandoss/coqui-tts-moore-V1"],
+              ["Les enfants apprennent les danses traditionnelles de leurs ancêtres, jouent à des jeux traditionnels dans les rues et aident leurs parents dans les tâches quotidiennes.", "exple_voix_masculine.wav", "ArissBandoss/coqui-tts-moore-V1"],
+              ["Ils achetèrent des troupeaux, firent construire des cases, parcoururent tout le pays pour offrir à leur mère et à leurs femmes les plus beaux bijoux, les plus belles étoffes.", "exple_voix_feminine.wav", "ArissBandoss/coqui-tts-moore-V1"]
              ],
     cache_examples=False,
     title="Mooré TTS & Traduction",
-    description=DESCRIPTION,
+    description=DESCRIPTION_TTS,
 )
 
 
@@ -141,21 +139,22 @@ goai_stt_ttt_pipeline_if = gr.Interface(
         gr.Textbox(label="Texte Mooré"), 
         gr.Textbox(label="Texte Francais"), 
     ],
-    examples=[["./audios/example1.mp3", "a ye ligdi"], 
-              ["./audios/example2.mp3", "zoe nimbãanega"],
-              ["./audios/example3.mp3", "zãng-zãnga"],
-              ["./audios/example4.mp3", "yõk foto"]
+    examples=[["./audios/example1.mp3", "ArissBandoss/whisper-small-mos"], 
+              ["./audios/example2.mp3", "ArissBandoss/whisper-small-mos"],
+              ["./audios/example3.mp3", "ArissBandoss/whisper-small-mos"],
+              ["./audios/example4.mp3", "ArissBandoss/whisper-small-mos"]
              ],
     cache_examples=False,
     title="Mooré ASR & Traduction",
-    description=DESCRIPTION,
+    description=DESCRIPTION_STT,
     flagging_mode="auto",
 )
 
+
 with demo:
     gr.TabbedInterface(
-        interface_list=[goai_traduction_if, goai_stt_if, goai_ttt_tts_pipeline_if, goai_stt_ttt_pipeline_if],
-        tab_names=["Traduction Mooré-Francais", "Mooré ASR", "Mooré TTS & Traduction", "Mooré ASR & Traduction"]
+        interface_list=[goai_traduction_if, goai_ttt_tts_pipeline_if, goai_stt_ttt_pipeline_if],
+        tab_names=["Traduction Mooré-Francais", "Mooré TTS & Traduction", "Mooré ASR & Traduction"]
     )
 
 demo.queue().launch(ssr_mode=False)

goai_helpers/goai_tts.py

@@ -32,7 +32,7 @@ def goai_tts(texte):
     start_time = time.time()
 
     # Charger le modèle TTS avec le token d'authentification
-    model_id = "ArissBandoss/mms-tts-mos-male-17-V5"
+    model_id = "ArissBandoss/mms-tts-mos-V2"
     synthesiser = pipeline("text-to-speech", model_id, device=device)
 
     # Inférence

goai_helpers/goai_ttt_tts_pipeline.py

@@ -21,7 +21,7 @@ def goai_many_tts(
         reference_audio=None,
     ):
 
-    if tts_model == "ArissBandoss/coqui-tts-moore-V1":
+    if "coqui" in tts_model:
         # TTS pipeline
         tts = MooreTTS(tts_model)
         reference_speaker = os.path.join("./exples_voix", reference_speaker)
@@ -34,7 +34,7 @@ def goai_many_tts(
 
         return sampling_rate, audio_array.numpy()
     
-    elif tts_model == "ArissBandoss/mms-tts-mos-male-17-V5":
+    elif "mms" in tts_model:
         sample_rate, audio_data = goai_tts(text)
         return sample_rate, audio_data