add slider process

new_app.py

@@ -3,6 +3,7 @@ import numpy as np
 import gradio as gr
 import asyncio
 from fastrtc.webrtc import WebRTC
+from fastrtc.utils import AdditionalOutputs
 from pydub import AudioSegment
 import time
 import os 
@@ -12,7 +13,7 @@ from app.logger_config import logger as logging
 from app.utils import (
     generate_coturn_config
 )
-from app.session_utils import (
+from app.new_session_utils import (
     on_load,
     on_unload,
     get_active_sessions,
@@ -57,6 +58,7 @@ def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streami
     try:
         segment = AudioSegment.from_file(filepath_to_stream)
         chunk_duree_ms = 1000
+        total_chunks = len(segment) // chunk_duree_ms + 1
         logging.info(f"[{session_id}] Début du streaming en chunks de {chunk_duree_ms}ms...")
 
         for i, chunk in enumerate(segment[::chunk_duree_ms]):
@@ -71,8 +73,12 @@ def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streami
                 chunk.frame_rate,
                 np.array(chunk.get_array_of_samples()).reshape(1, -1),
             )
+                # Calcul du pourcentage de progression
+            progress = round(((i + 1) / total_chunks) * 100, 2)
+            logging.debug(f"[{session_id}] Progression: {progress}%")
 
-            yield output_chunk
+            # Envoi du chunk et de la progression numérique
+            yield (output_chunk, AdditionalOutputs(progress))
 
             iter_end_time = time.perf_counter()
             processing_duration_ms = (iter_end_time - iter_start_time) * 1000
@@ -109,6 +115,8 @@ def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streami
         if isinstance(stop_streaming_flags, dict):
             stop_streaming_flags["stop"] = False
         logging.info(f"[{session_id}]Signal d'arrêt nettoyé.")
+        yield (None, AdditionalOutputs({"progress": 100}))  
+
 
 
 def stop_streaming(session_id: str, stop_streaming_flags: dict):
@@ -120,6 +128,23 @@ def stop_streaming(session_id: str, stop_streaming_flags: dict):
         stop_streaming_flags["stop"] = True
     return stop_streaming_flags
 
+def handle_additional_outputs(status_slider,progress_value):
+    """Met à jour le slider selon la valeur reçue et gère sa visibilité."""
+    logging.debug(f"📡 Additional output received: {progress_value}")
+
+    try:
+        progress = float(progress_value)
+    except (ValueError, TypeError):
+        progress = 0
+    # status_slider =  gr.update(interactive=True,visible=True,  value=max(0, min(progress, 100)))
+    # return status_slider
+    if progress >= 100:
+        return gr.update(visible=False, value=100)
+    elif progress <= 0:
+        return gr.update(visible=False, value=0)
+    else:
+        return gr.update(visible=True, value=progress)
+
 # --- Interface Gradio ---
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -131,10 +156,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     stop_streaming_flags = gr.State(value={"stop": False})
 
-
-
-
-
     gr.Markdown(
         "## Application 'Streamer' WebRTC (Serveur -> Client)\n"
         "Utilisez l'exemple fourni, uploadez un fichier ou enregistrez depuis votre micro, "
@@ -144,14 +165,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # 1. État pour stocker le chemin du fichier à lire
     active_filepath = gr.State(value=DEFAULT_FILE)
 
-    with gr.Row():
-        with gr.Column():
-            main_audio = gr.Audio(
-                label="Source Audio",
-                sources=["upload", "microphone"], # Combine les deux sources
-                type="filepath",
-                value=DEFAULT_FILE, # Défaut au premier exemple
-            )
+    with gr.Row(equal_height=True):
+        with gr.Column(elem_id="column_source", scale=1):
+            with gr.Group(elem_id="centered_content"):
+                main_audio = gr.Audio(
+                    label="Source Audio",
+                    sources=["upload", "microphone"], # Combine les deux sources
+                    type="filepath",
+                    value=DEFAULT_FILE, # Défaut au premier exemple
+                )
+                status_slider = gr.Slider(0, 100, value=0, label="Progression du streaming", interactive=False, visible=False )
+
         with gr.Column():
             webrtc_stream = WebRTC(
                 label="Stream Audio",
@@ -234,10 +258,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         inputs=[active_filepath, session_hash, stop_streaming_flags],  
         outputs=[webrtc_stream],
         trigger=start_button.click,
-        concurrency_id="audio_stream", # ID de concurrence
+        concurrency_id="audio_stream", 
+        concurrency_limit=10
+    )
+    webrtc_stream.on_additional_outputs(
+        fn=handle_additional_outputs,
+        inputs=[status_slider],
+        outputs=[status_slider],
+        concurrency_id="additional_outputs_audio_stream",
         concurrency_limit=10
     )
-
     # Mettre à jour l'interface au clic sur START
     start_button.click(
         fn=start_streaming_ui,
@@ -266,5 +296,30 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     gr.Timer(3.0).tick(fn=get_active_sessions, outputs=sessions_table)
 
+
+
+# --------------------------------------------------------
+# CSS
+# --------------------------------------------------------
+custom_css = """
+#column_source {
+    display: flex;
+    flex-direction: column;
+    justify-content: center;
+    align-items: center;
+    gap: 1rem;
+    margin-top: auto;
+    margin-bottom: auto;
+}
+#column_source .gr-row {
+    padding-top: 12px;
+    padding-bottom: 12px;
+}
+"""
+demo.css = custom_css
+# --------------------------------------------------------
+# MAIN
+# --------------------------------------------------------
+
 if __name__ == "__main__":
     demo.queue(max_size=10, api_open=False).launch(show_api=False, debug=True)