Update app.py

app.py

@@ -1,27 +1,27 @@
 import streamlit as st
 import torchaudio
-import torch
 from audiocraft.models import MusicGen
+import requests
 from pydub import AudioSegment
 import os
 
+# Set Streamlit page configuration
 st.set_page_config(page_title="Suno-like AI Music Generator", layout="centered")
 st.title("Suno-like AI Music Generator")
 
-# Load the pre-recorded male voice sample
+# Step 1: Load Pre-recorded AI Voice Sample
 st.subheader("Step 1: Pre-recorded AI Voice Sample")
-voice_path = "sample_voice.mp3"
-
-if os.path.exists(voice_path):
-    audio_file = open(voice_path, 'rb')
-    audio_bytes = audio_file.read()
-    st.audio(audio_bytes, format='audio/mp3')
-else:
-    st.error("Voice sample not found. Please make sure 'sample_voice.mp3' is in the same directory.")
-
-# Generate Instrumental with MusicGen
+voice_url = "https://www2.cs.uic.edu/~i101/SoundFiles/StarWars60.wav"  # Example URL for voice sample
+try:
+    audio_data = requests.get(voice_url).content
+    st.audio(audio_data, format="audio/wav")
+except Exception as e:
+    st.error(f"Failed to load audio: {e}")
+
+# Step 2: Generate Instrumental with MusicGen
 st.subheader("Step 2: Generate Instrumental with MusicGen")
-prompt = st.text_input("Enter music description (e.g. energetic trap beat with dark mood)", "emotional trap beat with dark pads and heavy drums")
+prompt = st.text_input("Enter music description (e.g. energetic trap beat with dark mood)", 
+                       "emotional trap beat with dark pads and heavy drums")
 
 if st.button("Generate Music"):
     with st.spinner("Generating music..."):
@@ -33,22 +33,29 @@ if st.button("Generate Music"):
         st.audio(output_path, format="audio/wav")
         st.success("Music generation complete!")
 
-# Combine voice and beat (optional step to simulate Suno-like output)
-st.subheader("Step 3: Simulated Combination")
-if os.path.exists("musicgen_output.wav") and os.path.exists("sample_voice.mp3"):
+# Step 3: Combine Voice and Instrumental (Simulated)
+st.subheader("Step 3: Simulated Combination of Voice + Instrumental")
+
+if os.path.exists("musicgen_output.wav"):
     if st.button("Combine Voice + Instrumental"):
-        # Convert MP3 voice to WAV for merging
-        voice = AudioSegment.from_mp3("sample_voice.mp3")
-        beat = AudioSegment.from_wav("musicgen_output.wav")
-
-        # Adjust volumes and overlay
-        voice = voice - 4
-        beat = beat - 2
-        combined = beat.overlay(voice, loop=False)
-        combined_path = "combined_output.mp3"
-        combined.export(combined_path, format="mp3")
-
-        st.audio(combined_path, format="audio/mp3")
-        st.success("Combined output ready!")
+        try:
+            # Load the generated instrumental and the pre-recorded voice
+            beat = AudioSegment.from_wav("musicgen_output.wav")
+            voice = AudioSegment.from_wav("StarWars60.wav")  # Use the public voice sample
+
+            # Adjust volumes and overlay (to avoid distortion)
+            voice = voice - 4  # Reduce the voice volume
+            beat = beat - 2    # Reduce the beat volume
+
+            # Combine voice and instrumental
+            combined = beat.overlay(voice, loop=False)
+            combined_path = "combined_output.mp3"
+            combined.export(combined_path, format="mp3")
+
+            # Play the combined audio
+            st.audio(combined_path, format="audio/mp3")
+            st.success("Combined output ready!")
+        except Exception as e:
+            st.error(f"Error while combining: {e}")
 else:
-    st.info("Generate the beat and make sure voice sample is available.")
+    st.info("Generate the music first to combine it with the voice.")