Update app.py

app.py

@@ -1,38 +1,49 @@
 import os
 
+import json
+import requests
+import io
+import base64
+import gradio as gr
+
+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 $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)
 
+# 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)
 
-import json
-import requests
-import io
-import base64
-from PIL import Image, PngImagePlugin
+# 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)
 
-url = "http://127.0.0.1:7861"
+import psutil
 
-payload = {
-    "prompt": "apple is beautiful",
-    "steps": 5
-}
 
-response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
+def current_working_directoy(cwd, ldir):
+    return os.getcwd(), psutil.net_connections()
 
-r = response.json()
+iface = gr.Interface(fn=current_working_directoy, 
 
-for i in r['images']:
-    image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0])))
+inputs=["text", "text"], outputs=["text", "text"])
 
-    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)
+iface.launch()