initial

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -0,0 +1,81 @@
+import spaces
+import gradio as gr
+import subprocess
+import re
+from PIL import Image
+
+import opencvinpaint
+
+def sanitize_prompt(prompt):
+  # Allow only alphanumeric characters, spaces, and basic punctuation
+  allowed_chars = re.compile(r"[^a-zA-Z0-9\s.,!?-]")
+  sanitized_prompt = allowed_chars.sub("", prompt)
+  return sanitized_prompt
+
+#@spaces.GPU(duration=120)
+def process_images(image1, image2,inpaint_radius,blur_radius,edge_expand):
+    if image1!=None:
+        pass
+        #image1.save("input.jpg","JPEG")
+    if image2!= None:
+        
+        pass
+        #image2.save("mask.jpg","JPEG")
+    if image1!=None and image2!=None:
+        
+        output = opencvinpaint.process_image(image1,image2,inpaint_radius,blur_radius,edge_expand)
+        
+        return output
+    
+    return None
+
+# code from https://huggingface.co/spaces/diffusers/stable-diffusion-xl-inpainting/blob/main/app.py
+def read_file(file_path: str) -> str:
+    """read the text of target file
+    """
+    with open(file_path, 'r', encoding='utf-8') as f:
+        content = f.read()
+
+    return content
+
+#css=css,
+demo_blocks = gr.Blocks( elem_id="demo-container")
+with demo_blocks as demo:
+    gr.HTML(read_file("demo_header.html"))
+    with gr.Row():
+                with gr.Column():
+                    image = gr.Image(sources=['upload','clipboard'],  elem_id="image_upload", type="pil", label="Upload",height=400)
+                    with gr.Row(elem_id="prompt-container",  equal_height=True):
+                        with gr.Row():
+                            btn = gr.Button("Inpaint!", elem_id="run_button")
+                    image_mask = gr.Image(sources=['upload','clipboard'],  elem_id="mask_upload", type="pil", label="Mask_Upload",height=400)
+                    with gr.Accordion(label="Advanced Settings", open=False):
+                        with gr.Row( equal_height=True):
+                            inpaint_radius = gr.Number(value=3, minimum=1.0, maximum=20.0, step=1, label="Inpaint Radius")
+                            blur_radius = gr.Number(value=25, minimum=0.0, maximum=50.0, step=1, label="Blur Radius")
+                            edge_expand = gr.Number(value=8, minimum=0.0, maximum=20.0, step=1, label="Edge Expand")
+                            
+                with gr.Column():
+                    image_out = gr.Image(sources=[],label="Output", elem_id="output-img", height=400)
+                    with gr.Group(elem_id="share-btn-container", visible=False) as share_btn_container:
+                        pass
+            
+
+    btn.click(fn=process_images, inputs=[image, image_mask,inpaint_radius,blur_radius,edge_expand], outputs =image_out, api_name='run')
+    gr.Examples(
+               examples=[["examples/street.jpg", "examples/street_mask.jpg"]]
+,
+                #fn=predict,
+                inputs=[image,image_mask],
+                cache_examples=False,
+    )
+    gr.HTML(
+        """
+            <div style="text-align: center;">
+                <p>Inpaint Code <a href="https://github.com/opencv/opencv/blob/da3debda6d233af90e421e95700c63fc08b83b75/samples/python/inpaint.py" style="text-decoration: underline;" target="_blank">OpenCV inpaint example</a> - Gradio Demo by 🤗 Hugging Face
+                </p>
+            </div>
+        """
+    )
+
+demo_blocks.queue(max_size=25).launch(share=True)

demo_header.html

@@ -0,0 +1,6 @@
+<div style="text-align: center;">
+    <h1>
+        OpenCV Inpaint(Classic Style Inapint) CPU
+    </h1>
+    <p>This is old style inpaint.but background image effects new-inapint app.for smooth fill</p>
+</div>

opencvinpaint.py

