update space app

app.py

@@ -1,62 +1,72 @@
 import gradio as gr
 from gryannote_audio import AudioLabeling
-from gryannote_pipeline import PipelineSelector
-from gryannote_rttm import RTTM
 from pyannote.audio import Pipeline
 import os
 
-def apply_pipeline(pipeline: Pipeline, audio):
+def apply_pipeline(audio):
     """Apply specified pipeline on the indicated audio file"""
+    pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1", use_auth_token=os.environ["HF_TOKEN"])
     annotations = pipeline(audio)
 
-    return ((audio, annotations), (audio, annotations))
-
-
-def update_annotations(data):
-    return rttm.on_edit(data)
+    return (audio, annotations)
 
 
 with gr.Blocks() as demo:
-    gr.Markdown(
-        "[Gryannote](): The [pyannote](https://github.com/pyannote/pyannote-audio) audio labeling tool"
-    )
+    with gr.Row(equal_height=True):
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown(
+                    '<a href="https://github.com/clement-pages/gryannote"><img src="https://repository-images.githubusercontent.com/744648524/e841cef0-fbd9-45b0-9bce-536a1822c7b1" alt="gryannote logo" width="220"/></a>',
+                )
+        with gr.Column(scale=6):
+                gr.Markdown('<h1 style="font-size: 3em;">gryannote</h1>')  
+                gr.Markdown("<h2>Make the audio labeling process easier and faster! </h2>")
 
-    pipeline_selector = PipelineSelector(token=os.environ["HF_TOKEN"])
-    pipeline_selector.select(
-        fn=pipeline_selector.on_select,
-        inputs=pipeline_selector,
-        outputs=pipeline_selector,
-        preprocess=False,
-        postprocess=False,
-    )
-    pipeline_selector.change(
-        fn=pipeline_selector.on_change,
-        inputs=pipeline_selector,
-        outputs=pipeline_selector,
-        preprocess=False,
-        postprocess=False,
-    )
-    audio_labeling = AudioLabeling(
-        type="filepath",
-        interactive=True,
-    )
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown(
+                "To use the component, start by loading or recording audio."
+                "Then apply the diarization pipeline (here [pyannote/speaker-diarization-3.1](https://huggingface.co/pyannote/speaker-diarization-3.1))" 
+                "or double-click directly on the waveform. The annotations produced can be edited."
+                "You can also use keyboard shortcuts to speed things up!"
+            )
+            gr.Markdown()
+            gr.Markdown()
+            gr.Markdown('<img src="https://github.com/clement-pages/gryannote/blob/main/docs/assets/poster-interspeech.jpg?raw=true" alt="gryannote poster"/>')
+        with gr.Column():
+            audio_labeling = AudioLabeling(
+                type="filepath",
+                interactive=True,
+            )
+            gr.Markdown()
+            gr.Markdown()
+            run_btn = gr.Button("Run pipeline")
 
-    run_btn = gr.Button("Run pipeline")
-
-    rttm = RTTM()
-
-    audio_labeling.edit(
-        fn=update_annotations,
-        inputs=audio_labeling,
-        outputs=rttm,
-        preprocess=False,
-        postprocess=False,
-    )
+            gr.Markdown(
+                """| Shortcut                                      | Action                                                                |
+                    | --------------------------------------------- | --------------------------------------------------------------------- |
+                    | `SPACE`                                       | Toggle play / pause                                                   |
+                    | `ENTER`                                       | Create annotation at current time                                     |
+                    | `SHIFT + ENTER`                               | Split annotation at current time                                      |
+                    | `A`, `B`, `C`, ..., `Z`                       | Set active label. If there is a selected annotation, update its label |
+                    | `LEFT` or `RIGHT`                             | Edit start time of selected annotation (if any) or move time cursor   |
+                    | `SHIFT + LEFT` or `SHIFT + RIGHT`             | Same, but faster                                                      |
+                    |`ALT + LEFT` or `ALT + RIGHT`                  | Edit end time of selected annotation                                  |
+                    | `SHIFT + ALT + LEFT` or `SHIFT + ALT + RIGHT` | Same, but faster                                                      |
+                    | `TAB`                                         | Select next annotation                                                |
+                    | `SHIFT + TAB`                                 | Select previous annotation                                            |
+                    |`BACKSPACE`                                    | Delete selected annotation and select the previous one                |
+                    |`DELETE` or `SHIFT + BACKSPACE`                | Delete selected region and select the next one                        |
+                    |`ESC`                                          | Unselect selected annotation and / or label                           |
+                    | `UP` or `DOWN`                                | Zoom in/out                                                           |
+                    | `F2`                                          | Open settings for the active label                                    |
+                    """
+            )
 
     run_btn.click(
         fn=apply_pipeline,
-        inputs=[pipeline_selector, audio_labeling],
-        outputs=[audio_labeling, rttm],
+        inputs=audio_labeling,
+        outputs=audio_labeling,
     )