refactoring

app.py

@@ -47,10 +47,10 @@ def sanitize_prompt(prompt):
   return sanitized_prompt
 
 @spaces.GPU(duration=120)
-def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
+def process_examples(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
     # I'm not sure when this happen
-    progress(0, desc="start-process-images")
-    print("start-process-images")
+    progress(0, desc="start-process-examples")
+    print("start-process-examples")
     if not isinstance(image, dict):
         if image2 == None:
              print("empty mask")
@@ -88,7 +88,7 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
         output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,width=width,height=height,
                       guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
 
-        return output.images[0]
+        return output.examples[0]
 
 
     output = process_inpaint(image["background"],mask,prompt,inpaint_model,strength,seed)
@@ -111,11 +111,28 @@ css="""
     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;
+}
 """
 
 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("tools.html"))
     with gr.Row():
                 with gr.Column():
                     image = gr.ImageEditor(height=800,sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#fff"], color_mode="fixed"))
@@ -141,12 +158,12 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                     
             
 
-    btn.click(fn=process_images, inputs=[image, image_mask,prompt,inpaint_model,strength,seed], outputs =[image_out,mask_out], api_name='infer')
+    btn.click(fn=process_examples, inputs=[image, image_mask,prompt,inpaint_model,strength,seed], outputs =[image_out,mask_out], api_name='infer')
     gr.Examples(
                examples=[
-                    ["images/00547245_99.jpg", "images/00547245_99_mask.jpg","a beautiful girl,eyes closed",0.8,"images/00547245.jpg"],
-                    ["images/00538245_paint.jpg", "images/00538245_mask.jpg","a beautiful girl,wearing t-shirt",0.7,"images/00538245.jpg"],
-                    ["images/00207245_18.jpg", "images/00207245_18_mask.jpg","a beautiful girl,mouth opened",0.2,"images/00207245.jpg"]
+                    ["examples/00547245_99.jpg", "examples/00547245_99_mask.jpg","a beautiful girl,eyes closed",0.8,"examples/00547245.jpg"],
+                    ["examples/00538245_paint.jpg", "examples/00538245_mask.jpg","a beautiful girl,wearing t-shirt",0.7,"examples/00538245.jpg"],
+                    ["examples/00207245_18.jpg", "examples/00207245_18_mask.jpg","a beautiful girl,mouth opened",0.2,"examples/00207245.jpg"]
                          ]
 ,
                 #fn=example_out,
@@ -155,9 +172,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                 #cache_examples=False,
     )
     gr.HTML(
-        """
-            
-        """
+       gr.HTML(read_file("demo_footer.html"))
     )
 
     demo.launch()

demo_footer.html

@@ -0,0 +1,3 @@
+<div>
+    <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>
+</div>

demo_header.html

@@ -2,12 +2,13 @@
     <h1>
         Flux.1-schnell mask-inpaint
     </h1>
-    <p>Making face images for <a href="https://huggingface.co/spaces/Akjava/godot-huggingface-chain">AI Diagram Chat with Voice/Face Character</a></p>
-    <p>
-        
-        See examples.
-        The image requires guide painting.(To generate an image with closed eyes, fill the eye area with skin color.)
-        You can draw mask or upload mask.BE CAREFUL UPLOAEDED MASK(If the uploaded mask exist,ignore your painted mask)
-    </p>
-    <p>Hint:step by step,reduce strength.blur guide-paint and mask</p>
+    <div class="grid-container">
+        <img src="https://akjava.github.io/AIDiagramChatWithVoice-FaceCharacter/webp/128/00206245_09_00003200.webp" alt="Flux.1-schnell-WebP3Frame-TalkingAnimation" class="image">
+        <p class="text">
+        See examples.<br>
+        The image requires guide painting.(To generate an image with closed eyes, fill the eye area with skin color.)<br>
+        You can draw mask or upload mask.BE CAREFUL UPLOAEDED MASK(If the uploaded mask exist,ignore your painted mask)<br>
+        Hint:step by step,reduce strength.blur guide-paint and mask
+        </p>
+    </div>
 </div>

tools.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>