Adding details for image watermarking

app.py

@@ -10,23 +10,27 @@ li {
     font-size: 110%;
 }
 
-.container {
-    height: initial;
+body {
+    overflow: scroll;
 }
 
 video {
     max-height: 400px;
 }
 
+.container {
+    height: initial;
+}
+
+
 .image-container {
     width: 200px; 
-    max-height: 300px;
+    max-height: auto;
     margin: auto;
 }
 
 img {
-    width: auto;
-    height: auto;
+    max-height: 400px;
 }
 """
 
@@ -52,6 +56,7 @@ w3 = os.path.join(os.path.dirname(__file__), "files/forest_qr_watermarking.png")
 w4 = os.path.join(os.path.dirname(__file__), "files/cheetah1.jpg")
 w5 = os.path.join(os.path.dirname(__file__), "files/frog.jpg")
 w6 = os.path.join(os.path.dirname(__file__), "files/Human_generated.png")
+w7 = os.path.join(os.path.dirname(__file__), "files/hf-logo_transpng.png")
 
 def generate_image(original_image, watermark):
     return gr.Image(original_image, watermark=watermark)
@@ -59,7 +64,7 @@ def generate_image(original_image, watermark):
 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:
     with gr.Row():
         with gr.Column():
@@ -79,27 +84,27 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 gr.Image('files/watermark_example.png', visible=False)
             with gr.Column():
-                #gr.Image('files/watermark_example.png', show_label=False, show_download_button=False, elem_id='example', container=False, interactive=False)
-                #gr.Markdown('**Image Watermark Code:**')
-                #gr.Code('import gradio as gr\n\nwatermarked_image = gr.Image(original_image_file, watermark=watermark_file)', lines=3)
+                gr.Image('files/watermark_example.png', show_label=False, show_download_button=False, elem_id='example', container=False, interactive=False)
+                gr.Markdown('**Image Watermark Code:**')
+                gr.Code('import gradio as gr\n\nwatermarked_image = gr.Image(original_image_file, watermark=watermark_file)', lines=3)
                 gr.Markdown('**Video Watermark Code:**')
                 gr.Code('import gradio as gr\n\nwatermarked_video = gr.Video(original_video_file, watermark=watermark_file)', lines=3)
             with gr.Column():
                 gr.Image('files/watermark_example.png', visible=False)
-    #with gr.Row():
-    #    with gr.Column():
-    #        gr.Markdown("**Inputs**: Image and watermark file")
-    #    with gr.Column():
-    #        gr.Markdown("**Output**: Watermarked image")
-    #gr.Interface(generate_image, [gr.Image(), gr.Image(type='filepath', image_mode=None)], gr.Image(),
-    #                        examples=[[c, w1], [d, w2], [w4, w5], [c, w6]])
-    with gr.Row():
+    with gr.Tab("Image Watermarking"):
+        with gr.Column():
+            gr.Markdown("**Inputs**: Image and watermark file")
+        with gr.Column():
+            gr.Markdown("**Output**: Watermarked image")
+        gr.Interface(generate_image, [gr.Image(), gr.Image(type='filepath', image_mode=None)], gr.Image(),
+                 examples=[[d, w7], [w4, w5], [c, w6],])
+    with gr.Tab("Video Watermarking"):
         with gr.Column():
             gr.Markdown("**Inputs**: Video and watermark file")
         with gr.Column():
             gr.Markdown("**Output**: Watermarked video")
-    gr.Interface(generate_video, [gr.Video(), gr.Image(type='filepath', image_mode=None)], gr.Video(),
-                            examples=[[a, w1], [b, w2], [a, w3], [b, w4]])
-
+        gr.Interface(generate_video, [gr.Video(), gr.Image(type='filepath', image_mode=None)], gr.Video(),
+                 examples=[[a, w1], [b, w2], [a, w3], [b, w4]])
+    
 if __name__ == "__main__":
     demo.launch()