Update app.py

app.py

@@ -180,7 +180,7 @@ def detect_watermark(det_im):
 # Gradio app
 with gr.Blocks() as app:
     with gr.Tab("Add Watermark"):
-        cho = gr.Radio(choices=["stegan", ""] ,value="stenography", label="Watermark Method")
+        cho = gr.Radio(choices=["stegan", "pnginfo"], value="stegan", label="Watermark Method")  # Correct default value
         with gr.Row():
             with gr.Column():
                 inp_im = gr.Image(label="Input Image", type="filepath")
@@ -203,6 +203,23 @@ with gr.Blocks() as app:
 
         det_btn.click(detect_watermark, inputs=[det_im], outputs=[det_msg])
 
+if __name__ == "__main__":
+    app.launch(
+        show_api=False
+    )
+
+        mark_btn.click(choose_encode, inputs=[inp_im, inp_mark, cho], outputs=[out_im, msg_box, file_output])
+
+    with gr.Tab("Detect Watermark"):
+        with gr.Row():
+            with gr.Column():
+                det_im = gr.Image(label="Watermarked Image", type="filepath")
+                det_btn = gr.Button("Detect Watermark")
+            with gr.Column():
+                det_msg = gr.Textbox(label="Detected Watermark", lines=6)
+
+        det_btn.click(detect_watermark, inputs=[det_im], outputs=[det_msg])
+
 if __name__ == "__main__":
     app.launch(
         show_api=False