Update app.py

app.py

@@ -3,6 +3,7 @@ import whisper
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 import base64
+from pydub import AudioSegment
 
 st.set_page_config(
     page_title="Sing It Forward App",
@@ -105,4 +106,17 @@ with tab2:
     if uploaded_file is not None:
         with open("raw_music.mp3", "wb") as f:
             f.write(uploaded_file.getbuffer())
-        st.audio("raw_music.mp3")
+        st.audio("raw_music.mp3")
+        st.write("Specify start and end times for trimming:")
+        audio = AudioSegment.from_file("raw_music.mp3")
+        duration = len(audio) // 1000 
+        start_time = st.number_input("Start Time (seconds)", min_value=0, max_value=duration, value=0)
+        end_time = st.number_input("End Time (seconds)", min_value=0, max_value=duration, value=duration)
+        if start_time < end_time:
+            trimmed_audio = audio[start_time * 1000:end_time * 1000]
+            trimmed_audio.export("trimmed_music.mp3", format="mp3")
+            st.write("Now type what user should sing:")
+            user_input = st.text_area("Lyrics to be singed:")
+            st.write(user_input)
+        else:
+            st.write("Start Time should be smaller than End Time!")