Update app.py

app.py

@@ -11,41 +11,41 @@ from PIL import Image, PngImagePlugin
 
 model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"
 
-#os.system(python /home/user/app/sd/stable-diffusion-webui/webui.py)
+os.system(/use/bin/python3 /home/user/app/sd/stable-diffusion-webui/webui.py)
 
-# url = "http://0.0.0.0:7860"
+url = "http://0.0.0.0:7860"
 
-# payload = {
-#     "prompt": "apple is beautiful",
-#     "steps": 5
-# }
+payload = {
+    "prompt": "apple is beautiful",
+    "steps": 5
+}
 
 
-# sleep(5)
-# response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
+sleep(5)
+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)
 
 ##############################################################################################
 
-def current_working_directoy(cwd, ldir, python_path):
-    return os.getcwd(), psutil.net_connections(), os.listdir('/usr/bin')
+# def current_working_directoy(cwd, ldir, python_path):
+#     return os.getcwd(), psutil.net_connections(), os.listdir('/usr/bin')
 
-iface = gr.Interface(fn=current_working_directoy, 
+# iface = gr.Interface(fn=current_working_directoy, 
 
-inputs=["text", "text", "text"], outputs=["text", "text", "text"])
+# inputs=["text", "text", "text"], outputs=["text", "text", "text"])
 
 
-iface.launch()
+# iface.launch()