add duplication notice banners

gradio_app.py

@@ -1,16 +1,28 @@
 import os
+import torch
 
 os.environ['HF_HOME'] = os.path.join(os.path.dirname(__file__), 'hf_download')
 result_dir = os.path.join('./', 'results')
 os.makedirs(result_dir, exist_ok=True)
 
+is_shared_ui = True if "fffiloni/Paints-UNDO" in os.environ['SPACE_ID'] else False
+
+is_gpu_associated = torch.cuda.is_available()
+
+if is_gpu_associated:
+    gpu_info = getoutput('nvidia-smi')
+    if("A10G" in gpu_info):
+        which_gpu = "A10G"
+    elif("T4" in gpu_info):
+        which_gpu = "T4"
+    else:
+        which_gpu = "CPU"
 
 import functools
-import os
 import random
 import gradio as gr
 import numpy as np
-import torch
+
 import wd14tagger
 import memory_management
 import uuid
@@ -112,12 +124,19 @@ def resize_without_crop(image, target_width, target_height):
 
 @torch.inference_mode()
 def interrogator_process(x):
+    if is_shared_ui:
+        raise gr.Error("This Space only works in duplicated instances")
+
     return wd14tagger.default_interrogator(x)
 
 
 @torch.inference_mode()
 def process(input_fg, prompt, input_undo_steps, image_width, image_height, seed, steps, n_prompt, cfg,
             progress=gr.Progress()):
+    
+    if is_shared_ui:
+        raise gr.Error("This Space only works in duplicated instances")
+
     rng = torch.Generator(device=memory_management.gpu).manual_seed(int(seed))
 
     memory_management.load_models_to_gpu(vae)
