Update app.py

app.py

@@ -1,23 +1,30 @@
+import time
+import 
 import streamlit as st
 from transformers import pipeline
-from PIL import Image
+import os
+import torch
+import datetime
+import numpy as np
+import soundfile
+from wavmark.utils import file_reader
 
-pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+# pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
-st.title("Hot Dog? Or Not?")
+# st.title("Hot Dog? Or Not?")
 
-file_name = st.file_uploader("Upload a hot dog candidate image")
+# file_name = st.file_uploader("Upload a hot dog candidate image")
 
-if file_name is not None:
-    col1, col2 = st.columns(2)
+# if file_name is not None:
+#     col1, col2 = st.columns(2)
 
-    image = Image.open(file_name)
-    col1.image(image, use_column_width=True)
-    predictions = pipeline(image)
+#     image = Image.open(file_name)
+#     col1.image(image, use_column_width=True)
+#     predictions = pipeline(image)
 
-    col2.header("Probabilities")
-    for p in predictions:
-        col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
+#     col2.header("Probabilities")
+#     for p in predictions:
+#         col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
 
 # Main web app
 def main():
@@ -50,23 +57,23 @@ def main():
 
         action = st.selectbox("Select Action", ["Add Watermark", "Decode Watermark"])
 
-        if action == "Add Watermark":
-            watermark_text = st.text_input("The watermark (0, 1 list of length-16):", value=st.session_state.def_value)
-            add_watermark_button = st.button("Add Watermark", key="add_watermark_btn")
-            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)
-                        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)
-
-                        # st.button("Add Watermark", disabled=False)
-        elif action == "Decode Watermark":
-            if st.button("Decode"):
-                with st.spinner("Decoding..."):
-                    decode_watermark(tmp_input_audio_file)
+#         if action == "Add Watermark":
+#             watermark_text = st.text_input("The watermark (0, 1 list of length-16):", value=st.session_state.def_value)
+#             add_watermark_button = st.button("Add Watermark", key="add_watermark_btn")
+#             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)
+#                         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)
+
+#                         # st.button("Add Watermark", disabled=False)
+#         elif action == "Decode Watermark":
+#             if st.button("Decode"):
+#                 with st.spinner("Decoding..."):
+#                     decode_watermark(tmp_input_audio_file)
 
 
 if __name__ == "__main__":