import gradio as gr
import os


css = """

p {
    font-size: 120%;
}

.container {
    height: initial;
}

video {
max-height: 400px;
}
.image-container {
width: 200px; max-height: 300px;
margin: auto;
}

img {
width: auto;
height: auto;
}
"""

a = os.path.join(os.path.dirname(__file__), "files/barkley_balloon.mp4")
b = os.path.join(os.path.dirname(__file__), "files/eiffel_tower.mp4")
w1 = os.path.join(os.path.dirname(__file__), "files/AI_generated.png")
w2 = os.path.join(os.path.dirname(__file__), "files/hf-logo.png")
w3 = os.path.join(os.path.dirname(__file__), "files/forest_qr_watermarking.png")


def generate_video(original_video, watermark):
    return gr.Video(original_video, watermark=watermark)

# Use RGBA image mode to preserve transparency for png images.
with gr.Blocks(css=css) as demo:
    gr.Markdown("# Watermarking with Gradio: Example")
    gr.Markdown("Watermarks can be **visible** or **invisible**.")
    gr.Markdown("""They can provide information directly, or provide a link for more information.  
                   Visible watermarks can help to disclose when content is AI-generated, so viewers know it's synthetic.  
                   Invisible watermarks can help to mark content as authentic.""")

    gr.Markdown()
    gr.Markdown("""For more information on watermarking -- what watermarking is, why it's important, and the tools available on Hugging Face --  
    please check out [our blogpost on AI watermarking](https://huggingface.co/blog/watermarking).""")
    gr.Interface(generate_video, [gr.Video(), gr.Image(type='filepath', image_mode="RGBA")], gr.Video(),
                        examples=[[a, w1], [b, w2], [a, w3]])

if __name__ == "__main__":
    demo.launch()