Update app.py

app.py

@@ -31,22 +31,13 @@ os.makedirs(TEMP_DIR, exist_ok=True)
 
 def init_metadata_state():
     return []
-    
-# Fonction de correction typographique complète avec détection préalable
-def correct_typography(text):
-    corrected_text = text
-    corrected_text = re.sub(r"\s+", " ", corrected_text)  # Suppression des doubles espaces
-    corrected_text = re.sub(r"\b([lLdDmMcCjJnNsStT]) ['’] (\w)", r"\1'\2", corrected_text)  # Correction des apostrophes
-    corrected_text = re.sub(r"(?<=\w) ([?!:;])", r"\1", corrected_text)  # Suppression de l'espace avant !, ?, : et ;
-    corrected_text = re.sub(r"([?!:;])(?=\w)", r"\1 ", corrected_text)  # Ajout d'un espace après !, ?, : et ; si nécessaire
-    corrected_text = re.sub(r"(?<!\d) (\.)", r"\1", corrected_text)  # Suppression de l'espace avant un point
-    corrected_text = re.sub(r"(\.) (?=\w)", r". \2", corrected_text)  # Ajout d'un espace après un point si nécessaire
-    
-    # Appliquer la correction uniquement si le texte a changé
-    return corrected_text.strip() if corrected_text != text else text
+
 # -------------------------------------------------
 # 2. Transcription de l'audio avec Whisper (Timestamps de fin + Marge de Sécurité)
 # -------------------------------------------------
+def correct_typography(text):
+    text = re.sub(r"\b([lLdDmMcCjJnNsStT]) ['’] (\w)", r"\1'\2", text)  # Corrige les espaces autour des apostrophes
+    return text
 
 @spaces.GPU(duration=120)
 def transcribe_audio(audio_path):
@@ -63,6 +54,7 @@ def transcribe_audio(audio_path):
         return "Erreur : Aucun timestamp détecté.", None, [], ""
     
     raw_transcription = " ".join([w["text"] for w in words])
+    raw_transcription = correct_typography(raw_transcription)
     
     MARGIN = 0.06  # 60ms
     word_timestamps = []
@@ -122,11 +114,29 @@ def save_segments(table_data):
 # 4. Génération du fichier ZIP
 # -------------------------------------------------
 def generate_zip(metadata_state, audio_path, zip_name):
+    if isinstance(metadata_state, tuple): 
+        metadata_state = metadata_state[0]
+    
     if metadata_state is None or metadata_state.empty:
+        print("[LOG ERROR] Aucun segment valide trouvé pour la génération du ZIP.")
         return None
     
-    if not zip_name.strip():
-        zip_name = "processed_audio"
+    zip_folder_name = f"{zip_name}_dataset"
+    zip_path = os.path.join(TEMP_DIR, f"{zip_folder_name}.zip")
+    metadata_csv_path = os.path.join(TEMP_DIR, "metadata.csv")
+    
+    metadata_state["ID"] = [f"{zip_name}_seg_{i+1:02d}" for i in range(len(metadata_state))]
+    metadata_state["Texte"] = metadata_state["Texte"].apply(correct_typography)
+    metadata_state["Commentaires"] = ""
+    metadata_state.to_csv(metadata_csv_path, sep="|", index=False)
+    
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        zf.write(metadata_csv_path, "metadata.csv")
+    
+    print("[LOG] Fichier ZIP généré avec succès.")
+    return zip_path
+    if metadata_state is None or metadata_state.empty:
+        return None
     
     zip_folder_name = f"{zip_name}_dataset"
     zip_path = os.path.join(TEMP_DIR, f"{zip_folder_name}.zip")
@@ -139,15 +149,6 @@ def generate_zip(metadata_state, audio_path, zip_name):
     
     with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
         zf.write(metadata_csv_path, "metadata.csv")
-        original_audio = AudioSegment.from_file(audio_path)
-        
-        for i, row in metadata_state.iterrows():
-            start_ms, end_ms = int(row["Début (s)"] * 1000), int(row["Fin (s)"] * 1000)
-            segment_audio = original_audio[start_ms:end_ms]
-            segment_filename = f"{zip_name}_seg_{i+1:02d}.wav"
-            segment_path = os.path.join(TEMP_DIR, segment_filename)
-            segment_audio.export(segment_path, format="wav")
-            zf.write(segment_path, segment_filename)
     
     return zip_path
 
@@ -158,7 +159,7 @@ 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")
-    zip_name = gr.Textbox(label="Nom du fichier ZIP", placeholder="Nom personnalisé", interactive=True)
+    zip_name = gr.Textbox(label="Nom du fichier ZIP", interactive=True)
     raw_transcription = gr.Textbox(label="Transcription", interactive=True)
     transcription_timestamps = gr.Textbox(label="Transcription avec Timestamps", interactive=True)
     table = gr.Dataframe(headers=["Texte", "Début (s)", "Fin (s)"], datatype=["str", "str", "str"], row_count=(1, "dynamic"))
@@ -166,8 +167,8 @@ with gr.Blocks() as demo:
     generate_button = gr.Button("Générer ZIP")
     zip_file = gr.File(label="Télécharger le ZIP")
     
-    audio_input.change(transcribe_audio, inputs=audio_input, outputs=[raw_transcription, transcription_timestamps, audio_input, zip_name])
-    save_button.click(save_segments, inputs=table, outputs=[metadata_state, zip_name])
+    audio_input.change(transcribe_audio, inputs=audio_input, outputs=[raw_transcription, transcription_timestamps, audio_input])
+    save_button.click(save_segments, inputs=table, outputs=[metadata_state])
     generate_button.click(generate_zip, inputs=[metadata_state, audio_input, zip_name], outputs=zip_file)
 
 demo.queue().launch()