clean up

app.py

@@ -40,18 +40,14 @@ def add_watermark(audio_path, watermark_text):
 def decode_watermark(audio_path):
     assert os.path.exists(audio_path)
 
-    #t1 = time.time()
     signal, sr, audio_length_second = my_read_file(audio_path, max_second_decode)
     payload_decoded, _ = wavmark.decode_watermark(model, signal, show_progress=False)
-    #decode_cost = time.time() - t1
 
     if payload_decoded is None:
-       #return "No Watermark" , decode_cost
        return "No Watermark"
 
     payload_decoded_str = "".join([str(i) for i in payload_decoded])
     st.write("Result:", payload_decoded_str)
-    #st.write("Time Cost:%d seconds" % (decode_cost))
 
 
 def create_default_value():
@@ -65,7 +61,7 @@ def create_default_value():
 def main():
     create_default_value()
 
-    st.title("AudioWaterMarking")
+    
     markdown_text = """
     # Audio WaterMarking
     You can upload an audio file and encode a custom 16-bit watermark or perform decoding from a watermarked audio.
@@ -83,8 +79,7 @@ def main():
         with open(tmp_input_audio_file, "wb") as f:
             f.write(audio_file.getbuffer())
 
-        
-        # st.audio(tmp_input_audio_file, format="audio/wav")
+
 
         action = st.selectbox("Select Action", ["Add Watermark", "Decode Watermark"])
 
@@ -94,13 +89,11 @@ def main():
             if add_watermark_button:  
                 if audio_file and watermark_text:
                     with st.spinner("Adding Watermark..."):
-                        #watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
                         watermarked_audio = add_watermark(tmp_input_audio_file, watermark_text)
                         st.write("Watermarked Audio:")
                         print("watermarked_audio:", watermarked_audio)
                         st.audio(watermarked_audio, format="audio/wav")
-                        #st.write("Time Cost: %d seconds" % encode_time_cost)
-
+                        
         elif action == "Decode Watermark":
             if st.button("Decode"):
                 with st.spinner("Decoding..."):