Update app.py

app.py

@@ -3,4 +3,36 @@ import os
 
 model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"
 
-os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py $share --api --disable-safe-unpickle --enable-insecure-extension-access --no-download-sd-model --no-half-vae  --ckpt "$model" --xformers $auth --disable-console-progressbars)
+os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py $share --api --disable-safe-unpickle --enable-insecure-extension-access --no-download-sd-model --no-half-vae  --ckpt "$model" --xformers $auth --disable-console-progressbars)
+
+
+
+
+import json
+import requests
+import io
+import base64
+from PIL import Image, PngImagePlugin
+
+url = "http://127.0.0.1:7861"
+
+payload = {
+    "prompt": "apple is beautiful",
+    "steps": 5
+}
+
+response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
+
+r = response.json()
+
+for i in r['images']:
+    image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0])))
+
+    png_payload = {
+        "image": "data:image/png;base64," + i
+    }
+    response2 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload)
+
+    pnginfo = PngImagePlugin.PngInfo()
+    pnginfo.add_text("parameters", response2.json().get("info"))
+    image.save('output.png', pnginfo=pnginfo)