Update app.py

app.py

@@ -1,16 +1,23 @@
 import streamlit as st
 import requests
 import json
-
+from PIL import Image
+import time
 
 ngrok_url = st.text_input("ngrok url")
 prompt_input = st.text_input("Prompt")
 
-payload = json.dumps({"text":prompt_input})
-headers = {
-  'Content-Type': 'text/plain'
-}
+if st.button('Generate Image'):
+    payload = json.dumps({"text":prompt_input})
+    headers = {
+      'Content-Type': 'text/plain'
+    }
 
-response = requests.request("POST", ngrok_url, headers=headers, data=payload)
+    response = requests.request("POST", ngrok_url, headers=headers, data=payload)
+    file_name = "static/"+prompt_input+time.strftime("%Y%m%d-%H%M%S")+".png"
+    
+    save_file = open(file_name, "wb")
+    save_file.write(response.content)
 
-print(response.text)
+    image = Image.open(file_name)
+    st.image(image, caption=prompt_input)