Update app.py

app.py

@@ -1,16 +1,9 @@
-# Version: Corrected After Test 4 (V2.4.1 - Fixed Validation & Timestamps Button Restored)
-# Description: Cette version corrige la validation des segments modifiés manuellement.
-# Réintégration du bouton "Générer les timestamps".
-# Correction du bug "too many values to unpack".
-# La génération du fichier ZIP fonctionne correctement après validation.
-
 import os
 import shutil
 import zipfile
 import torch
-import numpy as np
-from pathlib import Path
 import gradio as gr
+from pathlib import Path
 from pydub import AudioSegment
 from transformers import pipeline
 
@@ -41,7 +34,7 @@ def init_metadata_state():
 def transcribe_audio(audio_path):
     if not audio_path:
         print("[LOG] Aucun fichier audio fourni.")
-        return "Aucun fichier audio fourni", [], None, [], ""
+        return "Aucun fichier audio fourni", [], None, []
     
     print(f"[LOG] Début de la transcription de {audio_path}...")
     result = pipe(audio_path, return_timestamps="word")
@@ -49,7 +42,7 @@ def transcribe_audio(audio_path):
     
     if not words:
         print("[LOG ERROR] Erreur : Aucun timestamp détecté.")
-        return "Erreur : Aucun timestamp détecté.", [], None, [], ""
+        return "Erreur : Aucun timestamp détecté.", [], None, []
     
     raw_transcription = " ".join([w["text"] for w in words])
     word_timestamps = [(w["text"], w["timestamp"][0]) for w in words]
@@ -61,39 +54,31 @@ def transcribe_audio(audio_path):
     return raw_transcription, [], audio_path, word_timestamps, transcription_with_timestamps
 
 # -------------------------------------------------
-# 3. Prétraitement des segments : Associer les timestamps aux phrases sélectionnées
+# 3. Enregistrement des segments définis par l'utilisateur
 # -------------------------------------------------
-def preprocess_segments(table_data, word_timestamps):
-    print("[LOG] Début du prétraitement des segments...")
+def save_segments(table_data):
+    print("[LOG] Enregistrement des segments définis par l'utilisateur...")
     formatted_data = []
     for i, row in enumerate(table_data):
-        if not row or len(row) < 1 or not row[0].strip():
-            print(f"[LOG WARNING] Ignoré : ligne vide à l'index {i}.")
+        if len(row) < 3 or not row[0].strip():
+            print(f"[LOG WARNING] Ligne vide ignorée à l'index {i}.")
             continue
 
-        text = row[0].strip()
+        text, start_time, end_time = row[0].strip(), row[1], row[2]
         segment_id = f"seg_{i+1:02d}"
-        start_time = None
-        end_time = None
-        
-        for j, (word, start) in enumerate(word_timestamps):
-            if word in text.split():
-                if start_time is None:
-                    start_time = start
-                end_time = word_timestamps[j+1][1] - 0.01 if j+1 < len(word_timestamps) else start + 0.5
-        
-        formatted_data.append([text, start_time, end_time, segment_id])
-        print(f"[LOG] Segment ajouté : {text} | Début: {start_time}, Fin: {end_time}, ID: {segment_id}")
+
+        formatted_data.append([text, float(start_time), float(end_time), segment_id])
+        print(f"[LOG] Segment enregistré : {text} | Début: {start_time}, Fin: {end_time}, ID: {segment_id}")
     
     return formatted_data
 
 # -------------------------------------------------
 # 4. Validation et découpage des extraits audio
 # -------------------------------------------------
-def validate_segments(audio_path, table_data, metadata_state, word_timestamps):
+def validate_segments(audio_path, table_data, metadata_state):
     print("[LOG] Début de la validation des segments...")
-    if not audio_path or not word_timestamps:
-        print("[LOG ERROR] Erreur : Aucun timestamp valide trouvé !")
+    if not audio_path:
+        print("[LOG ERROR] Erreur : Aucun fichier audio fourni !")
         return [], metadata_state
     
     if os.path.exists(TEMP_DIR):
@@ -104,27 +89,18 @@ def validate_segments(audio_path, table_data, metadata_state, word_timestamps):
     segment_paths = []
     updated_metadata = []
 
-    for row in table_data:
-        if len(row) < 4:
-            print(f"[LOG ERROR] Données incorrectes pour la validation : {row}")
-            continue
-        text, start_time, end_time, segment_id = row
-        
-        if start_time is None or end_time is None:
-            print(f"[LOG ERROR] Timestamp manquant pour : {text}")
-            continue
-        
-        start_ms, end_ms = int(float(start_time) * 1000), int(float(end_time) * 1000)
+    for text, start_time, end_time, segment_id in table_data:
+        start_ms, end_ms = int(start_time * 1000), int(end_time * 1000)
         if start_ms < 0 or end_ms <= start_ms:
             print(f"[LOG ERROR] Problème de découpage : {text} | {start_time}s - {end_time}s")
             continue
 
         segment_filename = f"{Path(audio_path).stem}_{segment_id}.wav"
         segment_path = os.path.join(TEMP_DIR, segment_filename)
-        
+
         extract = original_audio[start_ms:end_ms]
         extract.export(segment_path, format="wav")
-        
+
         segment_paths.append(segment_path)
         updated_metadata.append({
             "audio_file": segment_filename,
@@ -136,6 +112,7 @@ def validate_segments(audio_path, table_data, metadata_state, word_timestamps):
         print(f"[LOG] Extrait généré : {segment_filename}")
     
     return segment_paths, updated_metadata
+
 # -------------------------------------------------
 # 5. Génération du fichier ZIP
 # -------------------------------------------------
@@ -163,6 +140,7 @@ def generate_zip(metadata_state):
     
     print("[LOG] Fichier ZIP généré avec succès.")
     return zip_path
+
 # -------------------------------------------------
 # 6. Interface utilisateur Gradio
 # -------------------------------------------------
@@ -174,16 +152,16 @@ with gr.Blocks() as demo:
     audio_input = gr.Audio(type="filepath", label="Fichier audio")
     raw_transcription = gr.Textbox(label="Transcription", interactive=False)
     transcription_timestamps = gr.Textbox(label="Transcription avec Timestamps", interactive=False)
-    table = gr.Dataframe(headers=["Texte"], datatype=["str"], row_count=(1, "dynamic"), col_count=1)
-    generate_timestamps_button = gr.Button("Générer les timestamps")
+    table = gr.Dataframe(headers=["Texte", "Début (s)", "Fin (s)"], datatype=["str", "number", "number"], row_count=(1, "dynamic"), col_count=3)
+    save_segments_button = gr.Button("Enregistrer les valeurs")
     validate_button = gr.Button("Valider")
     generate_button = gr.Button("Générer ZIP")
     zip_file = gr.File(label="Télécharger le ZIP")
     word_timestamps = gr.State()
 
     audio_input.change(transcribe_audio, inputs=audio_input, outputs=[raw_transcription, table, audio_input, word_timestamps, transcription_timestamps])
-    generate_timestamps_button.click(preprocess_segments, inputs=[table, word_timestamps], outputs=table)
-    validate_button.click(validate_segments, inputs=[audio_input, table, metadata_state, word_timestamps], outputs=[extracted_segments, metadata_state])
+    save_segments_button.click(save_segments, inputs=table, outputs=table)
+    validate_button.click(validate_segments, inputs=[audio_input, table, metadata_state], outputs=[extracted_segments, metadata_state])
     generate_button.click(generate_zip, inputs=metadata_state, outputs=zip_file)
 
 demo.queue().launch()