encode audio into bytes

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 import numpy as np
 import os
+import base64
 
 # Try to get API_URL from environment variables, if not found set to a default value
 try:
@@ -25,6 +26,8 @@ headers = {
 # Streamlit app title
 st.title("Songlabai")
 
+uploaded_file = st.file_uploader("Upload Music File", type=["mp3", "wav", "ogg", "flac", "aac"])
+
 genres = [
     "Pop", "Rock", "Hip Hop", "Jazz", "Blues",
     "Country", "Classical", "Electronic", "Reggae",
@@ -46,8 +49,12 @@ duration = st.slider("Duration (in seconds):", min_value=15, max_value=90, value
 # Generate audio based on the user's prompt
 if st.button("Generate Audio"):
     prompt = f"{genre} ,{energy_level}, {tempo}, {description}"
+    payload = {"inputs": {"prompt": prompt, "duration": duration}}
+    if uploaded_file:
+        audio_base64 = base64.b64encode(audio_bytes).decode('utf-8')
+        payload = {"inputs": {"prompt": prompt, "duration": duration, "track": audio_base64}}
     st.text("Generating audio...")
-    response = requests.post(API_URL, headers=headers, json={"inputs": {"prompt": prompt, "duration": duration}})
+    response = requests.post(API_URL, headers=headers, json=payload)
     audio = np.array(response.json()[0]['generated_audio'], dtype=np.float32)
     sample_rate = response.json()[0]['sample_rate']
     st.audio(audio, format="audio/wav", sample_rate=sample_rate, start_time=0)