Update app.py

app.py

@@ -60,15 +60,40 @@ def transcribe_audio(audio_path):
     return raw_transcription, word_timestamps, transcription_with_timestamps
 
 # -------------------------------------------------
-# 3. Gestion du tableau éditable
+# 3. Gestion du tableau éditable (Ajout dynamique)
 # -------------------------------------------------
-def add_row(metadata_state):
-    """Ajoute une ligne vide au tableau et met à jour l'affichage."""
-    new_row = pd.DataFrame([["", None, None, ""]], columns=["Texte", "Début (s)", "Fin (s)", "ID"])
-    updated_metadata = pd.concat([metadata_state, new_row], ignore_index=True)
+def add_row(metadata_state, new_rows):
+    """Ajoute dynamiquement des lignes au tableau en suivant le format structuré."""
+    if new_rows is None:
+        new_rows = []
 
-    print("[LOG] Nouvelle ligne ajoutée au tableau.")
-    return updated_metadata
+    formatted_rows = []
+    
+    # Gestion flexible des entrées (dictionnaires, listes, tuples, DataFrame)
+    if isinstance(new_rows, list):
+        for row in new_rows:
+            if isinstance(row, dict):
+                texte = row.get("Texte", "")
+                debut = row.get("Début (s)", None)
+                fin = row.get("Fin (s)", None)
+            elif isinstance(row, (list, tuple)) and len(row) >= 3:
+                texte, debut, fin = row[:3]
+            else:
+                continue
+            formatted_rows.append([texte, debut, fin, ""])
+
+    elif isinstance(new_rows, pd.DataFrame):
+        for _, row in new_rows.iterrows():
+            formatted_rows.append([row.get("Texte", ""), row.get("Début (s)", None), row.get("Fin (s)", None), ""])
+
+    # Conversion en DataFrame et fusion avec l'état actuel
+    if formatted_rows:
+        new_data = pd.DataFrame(formatted_rows, columns=["Texte", "Début (s)", "Fin (s)", "ID"])
+        metadata_state = pd.concat([metadata_state, new_data], ignore_index=True)
+
+        print(f"[LOG] {len(new_rows)} nouvelles lignes ajoutées.")
+
+    return metadata_state
 
 def save_segments(metadata_table):
     """Sauvegarde les modifications apportées par l'utilisateur."""
@@ -159,24 +184,16 @@ with gr.Blocks() as demo:
     gr.Markdown("# Application de Découpe Audio")
 
     metadata_state = gr.State(init_metadata_state())
-    
+
     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(label="Segments", headers=["Texte", "Début (s)", "Fin (s)", "ID"], datatype=["str", "number", "number", "str"], interactive=True)
 
-    add_row_button = gr.Button("Ajouter une ligne")
+    add_row_button = gr.Button("Ajouter des lignes")
     save_button = gr.Button("Enregistrer")
     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, word_timestamps, transcription_timestamps])
-    add_row_button.click(add_row, inputs=metadata_state, outputs=metadata_state)
+    add_row_button.click(add_row, inputs=[metadata_state, table], outputs=metadata_state)
     save_button.click(save_segments, inputs=table, outputs=metadata_state)
-    validate_button.click(validate_segments, inputs=[audio_input, metadata_state], outputs=metadata_state)
-    generate_button.click(generate_zip, inputs=metadata_state, outputs=zip_file)
 
 demo.queue().launch()