init

.gitignore

@@ -0,0 +1,6 @@
+__pycache__
+*.egg-info
+.ipynb_checkpoints
+dist
+.gradio
+build

app.py

@@ -0,0 +1,292 @@
+import spaces
+import gradio as gr
+import re
+from PIL import Image,ImageEnhance
+
+import os
+
+import numpy as np
+import hashlib
+import io
+import cv2
+import time
+
+def clear_old_files(dir,passed_time):
+    files = os.listdir(dir)
+    current_time = time.time()
+    for file in files:
+         file_path = os.path.join(dir,file)
+         
+         ctime = os.stat(file_path).st_ctime
+         diff = current_time - ctime
+         print(f"ctime={ctime},current_time={current_time},passed_time={passed_time},diff={diff}")
+         if diff > passed_time:
+              os.remove(file_path)
+              print(f"file removed {file_path}")#TODO remove later
+
+
+def get_image_id(image):
+    buffer = io.BytesIO()
+    image.save(buffer, format='PNG')
+    hash_object = hashlib.sha256(buffer.getvalue())
+    hex_dig = hash_object.hexdigest()
+    unique_id = hex_dig[:32]
+    return unique_id
+
+dir_name ="images"
+passed_time = 60*3
+
+
+def process_create_webp(images,duration=100, loop=0,quality=85):
+    frames = []
+    for image_file in images:
+        frames.append(image_file)
+    
+    output_buffer = io.BytesIO()
+    frames[0].save(output_buffer, 
+                   save_all=True, 
+                   append_images=frames[1:], 
+                   duration=duration, 
+                   loop=loop, 
+                   format='WebP',
+                   quality=quality
+                   )
+    
+    return output_buffer.getvalue()
+
+def process_create_apng(images,duration=100, disposal=1,blend=0,loop=0):
+    frames = []
+    for image_file in images:
+        frames.append(image_file)
+    
+    output_buffer = io.BytesIO()
+    frames[0].save(output_buffer, 
+                   save_all=True, 
+                   append_images=frames[1:], 
+                   #duration=duration, disposal=0,blend=1,
+                   duration=duration, disposal=disposal,blend=blend,
+                   loop=loop, 
+                   format='png')
+    
+    return output_buffer.getvalue()
+
+def process_create_gif(images,duration=100, quantize=False,disposal=1,blend=0,loop=0):
+    frames = []
+    for image_file in images:
+        if quantize:
+            image_file = image_file.quantize(colors=256, method=2)
+            #image_file = image_file.quantize()
+        
+
+        frames.append(image_file)
+    
+    output_buffer = io.BytesIO()
+    frames[0].save(output_buffer, 
+                   save_all=True, 
+                   append_images=frames[1:], 
+                   #duration=duration, disposal=0,blend=1,
+                   duration=duration, disposal=disposal,blend=blend,
+                   loop=loop, 
+                   format='gif',
+                   optimize=False
+                   #dither=Image.FLOYDSTEINBERG
+                   )
+    
+    return output_buffer.getvalue()
+
+def convert_webp_to_images(webp_path):
+     # 各フレームをリストに保存
+  frames = []
+  durations = []
+  Image.init()
+  image = Image.open(webp_path)
+  for i in range(image.n_frames):
+    image.seek(i)
+    frame = image.copy()
+    frames.append(frame)
+    durations.append(frame.info["duration"])
+  return frames,durations
+
+def buffer_to_id(buffer_value):
+    hash_object = hashlib.sha256(buffer_value)
+    hex_dig = hash_object.hexdigest()
+    unique_id = hex_dig[:32]
+    return unique_id
+
+def process_images(input_path,same_size=False,image_width=128,file_format="webp",webp_quality=85):
+    if input_path == None:
+         raise gr.Error("need image")
+    
+    # cache control
+    if not os.path.exists(dir_name):
+        os.mkdir(dir_name)
+    clear_old_files(dir_name,passed_time)
+
+
+
+    images =[]
+    frames,durations = convert_webp_to_images(input_path)
+    new_frames = []
+
+
+    new_width,new_height = frames[0].size
+    if not same_size and image_width!=new_width:
+       
+        ratio = new_width/new_height
+        new_height = int(image_width*ratio)
+        new_width = image_width
+        #print(f"new size {new_width} x {new_height}")
+        for frame in frames:
+            new_frame = frame.resize((new_width,new_height))# extremly slow Image.LANCZOS
+            new_frames.append(new_frame)
+    else:
+        new_frames = frames
+
+    if file_format == "webp":
+        webp_buffer = process_create_webp(new_frames,durations,webp_quality)
+        webp_id = buffer_to_id(webp_buffer)
+        webp_path = os.path.join(dir_name,f"{webp_id}.webp")
+        with open(webp_path, 'wb') as f:
+            f.write(webp_buffer)
+        
+        images.append((webp_path,"webp"))
+
+    elif file_format == "apng":
+        apng_buffer = process_create_apng(new_frames,durations)
+        apng_id = buffer_to_id(apng_buffer)
+        apng_path = os.path.join(dir_name,f"{apng_id}.apng")
+        with open(apng_path, 'wb') as f:
+            f.write(apng_buffer)
+        
+        images.append((apng_path,"apng"))
+
+    elif file_format == "gif":
+        gif_buffer = process_create_gif(new_frames,durations,False)
+        gif_id = buffer_to_id(gif_buffer)
+        gif_path = os.path.join(dir_name,f"{gif_id}.gif")
+        with open(gif_path, 'wb') as f:
+            f.write(gif_buffer)
+        
+        images.append((gif_path,"gif"))
+    else:
+        for i ,frame in enumerate(new_frames):
+            images.append((frame,f"index {i}"))
+
+    return images
+    
+
+def read_file(path: str) -> str:
+    with open(path, 'r', encoding='utf-8') as f:
+        content = f.read()
+
+    return content
+
+def test_echo(param):
+     return param
+
+def samesize_changed(check,slider):
+    return gr.Slider(label="Image Width",
+                            minimum=8,
+                            maximum=2048,
+                            step=1,
+                            value=slider,
+                            interactive=not check)
+
+css="""
+#col-left {
+    margin: 0 auto;
+    max-width: 640px;
+}
+#col-right {
+    margin: 0 auto;
+    max-width: 640px;
+}
+.grid-container {
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  gap:10px
+}
+
+.image {
+  width: 128px; 
+  height: 128px; 
+  object-fit: cover; 
+}
+}
+}
+
+.text {
+  font-size: 16px;
+}
+"""
+js = """
+function(path){
+console.log(path)
+return path
+}
+"""
+
+with gr.Blocks(css=css, elem_id="demo-container") as demo:
+    with gr.Column():
+        gr.HTML(read_file("demo_header.html"))
+        gr.HTML(read_file("links.html"))
+    with gr.Row():
+                with gr.Column():
+                    image = gr.Image(sources=[],image_mode='RGB',  elem_id="Image", type="filepath", label="Preview WebP")
+                    file = gr.File(label="WebP Upload")
+                    #file.select(fn=test_echo,inputs=[file])
+                    
+                    file.upload(fn=test_echo,inputs=[file],outputs=[image])
+                    file.clear(fn=test_echo,inputs=[file],outputs=[image])
+
+                    btn = gr.Button("Convert", elem_id="run_button",variant="primary")
+                    # type choice
+                    # size slider
+
+                    file_format=gr.Dropdown(
+                    ["webp", "apng", "gif","images"], label="Animation Format", info="Convert to  Animattion" )
+                    
+                    same_size = gr.Checkbox(label="Same Size",value=False)
+                           
+                    image_width = gr.Slider(
+                            label="Image Width",info = "new animation size",
+                            minimum=8,
+                            maximum=2048,
+                            step=1,
+                            value=128,
+                            interactive=True)
+                    same_size.change(fn=samesize_changed,inputs=[same_size,image_width],outputs=[image_width])
+                    
+                    with gr.Accordion(label="Advanced Settings", open=False):
+                        with gr.Row( equal_height=True):
+                           webp_quality = gr.Slider(
+                            label="WebP Quality",info = "this change file size",
+                            minimum=0,
+                            maximum=100,
+                            step=1,
+                            value=85,
+                            interactive=True)
+                
+                with gr.Column():
+                    image_out = gr.Gallery(height=800,label="Output", elem_id="output-img",format="webp", columns=[4],rows=[2])
+
+                    
+            
+
+    btn.click(fn=process_images, inputs=[file,same_size,image_width,file_format,webp_quality], outputs =[image_out], api_name='infer')
+    gr.Examples(
+               examples=[
+                   ["examples/1024.webp","examples/1024.webp"],
+                   #["images/00346245_00006200.jpg", "images/00346245_00003200.jpg","images/00346245_mask.jpg",10,0,"images/00346245_mixed.jpg"]
+                  
+                         ]
+                         ,
+                inputs=[image,file]#example not fire event
+    )
+    gr.HTML(
+        """
+        <P> Images are generated with <a href="https://huggingface.co/black-forest-labs/FLUX.1-schnell">FLUX.1-schnell</a> and licensed under <a href="http://www.apache.org/licenses/LICENSE-2.0">the Apache 2.0 License</a>
+        """)
+
+    demo.launch(allowed_paths=["examples/anime.webp"])#