@@ -0,0 +1,95 @@
+import os
+import sys
+
+import cv2
+import numpy as np
+from PIL import Image
+
+
+debug = False
+
+def gray3d_to_2d(grayscale: np.ndarray) -> np.ndarray:
+    channel = grayscale.shape[2] if grayscale.ndim == 3 else 1 
+    if channel!=1:
+        text = f"grayscale shape = {grayscale.shape} channel = {channel} ndim = {grayscale.ndim} size = {grayscale.size}"
+        raise ValueError(f"color maybe rgb or rgba {text}")
+
+    if grayscale.ndim == 2:
+        return grayscale
+    return np.squeeze(grayscale)
+
+def pil_to_cv(image):
+    cv_image = np.array(image, dtype=np.uint8)
+    if cv_image.shape[2] == 3:  # カラー
+        cv_image = cv2.cvtColor(cv_image, cv2.COLOR_RGB2BGR)
+    elif cv_image.shape[2] == 4: #
+        cv_image = cv2.cvtColor(cv_image, cv2.COLOR_RGBA2BGR)
+    return cv_image
+
+def blend_rgb_images(image1: np.ndarray, image2: np.ndarray, mask: np.ndarray) -> np.ndarray:
+
+    if image1.shape != image2.shape or image1.shape[:2] != mask.shape:
+        raise ValueError("not same shape")
+
+    # 画像を float 型に変換
+    image1 = image1.astype(float)
+    image2 = image2.astype(float)
+
+    # mask to 3 chan 0 -1 value
+    alpha = cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR).astype(float) / 255.0
+
+    # calcurate blend
+    blended = (1 - alpha) * image1 + alpha * image2
+
+    return blended.astype(np.uint8)
+
+def process_image(image,mask_image,inpaint_radius,blur_radius,edge_expand):
+    #print(blur_radius,",",edge_expand)
+    cv_image = pil_to_cv(image)
+
+    cv_mask = pil_to_cv(mask_image)
+    cv_gray = cv2.cvtColor(cv_mask,cv2.COLOR_BGR2GRAY)
+
+    mask = gray3d_to_2d(cv_gray)
+    cv2.imwrite("_mask.jpg",mask)
+    cv2.imwrite("_image.jpg",cv_image)
+    img_inpainted = cv2.inpaint(cv_image, mask,inpaint_radius, cv2.INPAINT_TELEA)
+    if debug:
+        cv2.imwrite("close_eye_inpaint.jpg",img_inpainted)
+    
+    
+    ## blur
+    if blur_radius > 0:
+        if blur_radius%2==0:
+            blur_radius += 1
+        print(blur_radius)
+        blurred_image = cv2.GaussianBlur(img_inpainted, (blur_radius, blur_radius), 0) #should be odd
+        if debug:
+            cv2.imwrite("close_eye_inpaint_burred.jpg",blurred_image)
+    else:
+        blurred_image = img_inpainted
+
+    # expand edge and blur
+    kernel = np.ones((edge_expand, edge_expand), np.uint8)
+    extend_mask = cv2.dilate(mask, kernel, iterations=1)
+
+    if edge_expand > 0 and blur_radius > 0:
+        extend_burred_mask = cv2.GaussianBlur(extend_mask, (blur_radius, blur_radius), 0)
+    else:
+        extend_burred_mask = extend_mask
+
+    
+    img_inpainted = blend_rgb_images(img_inpainted,blurred_image,extend_burred_mask)
+
+    output_image = img_inpainted.copy()
+    
+    if output_image.shape[2] == 3:  # カラー
+        output_image = cv2.cvtColor(output_image, cv2.COLOR_BGR2RGB)
+    
+    return Image.fromarray(output_image)
+
+if __name__ == "__main__":
+    image = Image.open(sys.argv[1])
+    mask  = Image.open(sys.argv[2])
+    output = process_image(image,mask)
+    output.save(sys.argv[3])

requirements.txt

@@ -0,0 +1,4 @@
+numpy
+torch
+spaces
+opencv