balthou commited on
Commit
55ca18f
Β·
1 Parent(s): 4e8fc18

fix QT issue

Browse files
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  title: Audio Separation
3
- emoji: πŸ”₯
4
  colorFrom: pink
5
  colorTo: purple
6
  sdk: gradio
 
1
  ---
2
  title: Audio Separation
3
+ emoji: πŸ”Š
4
  colorFrom: pink
5
  colorTo: purple
6
  sdk: gradio
src/gyraudio/audio_separation/visualization/interactive_audio.py CHANGED
@@ -22,8 +22,8 @@ import logging
22
  from interactive_pipe.data_objects.curves import Curve, SingleCurve
23
  from interactive_pipe import interactive, KeyboardControl, Control
24
  from interactive_pipe.headless.pipeline import HeadlessPipeline
25
- from interactive_pipe.graphical.qt_gui import InteractivePipeQT
26
- from interactive_pipe.graphical.mpl_gui import InteractivePipeMatplotlib
27
  from gyraudio.audio_separation.visualization.audio_player import audio_selector, audio_trim, audio_player
28
 
29
  default_device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -32,9 +32,9 @@ LEARNT_SAMPLING_RATE = 8000
32
 
33
  @interactive(
34
  idx=KeyboardControl(value_default=0, value_range=[
35
- 0, 1000], modulo=True, keyup="8", keydown="2"),
36
  idn=KeyboardControl(value_default=0, value_range=[
37
- 0, 1000], modulo=True, keyup="9", keydown="3")
38
  )
39
  def signal_selector(signals, idx=0, idn=0, global_params={}):
40
  if isinstance(signals, dict):
@@ -230,10 +230,15 @@ def interactive_audio_separation_visualization(
230
  ):
231
  pip = HeadlessPipeline.from_function(
232
  interactive_audio_separation_processing, cache=False)
233
- if gui == "qt":
 
 
 
 
234
  app = InteractivePipeQT(
235
  pipeline=pip, name="audio separation", size=(1000, 1000), audio=True)
236
  else:
 
237
  logging.warning("No support for audio player with Matplotlib")
238
  app = InteractivePipeMatplotlib(
239
  pipeline=pip, name="audio separation", size=None, audio=False)
 
22
  from interactive_pipe.data_objects.curves import Curve, SingleCurve
23
  from interactive_pipe import interactive, KeyboardControl, Control
24
  from interactive_pipe.headless.pipeline import HeadlessPipeline
25
+ from interactive_pipe.graphical.gradio_gui import InteractivePipeGradio
26
+
27
  from gyraudio.audio_separation.visualization.audio_player import audio_selector, audio_trim, audio_player
28
 
29
  default_device = "cuda" if torch.cuda.is_available() else "cpu"
 
32
 
33
  @interactive(
34
  idx=KeyboardControl(value_default=0, value_range=[
35
+ 0, 1000], modulo=True, keyup="8", keydown="2", name="clean signal index"),
36
  idn=KeyboardControl(value_default=0, value_range=[
37
+ 0, 1000], modulo=True, keyup="9", keydown="3", name="noisy signal index")
38
  )
39
  def signal_selector(signals, idx=0, idn=0, global_params={}):
40
  if isinstance(signals, dict):
 
230
  ):
231
  pip = HeadlessPipeline.from_function(
232
  interactive_audio_separation_processing, cache=False)
233
+ if gui == "gradio":
234
+ app = InteractivePipeGradio(
235
+ pipeline=pip, name="audio separation", audio=True)
236
+ elif gui == "qt":
237
+ from interactive_pipe.graphical.qt_gui import InteractivePipeQT
238
  app = InteractivePipeQT(
239
  pipeline=pip, name="audio separation", size=(1000, 1000), audio=True)
240
  else:
241
+ from interactive_pipe.graphical.mpl_gui import InteractivePipeMatplotlib
242
  logging.warning("No support for audio player with Matplotlib")
243
  app = InteractivePipeMatplotlib(
244
  pipeline=pip, name="audio separation", size=None, audio=False)