Synced repo using 'sync_with_huggingface' Github Action

app.py

@@ -5,7 +5,7 @@ import subprocess
 from huggingface_hub import hf_hub_download
 
 REPO_URL = "https://github.com/facebookresearch/watermark-anything.git"
-REPO_BRANCH = '45d56c2b61f2bc73caeafc90e14df33ad50b238c'
+REPO_BRANCH = 'bbec07bca82a416e5a6ff9d75a295cae5c166aaf'
 LOCAL_PATH = "./watermark-anything"
 MODEL_ID = "facebook/watermark-anything"
 

gradio_app.py

@@ -137,6 +137,21 @@ if '__main__' == __name__:
     wam = load_wam()
 
     def detect_watermark(image, multi, timeout):
+        """
+        Detects watermark(s) in the given image.
+
+        Args:
+            image (Union[PIL.Image.Image, str]): Input image, either a PIL Image object or a URL pointing to the image.
+            multi (bool): If True, enables detection of multiple watermarks in the image.
+            timeout (float): Timeout in seconds for image loading or processing; used to handle unresponsive or invalid images.
+
+        Returns:
+            tuple: A 4-element tuple containing:
+                - dict: An image dict of the predicted watermark positions.
+                - dict: An image dict of watermark clusters.
+                - Union[str, float, bool, list, dict]: Metadata or messages related to the detection.
+                - str: A markdown string summarizing the detection result (e.g., confidence level).
+        """
         if image is None:
             return None, None, None, {"status": "error", "messages": [], "error": "No image provided"}
 
@@ -198,6 +213,26 @@ if '__main__' == __name__:
         return pred_mask, cluster_viz, message_json, "\n".join(color_md)
 
     def embed_watermark(image, wm_num, wm_type, wm_str, wm_loc):
+        """
+        Embeds one or more watermarks into the input image.
+
+        Args:
+            image (Union[PIL.Image.Image, str]): The input image, either as a PIL Image object or a URL string.
+            wm_num (float): The number of watermarks to embed.
+            wm_type (Literal['random', 'input']): The type of watermark. 
+                - 'random': Use randomly generated watermark content.
+                - 'input': Use the content from `wm_str` as the watermark.
+            wm_str (str): The text content of the watermark when `wm_type='input'`.
+            wm_loc (Literal['random', 'bounding']): The placement mode of the watermark.
+                - 'random': Randomly place the watermark(s).
+                - 'bounding': Place within a bounding box or defined area.
+
+        Returns:
+            tuple: A 3-element tuple containing:
+                - dict: Watermarked image output (shown in "Watermarked Image" component).
+                - dict: Image showing the watermark position (shown in "Position of the watermark").
+                - Union[str, float, bool, list, dict]: Metadata or message related to the embedding process (shown in "Marked Messages").
+        """
         if image is None:
             return None, None, {
                 "status": "failure",
@@ -327,6 +362,7 @@ if '__main__' == __name__:
     with gr.Blocks(title="Watermark Anything Demo") as demo:
         gr.Markdown("""
         # Watermark Anything Demo
+        ![](https://badge.mcpx.dev?type=server 'MCP Server')
         This app demonstrates watermark detection and embedding using the Watermark Anything model.
         Find the project [here](https://github.com/facebookresearch/watermark-anything).
         """)
@@ -360,11 +396,13 @@ if '__main__' == __name__:
                 embedding_img.select(
                     fn=get_select_coordinates,
                     inputs=[embedding_img, embedding_num],
-                    outputs=embedding_box)
+                    outputs=embedding_box,
+                    api_name=False)
                 embedding_box.select(
                     fn=del_select_coordinates,
                     inputs=embedding_box,
-                    outputs=embedding_box
+                    outputs=embedding_box,
+                    api_name=False
                 )
 
                 # The inability to dynamically render `AnnotatedImage` is because,
@@ -397,7 +435,8 @@ if '__main__' == __name__:
                 embedding_loc.change(
                     fn=visible_box_image,
                     inputs=[embedding_img, embedding_loc],
-                    outputs=[embedding_box]
+                    outputs=[embedding_box],
+                    api_name=False
                 )
 
                 def visible_text_label(embedding_type, embedding_num):
@@ -418,17 +457,20 @@ if '__main__' == __name__:
                 embedding_num.change(
                     fn=visible_text_label,
                     inputs=[embedding_type, embedding_num],
-                    outputs=[embedding_str]
+                    outputs=[embedding_str],
+                    api_name=False
                 )
                 embedding_type.change(
                     fn=visible_text_label,
                     inputs=[embedding_type, embedding_num],
-                    outputs=[embedding_str]
+                    outputs=[embedding_str],
+                    api_name=False
                 )
                 embedding_str.change(
                     fn=check_embedding_str,
                     inputs=[embedding_str, embedding_num],
-                    outputs=[embedding_btn]
+                    outputs=[embedding_btn],
+                    api_name=False
                 )
 
                 embedding_btn.click(
@@ -460,12 +502,14 @@ if '__main__' == __name__:
                 detecting_img.change(
                     fn=lambda x: gr.update(value=False),
                     inputs=detecting_img,
-                    outputs=multi_ckb
+                    outputs=multi_ckb,
+                    api_name=False
                 )
                 multi_ckb.change(
                     fn=lambda x: gr.update(visible=x),
                     inputs=multi_ckb,
-                    outputs=timeout_sli
+                    outputs=timeout_sli,
+                    api_name=False
                 )
                 detecting_btn.click(
                     fn=detect_watermark,
@@ -473,6 +517,5 @@ if '__main__' == __name__:
                     outputs=[predicted_mask, predicted_cluster, predicted_messages, color_cluster]
                 )
 
-
     if __name__ == '__main__':
-        demo.launch()
+        demo.launch(server_name="0.0.0.0", server_port=7860, mcp_server=True, ssr_mode=False)

requirements.txt

@@ -1,5 +1,8 @@
 torch==2.5.1
 GitPython==3.1.43
-gradio==5.8.0
+gradio[mcp]==5.28.0
 huggingface-hub==0.26.3
-wrapt-timeout-decorator==1.5.1
+wrapt-timeout-decorator==1.5.1
+
+# gradio[mcp] 5.28.0 depends on pydantic>=2.11
+pydantic==2.11.4