@@ -213,6 +232,9 @@ def process_video_inner(image_1, image_2, prompt, seed=123, steps=25, cfg_scale=
 
 @torch.inference_mode()
 def process_video(keyframes, prompt, steps, cfg, fps, seed, progress=gr.Progress()):
+    if is_shared_ui:
+        raise gr.Error("This Space only works in duplicated instances")
+
     result_frames = []
     cropped_images = []
 
@@ -236,9 +258,112 @@ def process_video(keyframes, prompt, steps, cfg, fps, seed, progress=gr.Progress
     video = [x.cpu().numpy() for x in video]
     return output_filename, video
 
-
-block = gr.Blocks().queue()
+css = """
+div#warning-ready {
+    background-color: #ecfdf5;
+    padding: 0 16px 16px;
+    margin: 20px 0;
+    color: #030303!important;
+}
+div#warning-ready > .gr-prose > h2, div#warning-ready > .gr-prose > p {
+    color: #057857!important;
+}
+div#warning-duplicate {
+    background-color: #ebf5ff;
+    padding: 0 16px 16px;
+    margin: 20px 0;
+    color: #030303!important;
+}
+div#warning-duplicate > .gr-prose > h2, div#warning-duplicate > .gr-prose > p {
+    color: #0f4592!important;
+}
+div#warning-duplicate strong {
+    color: #0f4592;
+}
+p.actions {
+    display: flex;
+    align-items: center;
+    margin: 20px 0;
+}
+div#warning-duplicate .actions a {
+    display: inline-block;
+    margin-right: 10px;
+}
+div#warning-setgpu {
+    background-color: #fff4eb;
+    padding: 0 16px 16px;
+    margin: 20px 0;
+    color: #030303!important;
+}
+div#warning-setgpu > .gr-prose > h2, div#warning-setgpu > .gr-prose > p {
+    color: #92220f!important;
+}
+div#warning-setgpu a, div#warning-setgpu b {
+    color: #91230f;
+}
+div#warning-setgpu p.actions > a {
+    display: inline-block;
+    background: #1f1f23;
+    border-radius: 40px;
+    padding: 6px 24px;
+    color: antiquewhite;
+    text-decoration: none;
+    font-weight: 600;
+    font-size: 1.2em;
+}
+div#warning-setsleeptime {
+    background-color: #fff4eb;
+    padding: 10px 10px;
+    margin: 0!important;
+    color: #030303!important;
+}
+.custom-color {
+    color: #030303 !important;
+}
+"""
+
+block = gr.Blocks(css=css).queue()
 with block:
+    if is_shared_ui:
+            top_description = gr.HTML(f'''
+                <div class="gr-prose">
+                    <h2 class="custom-color"><svg xmlns="http://www.w3.org/2000/svg" width="18px" height="18px" style="margin-right: 0px;display: inline-block;"fill="none"><path fill="#fff" d="M7 13.2a6.3 6.3 0 0 0 4.4-10.7A6.3 6.3 0 0 0 .6 6.9 6.3 6.3 0 0 0 7 13.2Z"/><path fill="#fff" fill-rule="evenodd" d="M7 0a6.9 6.9 0 0 1 4.8 11.8A6.9 6.9 0 0 1 0 7 6.9 6.9 0 0 1 7 0Zm0 0v.7V0ZM0 7h.6H0Zm7 6.8v-.6.6ZM13.7 7h-.6.6ZM9.1 1.7c-.7-.3-1.4-.4-2.2-.4a5.6 5.6 0 0 0-4 1.6 5.6 5.6 0 0 0-1.6 4 5.6 5.6 0 0 0 1.6 4 5.6 5.6 0 0 0 4 1.7 5.6 5.6 0 0 0 4-1.7 5.6 5.6 0 0 0 1.7-4 5.6 5.6 0 0 0-1.7-4c-.5-.5-1.1-.9-1.8-1.2Z" clip-rule="evenodd"/><path fill="#000" fill-rule="evenodd" d="M7 2.9a.8.8 0 1 1 0 1.5A.8.8 0 0 1 7 3ZM5.8 5.7c0-.4.3-.6.6-.6h.7c.3 0 .6.2.6.6v3.7h.5a.6.6 0 0 1 0 1.3H6a.6.6 0 0 1 0-1.3h.4v-3a.6.6 0 0 1-.6-.7Z" clip-rule="evenodd"/></svg>
+                    Attention: this Space need to be duplicated to work</h2>
+                    <p class="main-message custom-color">
+                        To make it work, <strong>duplicate the Space</strong> and run it on your own profile using a <strong>private</strong> GPU (T4-small or A10G-small).<br />
+                        A T4 costs <strong>US$0.60/h</strong>.
+                    </p>
+                    <p class="actions custom-color">
+                        <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}?duplicate=true">
+                            <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-lg-dark.svg" alt="Duplicate this Space" />
+                        </a>
+                        to start experimenting with this demo
+                    </p>
+                </div>
+            ''', elem_id="warning-duplicate")
+        else:
+            if(is_gpu_associated):
+                top_description = gr.HTML(f'''
+                        <div class="gr-prose">
+                            <h2 class="custom-color"><svg xmlns="http://www.w3.org/2000/svg" width="18px" height="18px" style="margin-right: 0px;display: inline-block;"fill="none"><path fill="#fff" d="M7 13.2a6.3 6.3 0 0 0 4.4-10.7A6.3 6.3 0 0 0 .6 6.9 6.3 6.3 0 0 0 7 13.2Z"/><path fill="#fff" fill-rule="evenodd" d="M7 0a6.9 6.9 0 0 1 4.8 11.8A6.9 6.9 0 0 1 0 7 6.9 6.9 0 0 1 7 0Zm0 0v.7V0ZM0 7h.6H0Zm7 6.8v-.6.6ZM13.7 7h-.6.6ZM9.1 1.7c-.7-.3-1.4-.4-2.2-.4a5.6 5.6 0 0 0-4 1.6 5.6 5.6 0 0 0-1.6 4 5.6 5.6 0 0 0 1.6 4 5.6 5.6 0 0 0 4 1.7 5.6 5.6 0 0 0 4-1.7 5.6 5.6 0 0 0 1.7-4 5.6 5.6 0 0 0-1.7-4c-.5-.5-1.1-.9-1.8-1.2Z" clip-rule="evenodd"/><path fill="#000" fill-rule="evenodd" d="M7 2.9a.8.8 0 1 1 0 1.5A.8.8 0 0 1 7 3ZM5.8 5.7c0-.4.3-.6.6-.6h.7c.3 0 .6.2.6.6v3.7h.5a.6.6 0 0 1 0 1.3H6a.6.6 0 0 1 0-1.3h.4v-3a.6.6 0 0 1-.6-.7Z" clip-rule="evenodd"/></svg>
+                            You have successfully associated a {which_gpu} GPU to the Paints UNDO Space 🎉</h2>
+                            <p class="custom-color">
+                                You will be billed by the minute from when you activated the GPU until when it is turned off.
+                            </p> 
+                        </div>
+                ''', elem_id="warning-ready")
+            else:
+                top_description = gr.HTML(f'''
+                        <div class="gr-prose">
+                        <h2 class="custom-color"><svg xmlns="http://www.w3.org/2000/svg" width="18px" height="18px" style="margin-right: 0px;display: inline-block;"fill="none"><path fill="#fff" d="M7 13.2a6.3 6.3 0 0 0 4.4-10.7A6.3 6.3 0 0 0 .6 6.9 6.3 6.3 0 0 0 7 13.2Z"/><path fill="#fff" fill-rule="evenodd" d="M7 0a6.9 6.9 0 0 1 4.8 11.8A6.9 6.9 0 0 1 0 7 6.9 6.9 0 0 1 7 0Zm0 0v.7V0ZM0 7h.6H0Zm7 6.8v-.6.6ZM13.7 7h-.6.6ZM9.1 1.7c-.7-.3-1.4-.4-2.2-.4a5.6 5.6 0 0 0-4 1.6 5.6 5.6 0 0 0-1.6 4 5.6 5.6 0 0 0 1.6 4 5.6 5.6 0 0 0 4 1.7 5.6 5.6 0 0 0 4-1.7 5.6 5.6 0 0 0 1.7-4 5.6 5.6 0 0 0-1.7-4c-.5-.5-1.1-.9-1.8-1.2Z" clip-rule="evenodd"/><path fill="#000" fill-rule="evenodd" d="M7 2.9a.8.8 0 1 1 0 1.5A.8.8 0 0 1 7 3ZM5.8 5.7c0-.4.3-.6.6-.6h.7c.3 0 .6.2.6.6v3.7h.5a.6.6 0 0 1 0 1.3H6a.6.6 0 0 1 0-1.3h.4v-3a.6.6 0 0 1-.6-.7Z" clip-rule="evenodd"/></svg>
+                        You have successfully duplicated the Paints UNDO Space 🎉</h2>
+                        <p class="custom-color">There's only one step left before you can properly play with this demo: <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}/settings" style="text-decoration: underline" target="_blank">attribute a <b>T4-small or A10G-small GPU</b> to it (via the Settings tab)</a> and run the training below.
+                        You will be billed by the minute from when you activate the GPU until when it is turned off.</p> 
+                        <p class="actions custom-color">
+                            <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}/settings">🔥 &nbsp; Set recommended GPU</a>
+                        </p>
+                        </div>
+                ''', elem_id="warning-setgpu")
     gr.Markdown('# Paints-Undo')
 
     with gr.Accordion(label='Step 1: Upload Image and Generate Prompt', open=True):