Update app.py

app.py

@@ -472,6 +472,17 @@ def check_btn(btn):
 def transcribe():
     return 'Success'
 
+def transcribe_audio(audio_path):
+    # Prétraitement de l'audio
+    audio = preprocess_audio(audio_path)
+    
+    # Chargement du modèle
+    model = load_model()
+    
+    # Transcription
+    transcription = model.transcribe(audio)
+    
+    return transcription
 
 
 demo = gr.Blocks(css=".markdown-body { font-size: 18px; }")
@@ -526,26 +537,18 @@ with demo:
     # When an audio file is selected, it will display the file path
     audio_file_dropdown.change(fn=load_audio_file, inputs=[audio_file_dropdown], outputs=[audio_output])
 
-    
-    
-    ########################## Key Gen Part ##########################
+    with gr.Row():
+    transcribe_btn = gr.Button("Transcrire l'audio")
+    transcription_output = gr.Textbox(label="Transcription", lines=5)
 
-    gr.Markdown(
-        "## Step 1.2: Generate the keys\n\n"
-        """In Fully Homomorphic Encryption (FHE) methods, two types of keys are created. The first 
-        type, called secret keys, are used to encrypt and decrypt the user's data. The second type, 
-        called evaluation keys, enables a server to work on the encrypted data without seeing the 
-        actual data.
-        """
+    transcribe_btn.click(
+        fn=transcribe_audio,
+        inputs=[audio_output],
+        outputs=[transcription_output]
     )
+    
+    
 
-    gen_key_btn = gr.Button("Generate the secret and evaluation keys")
-
-    gen_key_btn.click(
-        key_gen_fn,
-        inputs=[],
-        outputs=[gen_key_btn],
-    )
 
     ########################## Step 1.1: Record Audio ##########################
 
@@ -580,6 +583,36 @@ with demo:
 
     clear.click(lambda: None, None, msg, queue=False)
 
+    ########################## Transcription ##########################
+    with gr.Row():
+    transcribe_btn = gr.Button("Transcrire l'audio")
+    transcription_output = gr.Textbox(label="Transcription", lines=5)
+
+transcribe_btn.click(
+    fn=transcribe_audio,
+    inputs=[audio_output],
+    outputs=[transcription_output]
+)
+
+    ########################## Key Gen Part ##########################
+
+    gr.Markdown(
+        "## Step 1.2: Generate the keys\n\n"
+        """In Fully Homomorphic Encryption (FHE) methods, two types of keys are created. The first 
+        type, called secret keys, are used to encrypt and decrypt the user's data. The second type, 
+        called evaluation keys, enables a server to work on the encrypted data without seeing the 
+        actual data.
+        """
+    )
+
+    gen_key_btn = gr.Button("Generate the secret and evaluation keys")
+
+    gen_key_btn.click(
+        key_gen_fn,
+        inputs=[],
+        outputs=[gen_key_btn],
+    )
+
     ########################## Main document Part ##########################
 
     gr.Markdown("<hr />")