Update app.py

app.py

@@ -115,18 +115,46 @@ else:
     style = ""
 
 # Define Gradio interface
-with gr.Blocks(css=style) as demo:
-    with gr.Tab("Watermark Audio"):
-        with gr.Column(scale=6):
-            audio_input_watermark = gr.Audio(label="Upload Audio File for Watermarking", type="filepath")
-            unique_message_output = gr.Textbox(label="Unique Message")
-            watermarked_audio_output = gr.Audio(label="Watermarked Audio")
-            message_output = gr.Textbox(label="Message Used for Watermarking")
-            enhanced_id_output = gr.Textbox(label="Enhanced ID")
-            generate_message_button = gr.Button("Generate Unique Message")
-            watermark_button = gr.Button("Apply Watermark")
-            generate_message_button.click(fn=generate_unique_message, inputs=None, outputs=unique_message_output)
-            watermark_button.click(fn=watermark_audio, inputs=[audio_input_watermark, unique_message_output], outputs=[watermarked_audio_output, message_output, enhanced_id_output])
+def gradio_interface():
+    with gr.Blocks(css=style) as demo:
+        with gr.Tab("Watermark Audio"):
+            with gr.Column(scale=6):
+                gr.Markdown("""**How to Watermark Your Audio**
+This tool embeds a unique, invisible watermark into your audio files to mark them as yours. Follow these steps:
+1. **Upload Your Audio**: Choose the audio file you want to watermark.
+2. **Generate Unique Message**: Click this button to generate a unique code that will serve as your audio's watermark.
+3. **Apply Watermark**: Embed the watermark into your audio file. This process does not alter the audio's quality.
+4. **Download Watermarked Audio**: After the watermark is applied, you can download the watermarked audio. It will sound identical to the original but now contains your unique watermark.
+5. **View Enhanced ID**: Along with the watermarked audio, you'll get an enhanced ID for additional tracking and verification purposes.
+""")
+                audio_input_watermark = gr.Audio(label="Upload Audio File for Watermarking", type="filepath")
+                unique_message_output = gr.Textbox(label="Unique Message")
+                watermarked_audio_output = gr.Audio(label="Watermarked Audio")
+                message_output = gr.Textbox(label="Message Used for Watermarking")
+                enhanced_id_output = gr.Textbox(label="Enhanced ID")
+                generate_message_button = gr.Button("Generate Unique Message")
+                watermark_button = gr.Button("Apply Watermark")
+                generate_message_button.click(fn=generate_unique_message, inputs=None, outputs=unique_message_output)
+                watermark_button.click(fn=watermark_audio, inputs=[audio_input_watermark, unique_message_output], outputs=[watermarked_audio_output, message_output, enhanced_id_output])
+
+        with gr.Tab("Detect Watermark"):
+            with gr.Column(scale=6):
+                gr.Markdown("""**How to Detect a Watermark in Your Audio**
+Use this feature to check if an audio file contains a specific watermark. Here's how:
+1. **Upload the Audio File**: Select the audio file you want to check.
+2. **Enter Original Hex Message**: If you know the hexadecimal code of the watermark, enter it here for a precise search.
+3. **Detect Watermark**: Click to analyze the audio for your watermark.
+4. **Review Results**: Find out whether your watermark was detected and if the detected code matches your input.
+""")
+                audio_input_detect_watermark = gr.Audio(label="Upload Audio File for Watermark Detection", type="filepath")
+                original_hex_input = gr.Textbox(label="Original Hex Message for Comparison", placeholder="Enter the original hex message here")
+                detect_watermark_button = gr.Button("Detect Watermark")
+                watermark_detection_output = gr.Textbox(label="Watermark Detection Result")
+                detected_message_output = gr.Textbox(label="Detected Hex Message")
+                match_result_output = gr.Textbox(label="Match Result")
+                detect_watermark_button.click(fn=detect_watermark, inputs=[audio_input_detect_watermark, original_hex_input], outputs=[watermark_detection_output, detected_message_output, match_result_output])
+
+    return demo enhanced_id_output])
 
     with gr.Tab("Detect Watermark"):
         with gr.Column(scale=6):