Update SDK version to 5.1.0, refactor image handling in app.py and added examples folder

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: gray
 colorTo: pink
 sdk: gradio
-sdk_version: 4.42.0
+sdk_version: 5.1.0
 app_file: app.py
 pinned: true
 license: mit

app.py

@@ -251,14 +251,13 @@ def hex_to_rgba(hex_color):
 
 def apply_bg_image(image, background_file=None, background_color=None, bg_type="Color"):
     try:
-        image_data = image.read()
-        input_image = Image.open(io.BytesIO(image_data))
+        input_image = load_img(image, output_type="pil")
         origin = input_image.copy()
         
         color_profile = input_image.info.get("icc_profile")
 
         if background_file is not None:
-            background_image = Image.open(io.BytesIO(background_file.read()))
+            background_image = load_img(background_file, output_type="pil")
         else:
             background_image = None
 
@@ -275,24 +274,20 @@ def apply_bg_image(image, background_file=None, background_color=None, bg_type="
                 output_frame = apply_background(frame, background_image)
                 frames.append(output_frame)
 
-            output_image = io.BytesIO()
-            frames[0].save(
-                output_image,
+            output_image = frames[0]
+            output_image.save(
+                "output.gif",
                 format="GIF",
                 save_all=True,
                 append_images=frames[1:],
                 loop=0,
                 icc_profile=color_profile,
             )
-            output_image_base64 = base64.b64encode(output_image.getvalue()).decode("utf-8")
         else:
             output_image = apply_background(input_image, background_image)
-            buffered = io.BytesIO()
-            output_image.save(buffered, format="PNG", optimize=True, icc_profile=color_profile)
-            output_image_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+            output_image.save("output.png", format="PNG", optimize=True, icc_profile=color_profile)
 
-        output_image_data = base64.b64decode(output_image_base64)
-        return (Image.open(io.BytesIO(output_image_data)), origin)
+        return (output_image, origin)
     except Exception as e:
         return str(e)
 
@@ -311,13 +306,13 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
         examples = gr.Examples(
             [
                 load_img(
-                    "https://images.rawpixel.com/image_800/cHJpdmF0ZS9sci9pbWFnZXMvd2Vic2l0ZS8yMDIzLTA4L3Jhd3BpeGVsX29mZmljZV8yX3Bob3RvX29mX2FfbGlvbl9pc29sYXRlZF9vbl9jb2xvcl9iYWNrZ3JvdW5kXzJhNzgwMjM1LWRlYTgtNDMyOS04OWVjLTY3ZWMwNjcxZDhiMV8xLmpwZw.jpg",
+                    "./examples/lion.jpg",
                     output_type="pil",
                 )
             ],
             inputs=image_input,
-            fn=remove_bg_fn,
             outputs=slider,
+            fn=remove_bg_fn,
             cache_examples=True,
             cache_mode="eager",
         )
@@ -376,17 +371,12 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
         
         examples = gr.Examples(
             [
-                ["https://pngimg.com/d/mario_PNG125.png", None, "#0cfa38", "Color"],
-                [
-                    "https://pngimg.com/d/mario_PNG125.png",
-                    "https://cdn.photoroom.com/v2/image-cache?path=gs://background-7ef44.appspot.com/backgrounds_v3/black/47_-_black.jpg",
-                    None,
-                    "Image",
-                ],
+                ["./examples/mario.png", None, "#0cfa38", "Color"],
+                ["./examples/mario.png", "./examples/bg-image.jpg", None, "Image"],
             ],
             inputs=[image_input, bg_image, color_picker, bg_type],
-            fn=apply_bg_image,
             outputs=slider,
+            fn=apply_bg_image,
             cache_examples=True,
             cache_mode="eager",
         )
@@ -494,18 +484,18 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
         examples = gr.Examples(
             [
                 [
-                    "https://www.w3schools.com/html/mov_bbb.mp4",
+                    "./examples/videoplayback.mp4",
                     "Video",
                     None,
-                    "https://www.w3schools.com/howto/rain.mp4",
+                    "./examples/mov_bbb.mp4",
                 ],
                 [
-                    "https://www.w3schools.com/html/mov_bbb.mp4",
+                    "./examples/videoplayback.mp4",
                     "Image",
-                    "https://cdn.photoroom.com/v2/image-cache?path=gs://background-7ef44.appspot.com/backgrounds_v3/black/47_-_black.jpg",
+                    "./examples/bg-image.jpg",
                     None,
                 ],
-                ["https://www.w3schools.com/html/mov_bbb.mp4", "Color", None, None],
+                ["./examples/videoplayback.mp4", "Color", None, None],
             ],
             inputs=[in_video, bg_type, bg_image, bg_video],
             outputs=[stream_image, out_video, time_textbox],

requirements.txt

@@ -11,7 +11,7 @@ typing
 scikit-image
 huggingface_hub
 transformers>=4.39.1
-gradio==4.42.0
+gradio==5.1.0
 schedule
 loadimg>=0.1.1
 moviepy==1.0.3