import os
import psutil
import json
import requests
import io
import base64
from time import sleep
import gradio as gr
from PIL import Image, PngImagePlugin


model = "/home/user/app/sd/stable-diffusion-webui/models/Stable-diffusion/"

os.system("/usr/bin/python3 /home/user/app/sd/stable-diffusion-webui/webui.py")

url = "http://0.0.0.0:7860"

payload = {
    "prompt": "apple is beautiful",
    "steps": 5
}


sleep(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)

##############################################################################################

# 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, 

# inputs=["text", "text", "text"], outputs=["text", "text", "text"])


# iface.launch()