Update app.py

app.py

@@ -0,0 +1,50 @@
+import streamlit as st
+import torchaudio
+import tempfile
+import os
+from pydub import AudioSegment
+from bark import generate_audio, SAMPLE_RATE as BARK_RATE
+from audiocraft.models import MusicGen
+
+# Load MusicGen model
+musicgen = MusicGen.get_pretrained('facebook/musicgen-small')
+
+st.set_page_config(page_title="Suno Clone", layout="centered")
+st.title("Suno-like Music Generator")
+st.markdown("Combining **Bark TTS**, **RVC Voice Cloning**, and **MusicLM** for full track creation.")
+
+lyrics = st.text_area("Enter your lyrics:", height=150)
+genre_prompt = st.text_input("Enter Music Style (e.g., 'trap beat with piano')", value="trap beat with piano")
+uploaded_voice = st.file_uploader("Upload your voice sample (for cloning - WAV only)", type=["wav"])
+
+if st.button("Generate Song"):
+    with st.spinner("Generating vocals with Bark..."):
+        vocals_tensor = generate_audio(lyrics, history_prompt="v2/en_speaker_6")
+        vocals_path = tempfile.mktemp(suffix=".wav")
+        torchaudio.save(vocals_path, vocals_tensor.squeeze(0).cpu(), BARK_RATE)
+
+    if uploaded_voice:
+        with st.spinner("Cloning voice with RVC..."):
+            uploaded_voice_path = tempfile.mktemp(suffix=".wav")
+            with open(uploaded_voice_path, "wb") as f:
+                f.write(uploaded_voice.read())
+
+            cloned_path = tempfile.mktemp(suffix=".wav")
+            os.system(f"python infer_rvc.py --input \"{vocals_path}\" --voice \"{uploaded_voice_path}\" --output \"{cloned_path}\"")
+            vocals_path = cloned_path
+
+    with st.spinner("Generating instrumental with MusicLM..."):
+        musicgen.set_generation_params(duration=15)
+        music = musicgen.generate([genre_prompt])
+        instrumental_path = tempfile.mktemp(suffix=".wav")
+        torchaudio.save(instrumental_path, music[0].cpu(), 32000)
+
+    with st.spinner("Mixing vocals and instrumental..."):
+        vocals_seg = AudioSegment.from_wav(vocals_path)
+        instrumental_seg = AudioSegment.from_wav(instrumental_path)
+        mixed = instrumental_seg.overlay(vocals_seg.set_frame_rate(32000).set_channels(1))
+        final_output_path = tempfile.mktemp(suffix=".wav")
+        mixed.export(final_output_path, format="wav")
+
+    st.audio(final_output_path)
+    st.success("Your song is ready!")