Update app.py

app.py

@@ -130,7 +130,7 @@ def main():
         #Unsqueeze for line 176
         # wav= wav.unsqueeze(0)
 
-        action = st.selectbox("Select Action", ["Add Watermark", "Decode Watermark"])
+        action = st.selectbox("Select Action", ["Add Watermark", "Detect Watermark"])
 
         if action == "Add Watermark":
             watermark_text = st.text_input("The watermark (0, 1 list of length-16):", value=st.session_state.def_value)
@@ -191,16 +191,17 @@ def main():
                             )
 
         elif action == "Detect Watermark":
-            Detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
-            if st.button("Detect"):
+            detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
+            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(result)
                     print(f"\nThis is likely a watermarked audio: {result}")
 
                     # Run on an unwatermarked audio
                     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(result)
 
 
 if __name__ == "__main__":