Update UI elements

app.py

@@ -13,7 +13,6 @@ import os
 from gradio.utils import get_space
 
 from app.utils import (
-    generate_coturn_config,
     raise_function
 )
 from app.session_utils import (
@@ -36,6 +35,7 @@ from app.ui_utils import (
 )
 
 from app.stream_utils import (
+    generate_coturn_config,
     read_and_stream_audio,
     stop_streaming
 )
@@ -100,16 +100,15 @@ with gr.Blocks(theme=theme, css=css_style) as demo:
                             """
                         )
 
-            btn = gr.Button("Proceed to Streaming", visible=True)
-
-            # ui_components_one = [active_filepath, btn]
-            # main_audio.change(fn=on_file_load, inputs=[main_audio], outputs=ui_components_one)
+            btn_proceed_streaming = gr.Button("Proceed to Streaming", visible=False)
+            ui_components_oload_audio = [active_filepath, btn_proceed_streaming]
+            main_audio.change(fn=on_file_load, inputs=[main_audio], outputs=ui_components_oload_audio)
             # main_audio.stop_recording(fn=on_file_load, inputs=[main_audio], outputs=ui_components_one)
             # main_audio.clear(fn=on_file_load, inputs=[main_audio], outputs=ui_components_one)
 
-            btn.click(lambda: gr.Walkthrough(selected=1), outputs=walkthrough)
+            btn_proceed_streaming.click(lambda: gr.Walkthrough(selected=1), outputs=walkthrough)
 
-         # === STEP 2 ===
+        # === STEP 2 ===
         with gr.Step("Stream", id=1) as audio_stream:
             gr.Markdown("### Step 2: Start audio streaming")
             with gr.Group():
@@ -120,18 +119,19 @@ with gr.Blocks(theme=theme, css=css_style) as demo:
                         modality="audio",
                         rtc_configuration=generate_coturn_config(),
                         visible=True,
+                        inputs=main_audio
                     )
-            start_button = gr.Button("Start Streaming")
+            start_stream_button = gr.Button("Start Streaming")
 
             webrtc_stream.stream(
                 fn=read_and_stream_audio,
                 inputs=[active_filepath, session_hash, stop_streaming_flags],
                 outputs=[webrtc_stream],
-                trigger=start_button.click,
+                trigger=start_stream_button.click,
                 concurrency_id="audio_stream",
                 concurrency_limit=10,
             )
-
+            status_message_stream = gr.Markdown("", elem_id="status-message-stream", visible=False)
             go_to_config = gr.Button("Go to Configuration", visible=False)
             go_to_config.click(lambda: gr.Walkthrough(selected=2), outputs=walkthrough)
 
@@ -236,10 +236,10 @@ with gr.Blocks(theme=theme, css=css_style) as demo:
                     )
 
                     start_task_button = gr.Button("Start Task", visible=True)
-                    stop_button = gr.Button("Stop Streaming", visible=False)
+                    stop_stream_button = gr.Button("Stop Streaming", visible=False)
                     stop_task_button = gr.Button("Stop Task", visible=False)
 
-                    stop_button.click(
+                    stop_stream_button.click(
                         fn=stop_streaming,
                         inputs=[session_hash, stop_streaming_flags],
                         outputs=[stop_streaming_flags],
@@ -255,12 +255,13 @@ with gr.Blocks(theme=theme, css=css_style) as demo:
                     )
 
                     ui_components = [
-                        start_button, stop_button,
-                        go_to_config, audio_source_step, status_slider
+                        start_stream_button, stop_stream_button,
+                        go_to_config, audio_source_step, status_slider,walkthrough,status_message_stream
                     ]
 
                     webrtc_stream.on_additional_outputs(
                         fn=handle_additional_outputs,
+                        inputs=[webrtc_stream],
                         outputs=ui_components,
                         concurrency_id="additional_outputs_audio_stream",
                         concurrency_limit=10,
@@ -300,10 +301,10 @@ with gr.Blocks(theme=theme, css=css_style) as demo:
                     # start_task_button.click(
                     #     fn=toggle_task_buttons,
                     #     inputs=None,
-                    #     outputs=[start_task_button, stop_task_button, stop_button],
+                    #     outputs=[start_task_button, stop_task_button, stop_stream_button],
                     #     queue=False
                     # )
 
 
 if __name__ == "__main__":
-    demo.queue(max_size=10, api_open=False).launch(show_api=False, debug=True)
+    demo.queue(max_size=10, api_open=False).launch(show_api=False,show_error=True, debug=True)

app/stream_utils.py

@@ -7,10 +7,56 @@ import os
 import time
 import numpy as np
 
+import hmac
+import hashlib
+import base64
+import os
+import time
+import random
+
+
 # --------------------------------------------------------
 # Utility functions
 # --------------------------------------------------------
 
+
+
+
+def generate_coturn_config():
+    """
+    Génère une configuration Coturn complète avec authentification dynamique (use-auth-secret).
+    Returns:
+        dict: Objet coturn_config prêt à être utilisé côté client WebRTC.
+    """
+
+    secret_key = os.getenv("TURN_SECRET_KEY", "your_secret_key")
+    ttl = int(os.getenv("TURN_TTL", 3600))
+    turn_url = os.getenv("TURN_URL", "turn:*******")
+    turn_s_url = os.getenv("TURN_S_URL", "turns:*****")
+    user = os.getenv("TURN_USER", "client")
+
+    timestamp = int(time.time()) + ttl
+    username = f"{timestamp}:{user}"
+    password = base64.b64encode(
+        hmac.new(secret_key.encode(), username.encode(), hashlib.sha1).digest()
+    ).decode()
+ 
+    coturn_config = {
+        "iceServers": [
+            {
+                "urls": [
+                    f"{turn_url}",
+                    f"{turn_s_url}",
+                ],
+                "username": username,
+                "credential": password,
+            }
+        ]
+    }
+    return coturn_config
+
+
+
 def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streaming_flags: dict):
     """
     Read an audio file and stream it chunk by chunk (1s per chunk).
@@ -28,18 +74,20 @@ def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streami
         segment = AudioSegment.from_file(filepath_to_stream)
         chunk_duration_ms = 1000
         total_chunks = len(segment) // chunk_duration_ms + 1
-        logging.info(f"[{session_id}] Starting audio streaming ({total_chunks} chunks).")
+        logging.info(f"[{session_id}] Starting audio streaming {filepath_to_stream} ({total_chunks} chunks).")
 
         for i, chunk in enumerate(segment[::chunk_duration_ms]):
-            if _is_stop_requested(stop_streaming_flags):
-                logging.info(f"[{session_id}] Stop signal received. Terminating stream.")
-                break
+         
 
             frame_rate = chunk.frame_rate
             samples = np.array(chunk.get_array_of_samples()).reshape(1, -1)
             progress = round(((i + 1) / total_chunks) * 100, 2)
-
-            yield ((frame_rate, samples), AdditionalOutputs(progress))
+            if _is_stop_requested(stop_streaming_flags):
+                logging.info(f"[{session_id}] Stop signal received. Terminating stream.")
+                yield ((frame_rate, samples), AdditionalOutputs({"stoped": True, "value": "STREAM_STOPED"} ) )
+                break
+            
+            yield ((frame_rate, samples), AdditionalOutputs({"progressed": True, "value": progress} ))
             logging.debug(f"[{session_id}] Sent chunk {i+1}/{total_chunks} ({progress}%).")
 
             time.sleep(1)
@@ -53,11 +101,11 @@ def read_and_stream_audio(filepath_to_stream: str, session_id: str, stop_streami
         yield from handle_stream_error(session_id, e, stop_streaming_flags)
     except Exception as e:
         yield from handle_stream_error(session_id, e, stop_streaming_flags)
+        
     finally:
         if isinstance(stop_streaming_flags, dict):
             stop_streaming_flags["stop"] = False
         logging.info(f"[{session_id}] Stop flag reset.")
-        yield (None, AdditionalOutputs("STREAM_DONE"))
 
 
 
@@ -78,8 +126,8 @@ def handle_stream_error(session_id: str, error: Exception | str, stop_streaming_
     if isinstance(stop_streaming_flags, dict):
         stop_streaming_flags["stop"] = False
 
-    yield (None, AdditionalOutputs({"error": True, "message": msg}))
-    yield (None, AdditionalOutputs("STREAM_DONE"))
+    yield ((16000,np.zeros(16000, dtype=np.float32).reshape(1, -1)), AdditionalOutputs({"errored": True, "value": msg}))
+    
 
 
 def _is_stop_requested(stop_streaming_flags: dict) -> bool:
@@ -100,3 +148,7 @@ def stop_streaming(session_id: str, stop_streaming_flags: dict):
     return stop_streaming_flags
 
 
+def raise_function():
+    """Raise an error randomly (1 out of 10 times)."""
+    if random.randint(1, 10) == 1:
+        raise RuntimeError("Random failure triggered!")

app/ui_utils.py

@@ -74,7 +74,6 @@ def to_updates(cfg):
 
 def apply_preset_if_example(filepath, auto_apply):
     """Si fichier = exemple ET auto_apply=True -> applique preset. Sinon, ne rien changer."""
-    logging.info(f"apply_preset_if_example {filepath} {auto_apply} ")
     if not filepath or not auto_apply:
         updates = [gr.update() for _ in range(12)]
         updates.append(gr.update())
@@ -127,48 +126,67 @@ def summarize_config(
     )
     return txt
 
-def handle_additional_outputs( progress_value):
+def handle_additional_outputs(webrtc_stream, msg):
     """
     Update UI elements based on streaming progress or errors.
     Controls button states, audio visibility, and progress slider.
     """
-    logging.debug(f"Additional output received: {progress_value}")
-    # ui_components = [start_button, stop_button,go_to_task, audio_source_step, status_slider]
+    logging.debug(f"Additional output received: {msg}")
+    # ui_components = [start_stream_button, stop_stream_button,go_to_task, audio_source_step, status_slider,walkthrough]
 
+    progress = float(0)
     # Handle structured error message
-    non_ok= (
-            gr.update(visible=True),   # start_button enabled
-            gr.update(visible=False),  # stop_button disabled
-            gr.update(visible=False),  # go_to_task disabled
-            gr.update(interactive=True),       # audio_source_step re-shown
-            gr.update(visible=False, value=0),  # slider hidden
+
+    if isinstance(msg, dict) and msg.get("errored"):
+        value = msg.get("value", "Unknown error.")
+        logging.error(f"[stream_ui] Client-side error: {value}")
+        
+        return (
+                gr.update(visible=True),   # start_stream_button enabled
+                gr.update(visible=False),  # stop_stream_button disabled
+                gr.update(visible=False),  # go_to_task disabled
+                gr.update(interactive=True),       # audio_source_step re-shown
+                gr.update(visible=False, value=0),  # slider hidden
+                gr.update(selected=1),              #walkthrough
+                gr.update(value=f"**Error:** {value}", visible=True)
+            )
+        
+       
+    elif msg.get("progressed") :
+        value = msg.get("value", 0)
+        progress = float(value)
+        if progress == 100.00 :
+            return (
+                gr.update(visible=True),      # start_stream_button disabled
+                gr.update(visible=False),       # stop_stream_button enabled
+                gr.update(visible=True),  # go_to_task enabled
+                gr.update(interactive=True),          # hide audio_source_step
+                gr.update(visible=True, value=progress), # show progress
+                gr.update(),                             #walkthrough
+                gr.update(value="", visible=False) 
+            )
+        else :    
+            return (
+                gr.update(visible=False),      # start_stream_button disabled
+                gr.update(visible=True),       # stop_stream_button enabled
+                gr.update(visible=True),  # go_to_task enabled
+                gr.update(interactive=False),          # hide audio_source_step
+                gr.update(visible=True, value=progress), # show progress
+                gr.update(),                            #walkthrough
+                gr.update(value="", visible=False) 
+            )
+    elif msg.get("stoped") :
+        
+        return (
+            gr.update(visible=True),      # start_stream_button disabled
+            gr.update(visible=False),       # stop_stream_button enabled
+            gr.update(visible=False),  # go_to_task enabled
+            gr.update(interactive=True),          # hide audio_source_step
+            gr.update(visible=True, value=0), # show progress
+            gr.update(selected=1),                       #walkthrough
+            gr.update(value="ℹStream stopped by user.", visible=True)
         )
-    if isinstance(progress_value, dict) and progress_value.get("error"):
-        msg = progress_value.get("message", "Unknown error.")
-        logging.error(f"[stream_ui] Client-side error: {msg}")
-        return non_ok
-
-    try:
-        progress = float(progress_value)
-    except (ValueError, TypeError):
-        progress = 0
-
-    # --- Stream not started ---
-    if progress <= 0:
-        return non_ok
-
-    # --- Stream finished ---
-    if progress >= 100:
-        return non_ok
-
-    # --- Stream in progress ---
-    return (
-        gr.update(visible=False),      # start_button disabled
-        gr.update(visible=True),       # stop_button enabled
-        gr.update(visible=True),  # go_to_task enabled
-        gr.update(interactive=False),          # hide audio_source_step
-        gr.update(visible=True, value=progress), # show progress
-    )
+
 
 
 def on_file_load(filepath):
@@ -200,4 +218,10 @@ def get_custom_theme() :
     css_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "assets", "custom_style.css")
     with open(css_path, encoding="utf-8") as f:
         css_style = f.read()
-    return theme, css_style
+    return theme, css_style
+
+
+
+def raise_error(message="Une erreur est survenue."):
+    raise gr.Error(message)
+

app/utils.py

@@ -1,12 +1,6 @@
 import torch
 from app.logger_config import logger as logging
-import hmac
-import hashlib
-import base64
-import os
-import time
 import random
-
 def debug_current_device():
     """Safely logs GPU or CPU information without crashing on stateless GPU."""
     logging.debug("=== Debugging current device ===")
@@ -55,41 +49,6 @@ def get_current_device():
 
 
 
-def generate_coturn_config():
-    """
-    Génère une configuration Coturn complète avec authentification dynamique (use-auth-secret).
-    Returns:
-        dict: Objet coturn_config prêt à être utilisé côté client WebRTC.
-    """
-
-    secret_key = os.getenv("TURN_SECRET_KEY", "your_secret_key")
-    ttl = int(os.getenv("TURN_TTL", 3600))
-    turn_url = os.getenv("TURN_URL", "turn:*******")
-    turn_s_url = os.getenv("TURN_S_URL", "turns:*****")
-    user = os.getenv("TURN_USER", "client")
-
-    timestamp = int(time.time()) + ttl
-    username = f"{timestamp}:{user}"
-    password = base64.b64encode(
-        hmac.new(secret_key.encode(), username.encode(), hashlib.sha1).digest()
-    ).decode()
- 
-    coturn_config = {
-        "iceServers": [
-            {
-                "urls": [
-                    f"{turn_url}",
-                    f"{turn_s_url}",
-                ],
-                "username": username,
-                "credential": password,
-            }
-        ]
-    }
-    return coturn_config
-
-
-
 
 
 

assets/custom_style.css

@@ -134,4 +134,14 @@ body {
   background: #EFF6FF !important;
   border-radius: 6px;
   box-shadow: 0 0 6px rgba(59,130,246,0.3);
+}
+
+
+#status-message-stream {
+  background-color: #FFF5F5;
+  border: 1px solid #DC2626;
+  border-radius: 6px;
+  padding: 0.75rem;
+  color: #991B1B;
+  font-weight: 500;
 }