Update app.py

app.py

@@ -72,23 +72,37 @@ def cosine_sim(text1, text2):
     
 
 model = whisper.load_model("base")
-st.write("Listen to music since you have to record 15seconds after that")
-st.audio("titanic.mp3")
-
-
-audio_value = st.experimental_audio_input("Sing Rest of music:🎙️")
-lyrics = "Far across the distance And spaces between us You have come to show you go on"
-
-if audio_value:
-    with open("user_sing.mp3", "wb") as f:
-        f.write(audio_value.getbuffer())
-
-    user_lyrics = model.transcribe("user_sing.mp3")["text"]
-    st.write(user_lyrics)
-    similarity_score = cosine_sim(lyrics, user_lyrics)
-    if similarity_score > 0.85:
-        st.success('Awsome! You are doing great', icon="✅")
-        st.markdown('<style>div.stAlert { background-color: rgba(3, 67, 24, 0.9); }</style>', unsafe_allow_html=True)
-    else:
-        st.error('Awful! Try harder next time', icon="🚨")
-        st.markdown('<style>div.stAlert { background-color: rgba(241, 36, 36, 0.9); }</style>', unsafe_allow_html=True)
+
+
+tab1, tab2 = st.tabs(["Take Challenge", "Make Challenge"])
+
+with tab1:
+    st.write("Listen to music since you have to record 15seconds after that")
+    st.audio("titanic.mp3")
+    
+    
+    audio_value = st.experimental_audio_input("Sing Rest of music:🎙️")
+    lyrics = "Far across the distance And spaces between us You have come to show you go on"
+    
+    if audio_value:
+        with open("user_sing.mp3", "wb") as f:
+            f.write(audio_value.getbuffer())
+    
+        user_lyrics = model.transcribe("user_sing.mp3")["text"]
+        st.write(user_lyrics)
+        similarity_score = cosine_sim(lyrics, user_lyrics)
+        if similarity_score > 0.85:
+            st.success('Awsome! You are doing great', icon="✅")
+            st.markdown('<style>div.stAlert { background-color: rgba(3, 67, 24, 0.9); }</style>', unsafe_allow_html=True)
+        else:
+            st.error('Awful! Try harder next time', icon="🚨")
+            st.markdown('<style>div.stAlert { background-color: rgba(241, 36, 36, 0.9); }</style>', unsafe_allow_html=True)
+
+
+with tab2:
+    st.write("Upload music to make challenge:")
+    uploaded_file = st.file_uploader("Choose a music file", type=["mp3", "wav"])
+    if uploaded_file is not None:
+        with open("raw_music.mp3", "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        st.audio("raw_music.mp3")