Update app.py

app.py

@@ -141,7 +141,8 @@ This tool embeds a unique, invisible watermark into your audio files to mark the
             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.
+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.
@@ -154,16 +155,8 @@ Use this feature to check if an audio file contains a specific watermark. Here's
                 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])
+    return demo
 
-    with gr.Tab("Detect Watermark"):
-        with gr.Column(scale=6):
-            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])
-
-demo.launch()
+if __name__ == "__main__":
+    demo = gradio_interface()
+    demo.launch()