rick commited on
Commit
c556e09
·
unverified ·
1 Parent(s): b9f059b

add audio isolation feature

Browse files
Files changed (1) hide show
  1. pages/main.py +8 -4
pages/main.py CHANGED
@@ -22,7 +22,7 @@ from copy import deepcopy
22
  # Third-party libraries
23
  import requests
24
  import streamlit as st
25
- import streamlit.components.v1 as components
26
 
27
  #from audiorecorder import audiorecorder
28
  from openai import OpenAI
@@ -54,7 +54,7 @@ from core.files import load_ui_language
54
  from core.core import process_message
55
  from core.core import init_process_mode
56
  from core.moderation import api_moderation_openai_text
57
-
58
 
59
 
60
  def save_attachment(attachment):
@@ -127,8 +127,11 @@ def recorder_released():
127
  if audio_recorded:
128
  audio = AudioSegment.from_wav(io.BytesIO(st.session_state.rec_widget.getvalue()))
129
  st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
130
-
131
- if not (st.session_state.language_detected):
 
 
 
132
  # 1. Verifie si audio.duration_seconds est superieur a 600 secondes (10 minutes)
133
  if audio.duration_seconds > 600:
134
  # PyDub handles time in milliseconds
@@ -478,3 +481,4 @@ main_page()
478
 
479
 
480
 
 
 
22
  # Third-party libraries
23
  import requests
24
  import streamlit as st
25
+ #import streamlit.components.v1 as components
26
 
27
  #from audiorecorder import audiorecorder
28
  from openai import OpenAI
 
54
  from core.core import process_message
55
  from core.core import init_process_mode
56
  from core.moderation import api_moderation_openai_text
57
+ from core.audio_isolation import isolate_audio
58
 
59
 
60
  def save_attachment(attachment):
 
127
  if audio_recorded:
128
  audio = AudioSegment.from_wav(io.BytesIO(st.session_state.rec_widget.getvalue()))
129
  st.write(f"Frame rate: {audio.frame_rate}, Frame width: {audio.frame_width}, Duration: {audio.duration_seconds} seconds")
130
+
131
+ # Isoler l'audio ici
132
+ audio = isolate_audio(audio)
133
+
134
+ if not st.session_state.language_detected:
135
  # 1. Verifie si audio.duration_seconds est superieur a 600 secondes (10 minutes)
136
  if audio.duration_seconds > 600:
137
  # PyDub handles time in milliseconds
 
481
 
482
 
483
 
484
+