Update app.py

app.py

@@ -114,11 +114,48 @@ def plex(qr_code_value, text, neg_prompt, modil, one, two, three):
     image = pipe([prompt]*2,images,num_inference_steps=one,generator=generator,strength=two,negative_prompt=[neg_prompt]*2,controlnet_conditioning_scale=three,)
     for i, imge in enumerate(image["images"]):
         apol.append(imge)
+
+        img = load_image(imge)
+        img.save('./image.png', 'PNG')
+        img = img.resize((512, 512))
+        img = img.convert("RGBA")
+        img.save('./image.png', 'PNG')
+
+        iog = load_image(original)
+        iog.save('./imoge.png', 'PNG')
+        iog = iog.resize((512, 512))
+        iog = iog.convert("RGBA")
+        iog.save('./imoge.png', 'PNG')
+        doto = iog.getdata()
+    
+        new_data = []
+        for item in doto:
+            if item[0] in list(range(200, 256)):
+                new_data.append((255, 255, 255, 0))
+            else:
+                new_data.append(item)
+        iog.putdata(new_data)
+        iog.save('./image.png', 'PNG')
+
+        pixel_data1 = list(iog.getdata())
+        pixel_data2 = list(img.getdata())
+
+        new_pixel_data = [pixel if pixel[3] > 0 else pixel_data2[i] for i, pixel in enumerate(pixel_data1)]
+        if new_image:
+            new_imoge = Image.new("RGBA", img.size)
+            new_imoge.putdata(new_pixel_data)
+            new_imoge.save('./new_imoge.png', 'PNG')
+            apol.append(new_imoge)
+        else:
+            new_image = Image.new("RGBA", img.size)
+            new_image.putdata(new_pixel_data)
+            new_image.save('./new_image.png', 'PNG')
+            apol.append(new_image)
     apol.append(original)
     apol.append(cannyimage)
     apol.append(pannyimage)
     return apol
 
-iface = gr.Interface(fn=plex, inputs=[gr.Textbox(label="QR Code URL"),gr.Textbox(label="prompt"),gr.Textbox(label="neg prompt"),gr.Dropdown(choices=models, label="some sd models", value=models[0], type="value"), gr.Slider(label="num inference steps", minimum=1, step=1, maximum=5, value=5), gr.Slider(label="prompt strength", minimum=0.01, step=0.01, maximum=0.99, value=0.20), gr.Slider(label="controlnet scale", minimum=0.01, step=0.01, maximum=0.99, value=0.80)], outputs=gr.Gallery(label="out", columns=1))
+iface = gr.Interface(fn=plex, inputs=[gr.Textbox(label="QR Code URL"),gr.Textbox(label="prompt"),gr.Textbox(label="neg prompt"),gr.Dropdown(choices=models, label="some sd models", value=models[0], type="value"), gr.Slider(label="num inference steps", minimum=1, step=1, maximum=5, value=5), gr.Slider(label="prompt strength", minimum=0.01, step=0.01, maximum=0.99, value=0.20), gr.Slider(label="controlnet scale", minimum=0.01, step=0.01, maximum=0.99, value=0.80)], outputs=gr.Gallery(label="out", columns=2),description="Running on cpu, very slow! by JoPmt.")
 iface.queue(max_size=1)
 iface.launch(max_threads=1)