Spaces:
Paused
Paused
rick
commited on
add audio isolation feature
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
+
|