Update app.py

app.py

@@ -190,26 +190,28 @@ def main():
                                 file_name="output.mp3",
                                 mime="audio/mpeg",
                             )
-
+                    except RuntimeError:
+                        st.error("Please input audio with one channel (mono-channel)")
+                                 
         elif action == "Detect Watermark":
             detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
 
-            if audio_file:
-                #1st attempt
-                watermark = model.get_watermark(wav, default_sr)
-                watermarked_audio = wav + watermark
-                print(watermarked_audio.size())
-                size = watermarked_audio.size()
-                #st.markdown(size)
+            # if audio_file:
+            #     #1st attempt
+            #     watermark = model.get_watermark(wav, default_sr)
+            #     watermarked_audio = wav + watermark
+            #     print(watermarked_audio.size())
+            #     size = watermarked_audio.size()
+            #     #st.markdown(size)
 
                 
             if detect_watermark_button:
                 with st.spinner("Detecting..."):
-                    result, message = detector.detect_watermark(watermarked_audio, sample_rate=default_sr, message_threshold=0.5)
-                    st.markdown("Probability of audio being watermarked: ")
-                    st.markdown(result)
-                    st.markdown("This is likely a watermarked audio!")
-                    print(f"\nThis is likely a watermarked audio: {result}")
+                    # result, message = detector.detect_watermark(watermarked_audio, sample_rate=default_sr, message_threshold=0.5)
+                    # st.markdown("Probability of audio being watermarked: ")
+                    # st.markdown(result)
+                    # st.markdown("This is likely a watermarked audio!")
+                    # print(f"\nThis is likely a watermarked audio: {result}")
 
                     #Run on an unwatermarked audio
 
@@ -227,7 +229,7 @@ def main():
                         
                     result2, message2 = detector.detect_watermark(wav, sample_rate=default_sr, message_threshold=0.5)
                     print(f"This is likely an unwatermarked audio: {result2}")
-                    st.markdown("Probability of audio being watermarked (Original Audio): ")
+                    st.markdown("Probability of audio being watermarked: ")
                     st.markdown(result2)
                     st.markdown("This is likely an unwatermarked audio!")