demo_header.html

@@ -0,0 +1,18 @@
+<div style="text-align: center;">
+    <h1>
+        WebP Resize and Convert
+    </h1>
+    
+    <div class="grid-container">
+        <img src="/gradio_api/file=examples/anime.webp" alt="Flux.1-schnell-WebP3Frame-TalkingAnimation" class="image">
+        <p class="text">
+            Hint:Aspect ratio will always keep<br>
+            I'm not familiar with apng/gif,maybe quality is low<br>
+        </p>
+    </div>
+
+    <p>
+       
+    </p>
+    
+</div>

links.html

@@ -0,0 +1,12 @@
+<div style="text-align: center;">
+    <p><a href="https://huggingface.co/spaces/Akjava/godot-huggingface-chain">AI Diagram Chat with Voice/Face Character Making Tools</a></p>
+    <p><a href="https://huggingface.co/spaces/Akjava/WebPTalkHead">[WebP-3F-TH]</a> 
+        <a href="https://huggingface.co/spaces/Akjava/flux1-schnell-mask-inpaint">[Flux1-Inpaint(GPU)]</a> 
+        <a href="https://huggingface.co/spaces/Akjava/OpenCVInpaintCPU">[OpenCV-Inapint]</a>
+        <a href="https://huggingface.co/spaces/Akjava/Simple-Whitebalance-Image">[Whitebalance]</a>
+        <a href="https://huggingface.co/spaces/Akjava/Simple-Mask-Paste-Image">[Paste Image]</a>
+        <a href=" https://huggingface.co/spaces/Akjava/WebP-Resize-Convert">[WebP Resize Convert]</a></p>
+
+       
+        
+</div>

requirements.txt

@@ -0,0 +1,3 @@
+spaces
+numpy
+opencv-contrib-python