Duplicate from Subhraj07/audio_vocab_spleeter

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Audio Vocab Spleeter
+emoji: 📊
+colorFrom: red
+colorTo: pink
+sdk: streamlit
+sdk_version: 1.21.0
+app_file: app.py
+pinned: false
+license: openrail
+duplicated_from: Subhraj07/audio_vocab_spleeter
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,91 @@
+import streamlit as st
+import tempfile
+import shutil
+import os
+import random
+import string
+import subprocess
+
+# os.system("conda install -c conda-forge ffmpeg libsndfile -y")
+
+st.set_page_config(
+    page_title="Video Splitter",
+    page_icon=None,
+    layout="wide",
+    initial_sidebar_state="auto",
+)
+
+def run_shell_command(command):
+    process = subprocess.Popen(
+        command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, shell=True
+    )
+
+    while True:
+        output = process.stdout.readline()
+        if output == '' and process.poll() is not None:
+            break
+        if output:
+            st.write(output.strip())
+
+    rc = process.poll()
+    return rc
+
+def create_download_link(video_file):
+    with open(video_file.name, "rb") as f:
+        file_bytes = f.read()
+        b64 = base64.b64encode(file_bytes).decode()
+        download_link = f'<a href="data:video/mp4;base64,{b64}" download="{video_file.name}">Download video file</a>'
+        st.markdown(download_link, unsafe_allow_html=True)
+
+def save_uploaded_file(uploaded_file):
+    save_dir = "/tmp/uploaded_videos"  # Change this to the desired folder
+    if not os.path.exists(save_dir):
+        os.makedirs(save_dir)
+
+    with open(os.path.join(save_dir, uploaded_file.name), "wb") as f:
+        f.write(uploaded_file.getbuffer())
+
+    st.success(f"File '{uploaded_file.name}' saved to folder '{save_dir}'")
+    return uploaded_file.name, save_dir
+
+video_file = st.file_uploader("Upload a video file", type=['mp4', 'mov', 'avi', 'flv', 'wmv', 'webm','mp3'])
+
+if video_file is not None:
+    tfile = tempfile.NamedTemporaryFile(delete=False) 
+    tfile.write(video_file.read())
+    st.video(tfile.name)
+
+    filename, video_path = save_uploaded_file(video_file)
+    # create_download_link(video_file)
+
+    save_dir = "/tmp/vocals"
+
+    ip_dir = video_path + "/" + filename
+    
+    # print(f"spleeter separate -o {save_dir} '{ip_dir}'")
+
+    # Example command to run
+    command = f"spleeter separate -o {save_dir} '{ip_dir}'"
+
+    # Run the shell command and display live results
+    return_code = run_shell_command(command)
+    st.write(f"\nCommand finished with return code: {return_code}")
+
+    op_file_vocals = save_dir + "/" + filename.split(".")[0] + "/vocals.wav"
+    op_file_music = save_dir + "/" + filename.split(".")[0] + "/accompaniment.wav"
+
+    st.success(f"File '{ip_dir}' processed into vocals {op_file_vocals} and music {op_file_music}")
+
+    st.audio(op_file_vocals)
+    # create_download_link(op_file_vocals)
+
+    st.audio(op_file_music)
+    # create_download_link(op_file_music)
+
+if st.button("Clear Data"):
+
+    command = f"rm -R uploaded_videos"
+    run_shell_command(command)
+
+    command = f"rm -R vocals"
+    run_shell_command(command)

requirements.txt

@@ -0,0 +1,2 @@
+streamlit==1.19.0
+spleeter==2.3.2