Spaces:
Sleeping
Sleeping
File size: 2,059 Bytes
441f847 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import gradio as gr
import requests
import time
import json
import base64
import os
from PIL import Image
from io import BytesIO
class SuperCanvas:
def __init__(self, api_key):
self.base = "https://api.supercanvas.ai/v1/run/dfd23676-575c-498a-84f4-10d19af1e65b"
self.headers = {
"accept": "application/json",
"content-type": "application/json",
"authorization": f"Bearer {api_key}"
}
def generate(self, prompt):
payload = { "properties": {
"user_prompt": prompt
}}
response = requests.post(self.base, json=payload, headers=self.headers)
return response.json()
def wait(self, job):
job_result = job
while job_result['status'] in ['running']:
time.sleep(0.25)
return job_result
supercanvas_client = SuperCanvas(api_key=os.getenv("SUPERCANVAS_API_TOKEN"))
def get_result(prompt):
result = supercanvas_client.generate(prompt)
job = supercanvas_client.wait(result["run_id"])
return job["urls"][0]
css = """
#generate {
height: 100%;
}
"""
with gr.Blocks(css=css) as demo:
with gr.Row():
with gr.Column(scale=1):
gr.Markdown(elem_id="powered-by-supercanvas-ai", value="Powered by [SuperCanvas.AI](https://www.supercanvas.ai/).")
with gr.Tab("Prompt"):
with gr.Row():
with gr.Column(scale=6, min_width=600):
prompt = gr.Textbox("hogwarts school", placeholder="Prompt", show_label=False, lines=3)
with gr.Column():
text_button = gr.Button("Generate", variant='primary', elem_id="generate")
with gr.Row():
with gr.Column(scale=2):
image_output = gr.Image(value="https://storage.googleapis.com/ai-pipeline-app.appspot.com/JGoawqXDF1x2zKuKXj8ZWg/95bdb06c-03a5-4902-a0a0-0d5a206f4ca1-0.png")
text_button.click(get_result, inputs=[prompt], outputs=image_output)
demo.queue(concurrency_count=24)
demo.launch() |