Update app.py

app.py

@@ -1,8 +1,17 @@
 import gradio as gr
 import utils
 
+def choose_encode(inp_im,inp_mark,cho):
+    if cho == "stegan":
+        out_im, out_msg = utils.encode(inp_im,inp_mark)
+        return out_im,out_msg
+    if cho == "pnginfo"
+        out_im, out_msg = utils.png_encode(inp_im,inp_mark)
+        return out_im,out_msg
+
 with gr.Blocks() as app:
     with gr.Tab("Add Watermark"):
+        cho = gr.Radio(choices=["stegan","pnginfo"])
         with gr.Row():
             with gr.Column():
                 inp_im = gr.Image(label="Input Image",type="filepath")
@@ -18,6 +27,6 @@ with gr.Blocks() as app:
                 det_btn = gr.Button("Detect")
             with gr.Column():
                 det_msg = gr.Textbox(label="Detected Watermark",lines=6, max_lines=50)
-    mark_btn.click(utils.encode,[inp_im,inp_mark],[out_im,msg_box])
+    mark_btn.click(choose_encode,[inp_im,inp_mark,cho],[out_im,msg_box])
     det_btn.click(utils.decode,[det_im],det_msg)
 app.launch()