Update app.py

app.py

@@ -65,6 +65,29 @@ def sd_call(prompt, width, height):
     return image_link, fetch_result
 
 
+
+def sd_recall(fetch_result):
+    url = fetch_result
+    payload = json.dumps({
+        "key": "xbYWUXEs8U4iUweAQjOoH6mrasutoExqC3a2H6NtlnYAWTRXui9y0t2w1g8P"
+    })
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    response = requests.request("POST", url, headers=headers, data=payload)
+    response_data = json.loads(response.text)
+    image_link = response_data["output"]
+
+    if not image_link:
+        image_link = None
+    else:
+        image_link = image_link[0]
+    return image_link
+
+
+
+
+
 with gr.Blocks(theme=gr.themes.Default()) as app:
     with gr.Tab("삽화 생성"):
         with gr.Row():