Update app.py

app.py

@@ -1,111 +1,132 @@
-from pathlib import Path
-
-import gradio as gr
-import pi_heif
-import spaces
-import requests
-import base64
-import io
-import os
-import json
-from huggingface_hub import hf_hub_download
-from PIL import Image
-
-css = """
-footer {display: none !important;}
-"""
-
-TITLE = """
-<p>
-    Erase unwanted watermarks automatically with AI, no manual editing needed.
-</p><p>
-    For premium-quality results,
-    <a href="https://nowatermark.cloud">try NoWatermark API</a>
-    — it's free to test!
-</p>
-"""
-
-HEADERS = os.getenv("HEADERS")
-API_HOST = os.getenv("API_HOST")
-
-
-def remove_watermark(img_bytes):
-    headers = json.loads(HEADERS)
-
-    files = {
-        "original_preview_image": ("1.webp", img_bytes, "image/webp"),
-    }
-
-    data = {
-        "zoom_factor": 1,
-        "predict_mode": "old",
-    }
-
-    resp = requests.post(
-        API_HOST,
-        headers=headers,
-        files=files,
-        data=data,
-    )
-    if resp.status_code != 200:
-        print(f"Error: {resp.status_code} - {resp.text}")
-        return None
-
-    base64_image = resp.json().get("edited_image").get("image")
-    return base64.b64decode(base64_image)
-
-def process(input_image):
-    """
-    input_image: PIL.Image from Gradio
-    returns: PIL.Image (dewatermarked)
-    """
-    # Convert PIL image to bytes (WebP, since your API expects webp)
-    img_bytes_io = io.BytesIO()
-    input_image.save(img_bytes_io, format="WEBP")
-    img_bytes = img_bytes_io.getvalue()
-
-    # Call your dewatermark function
-    result_bytes = remove_watermark(img_bytes)
-    if result_bytes is None:
-        return [input_image, input_image]
-        # return None  # or return input_image as fallback
-
-    # Convert bytes back to PIL image
-    result_img = Image.open(io.BytesIO(result_bytes)).convert("RGB")
-    return [input_image, result_img]
-    
-with gr.Blocks(css=css) as demo:
-    gr.HTML(TITLE)
-
-    with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(type="pil", label="Input Image")
-            run_button = gr.ClearButton(components=None, value="Remove Watermark")
-        with gr.Column():
-            output_slider = gr.ImageSlider(label="Before / After", max_height=1200, show_fullscreen_button=False)
-            run_button.add(output_slider)
-
-    run_button.click(fn=process, inputs=input_image, outputs=output_slider, api_name=False)
-
-    gr.Examples(
-        examples=[
-            "examples/1.jpg",
-            "examples/2.jpg",
-            "examples/3.jpg",
-            "examples/4.jpg",
-            "examples/5.jpg",
-            "examples/6.jpg",
-            "examples/7.jpg",
-            "examples/8.jpg",
-            "examples/9.jpg",
-            "examples/10.jpg",
-        ],
-        inputs=[input_image],
-        outputs=output_slider,
-        fn=process,
-        cache_examples=True,
-        cache_mode="lazy",
-        run_on_click=False,
-    )
-
+from pathlib import Path
+
+import gradio as gr
+import pi_heif
+import spaces
+import requests
+import base64
+import io
+import os
+import json
+from huggingface_hub import hf_hub_download
+from PIL import Image
+
+css = """
+footer {display: none !important;}
+"""
+
+TITLE = """
+<p>
+    Erase unwanted watermarks automatically with AI, no manual editing needed.
+</p><p>
+    For premium-quality results,
+    <a href="https://nowatermark.cloud">try NoWatermark API</a>
+    — it's free to test!
+</p>
+"""
+
+LIKE_BUTTON = """
+<div style="text-align: center; margin-top: 20px;">
+  <a href="https://huggingface.co/spaces/abdul9999/NoWatermark" target="_blank">
+    <button style="
+      background-color: #ff4081;
+      color: white;
+      border: none;
+      padding: 12px 24px;
+      font-size: 16px;
+      font-weight: bold;
+      border-radius: 12px;
+      cursor: pointer;
+      transition: 0.3s;
+    ">
+      ❤️ Give a Like
+    </button>
+  </a>
+</div>
+"""
+
+HEADERS = os.getenv("HEADERS")
+API_HOST = os.getenv("API_HOST")
+
+
+def remove_watermark(img_bytes):
+    headers = json.loads(HEADERS)
+
+    files = {
+        "original_preview_image": ("1.webp", img_bytes, "image/webp"),
+    }
+
+    data = {
+        "zoom_factor": 1,
+        "predict_mode": "old",
+    }
+
+    resp = requests.post(
+        API_HOST,
+        headers=headers,
+        files=files,
+        data=data,
+    )
+    if resp.status_code != 200:
+        print(f"Error: {resp.status_code} - {resp.text}")
+        return None
+
+    base64_image = resp.json().get("edited_image").get("image")
+    return base64.b64decode(base64_image)
+
+def process(input_image):
+    """
+    input_image: PIL.Image from Gradio
+    returns: PIL.Image (dewatermarked)
+    """
+    # Convert PIL image to bytes (WebP, since your API expects webp)
+    img_bytes_io = io.BytesIO()
+    input_image.save(img_bytes_io, format="WEBP")
+    img_bytes = img_bytes_io.getvalue()
+
+    # Call your dewatermark function
+    result_bytes = remove_watermark(img_bytes)
+    if result_bytes is None:
+        return [input_image, input_image]
+        # return None  # or return input_image as fallback
+
+    # Convert bytes back to PIL image
+    result_img = Image.open(io.BytesIO(result_bytes)).convert("RGB")
+    return [input_image, result_img]
+    
+with gr.Blocks(css=css) as demo:
+    gr.HTML(TITLE)
+    gr.HTML(LIKE_BUTTON)
+
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Input Image")
+            run_button = gr.ClearButton(components=None, value="Remove Watermark")
+        with gr.Column():
+            output_slider = gr.ImageSlider(label="Before / After", max_height=1200, show_fullscreen_button=False)
+            run_button.add(output_slider)
+
+    run_button.click(fn=process, inputs=input_image, outputs=output_slider, api_name=False)
+
+    gr.Examples(
+        examples=[
+            "examples/1.jpg",
+            "examples/2.jpg",
+            "examples/3.jpg",
+            "examples/4.jpg",
+            "examples/5.jpg",
+            "examples/6.jpg",
+            "examples/7.jpg",
+            "examples/8.jpg",
+            "examples/9.jpg",
+            "examples/10.jpg",
+        ],
+        inputs=[input_image],
+        outputs=output_slider,
+        fn=process,
+        cache_examples=True,
+        cache_mode="lazy",
+        run_on_click=False,
+    )
+
 demo.launch(share=False, ssr_mode=False, show_api=False)