Update app.py

app.py

@@ -1,10 +1,44 @@
+# import gradio as gr
+# from transformers import pipeline
+# import numpy as np
+
+# transcriber = pipeline("automatic-speech-recognition", model="jonatasgrosman/wav2vec2-large-xlsr-53-arabic")
+
+# def transcribe(stream, new_chunk):
+#     sr, y = new_chunk
+#     y = y.astype(np.float32)
+#     y /= np.max(np.abs(y))
+
+#     if stream is not None:
+#         stream = np.concatenate([stream, y])
+#     else:
+#         stream = y
+#     return stream, transcriber({"sampling_rate": sr, "raw": stream})["text"]
+
+
+# demo = gr.Interface(
+#     transcribe,
+#     ["state", gr.Audio(sources=["microphone"], streaming=True)],
+#     ["state", "text"],
+#     live=True, title="S2T: Transcription automatique en LIVE de l'arabe en text by PS-WADE", 
+#     description="Parlez en arabe dans le microphone. La transcription commence en quelques secondes. Attendez un moment après avoir terminé pour voir le reste de la transcription"
+# )
+
+    
+# demo.launch(share=True, show_error=True)
+
+
 import gradio as gr
 from transformers import pipeline
 import numpy as np
+from googletrans import Translator
 
+# Initialisation du modèle de transcription
 transcriber = pipeline("automatic-speech-recognition", model="jonatasgrosman/wav2vec2-large-xlsr-53-arabic")
+# Initialisation du traducteur
+translator = Translator()
 
-def transcribe(stream, new_chunk):
+def transcribe_and_translate(stream, new_chunk):
     sr, y = new_chunk
     y = y.astype(np.float32)
     y /= np.max(np.abs(y))
@@ -13,16 +47,21 @@ def transcribe(stream, new_chunk):
         stream = np.concatenate([stream, y])
     else:
         stream = y
-    return stream, transcriber({"sampling_rate": sr, "raw": stream})["text"]
-
+    # Transcription du texte
+    transcription = transcriber({"sampling_rate": sr, "raw": stream})["text"]
+    # Traduction du texte transcrit
+    translation = translator.translate(transcription, src='ar', dest='fr').text
+    return stream, transcription, translation
 
+# Création de l'interface Gradio
 demo = gr.Interface(
-    transcribe,
+    transcribe_and_translate,
     ["state", gr.Audio(sources=["microphone"], streaming=True)],
-    ["state", "text"],
-    live=True, title="S2T: Transcription automatique en LIVE de l'arabe en text by PS-WADE", 
-    description="Parlez en arabe dans le microphone. La transcription commence en quelques secondes. Attendez un moment après avoir terminé pour voir le reste de la transcription"
+    ["state", "text", "text"],
+    live=True,
+    title="S2T: Transcription et traduction automatiques en LIVE de l'arabe en texte français by PS-WADE",
+    description="Parlez en arabe dans le microphone. La transcription et la traduction commencent en quelques secondes. Attendez un moment après avoir terminé pour voir le reste de la transcription et la traduction."
 )
 
-    
-demo.launch(share=True, show_error=True)
+# Lancement de l'application Gradio
+demo.launch(share=True, show_error=True)