Update app.py

app.py

@@ -12,28 +12,28 @@ model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"
 
 #os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py)
 
-# url = "http://127.0.0.1:7861"
+url = "http://0.0.0.0:7860"
 
-# payload = {
-#     "prompt": "apple is beautiful",
-#     "steps": 5
-# }
+payload = {
+    "prompt": "apple is beautiful",
+    "steps": 5
+}
 
-# response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
+response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
 
-# r = response.json()
+r = response.json()
 
-# for i in r['images']:
-#     image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0])))
+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)
+    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)
+    pnginfo = PngImagePlugin.PngInfo()
+    pnginfo.add_text("parameters", response2.json().get("info"))
+    image.save('output.png', pnginfo=pnginfo)
 
 import psutil