Update index.html

app.py

@@ -126,7 +126,22 @@ def run_outpaint(
     state,
 ):
     base64_str = "base64"
-    return sel_buffer_str
+    data = base64.b64decode(str(sel_buffer_str))
+    pil = Image.open(io.BytesIO(data))
+    sel_buffer = np.array(pil)
+    sel_buffer[:, :, 3]=255
+    sel_buffer[:, :, 0]=255
+    out_pil = Image.fromarray(sel_buffer)
+    out_buffer = io.BytesIO()
+    out_pil.save(out_buffer, format="PNG")
+    out_buffer.seek(0)
+    base64_bytes = base64.b64encode(out_buffer.read())
+    base64_str = base64_bytes.decode("ascii")
+    return (
+        gr.update(label=str(state + 1), value=base64_str,),
+        gr.update(label="Prompt"),
+        state + 1,
+    )
     if True:
         text2img, inpaint = get_model()
         if enable_safety:

index.html

@@ -152,7 +152,7 @@ async def transfer_func(event):
     base.read_selection_from_buffer()
     sel_buffer=base.sel_buffer
     sel_buffer_str=base.numpy_to_base64(sel_buffer)
-    parent.postMessage(["transfer",sel_buffer_str],"*")
+    parent.postMessage(to_js(["transfer",str(sel_buffer_str)]),"*")
 
 async def upload_func(event):
     base=base_lst[0]