fix QT issue

README.md

@@ -1,6 +1,6 @@
 ---
 title: Audio Separation
-emoji: 🔥
+emoji: 🔊
 colorFrom: pink
 colorTo: purple
 sdk: gradio

src/gyraudio/audio_separation/visualization/interactive_audio.py

@@ -22,8 +22,8 @@ import logging
 from interactive_pipe.data_objects.curves import Curve, SingleCurve
 from interactive_pipe import interactive, KeyboardControl, Control
 from interactive_pipe.headless.pipeline import HeadlessPipeline
-from interactive_pipe.graphical.qt_gui import InteractivePipeQT
-from interactive_pipe.graphical.mpl_gui import InteractivePipeMatplotlib
+from interactive_pipe.graphical.gradio_gui import InteractivePipeGradio
+
 from gyraudio.audio_separation.visualization.audio_player import audio_selector, audio_trim, audio_player
 
 default_device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -32,9 +32,9 @@ LEARNT_SAMPLING_RATE = 8000
 
 @interactive(
     idx=KeyboardControl(value_default=0, value_range=[
-                        0, 1000], modulo=True, keyup="8", keydown="2"),
+                        0, 1000], modulo=True, keyup="8", keydown="2", name="clean signal index"),
     idn=KeyboardControl(value_default=0, value_range=[
-                        0, 1000], modulo=True, keyup="9", keydown="3")
+                        0, 1000], modulo=True, keyup="9", keydown="3", name="noisy signal index")
 )
 def signal_selector(signals, idx=0, idn=0, global_params={}):
     if isinstance(signals, dict):
@@ -230,10 +230,15 @@ def interactive_audio_separation_visualization(
 ):
     pip = HeadlessPipeline.from_function(
         interactive_audio_separation_processing, cache=False)
-    if gui == "qt":
+    if gui == "gradio":
+        app = InteractivePipeGradio(
+            pipeline=pip, name="audio separation", audio=True)
+    elif gui == "qt":
+        from interactive_pipe.graphical.qt_gui import InteractivePipeQT
         app = InteractivePipeQT(
             pipeline=pip, name="audio separation", size=(1000, 1000), audio=True)
     else:
+        from interactive_pipe.graphical.mpl_gui import InteractivePipeMatplotlib
         logging.warning("No support for audio player with Matplotlib")
         app = InteractivePipeMatplotlib(
             pipeline=pip, name="audio separation", size=None, audio=False)