Update app.py

app.py

@@ -1,7 +1,7 @@
-# Version: Corrected After Test 3 (V2.3.0 - Adjusted Timestamp Handling)
-# Description: Cette version ajuste la gestion des timestamps.
-# La fin d'un mot est désormais définie par le début du mot suivant - quelques millisecondes.
-# Respecte la structure et les commentaires existants.
+# Version: Corrected After Test 3 (V2.3.0 - Structured, Commented, and ZIP Generation Restored)
+# Description: Cette version structure le script selon l'ordre des étapes du processus.
+# La génération du fichier ZIP a été réintégrée après avoir été omise dans la version précédente.
+# Chaque section est commentée pour assurer une meilleure lisibilité et une logique claire.
 
 import os
 import shutil
@@ -141,6 +141,34 @@ def validate_segments(audio_path, table_data, metadata_state, word_timestamps):
     
     return segment_paths, updated_metadata
 
+# -------------------------------------------------
+# 5. Génération du fichier ZIP
+# -------------------------------------------------
+def generate_zip(metadata_state):
+    if not metadata_state:
+        print("[LOG ERROR] Aucun segment valide trouvé pour la génération du ZIP.")
+        return None
+    
+    zip_path = os.path.join(TEMP_DIR, "dataset.zip")
+    if os.path.exists(zip_path):
+        os.remove(zip_path)
+    
+    metadata_csv_path = os.path.join(TEMP_DIR, "metadata.csv")
+    with open(metadata_csv_path, "w", encoding="utf-8") as f:
+        f.write("audio_file|text|speaker_name|API\n")
+        for seg in metadata_state:
+            f.write(f"{seg['audio_file']}|{seg['text']}|projectname|/API_PHONETIC/\n")
+    
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        zf.write(metadata_csv_path, "metadata.csv")
+        for seg in metadata_state:
+            file_path = os.path.join(TEMP_DIR, seg["audio_file"])
+            if os.path.exists(file_path):
+                zf.write(file_path, seg["audio_file"])
+    
+    print("[LOG] Fichier ZIP généré avec succès.")
+    return zip_path
+
 # -------------------------------------------------
 # 6. Interface utilisateur Gradio
 # -------------------------------------------------
@@ -157,5 +185,11 @@ with gr.Blocks() as demo:
     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])
+    generate_button.click(generate_zip, inputs=metadata_state, outputs=zip_file)
 
 demo.queue().launch()