fix error

app.py

@@ -1,46 +1,27 @@
 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
+def get_result(prompt):
 
+    API_URL = "https://api.supercanvas.ai/v1/run/dfd23676-575c-498a-84f4-10d19af1e65b"
 
+    payload = { "properties": {
+        "user_prompt": prompt
+    }}
 
-supercanvas_client = SuperCanvas(api_key=os.getenv("SUPERCANVAS_API_TOKEN"))
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "authorization": f"Bearer {os.getenv('SUPERCANVAS_API_TOKEN')}"
+    }
 
-def get_result(prompt):
-    result = supercanvas_client.generate(prompt)
+    response = requests.post(API_URL, json=payload, headers=headers)
 
-    job = supercanvas_client.wait(result["run_id"])
+    result = response.json()
 
-    return job["urls"][0]
+    return result["output"]["urls"][0]
 
 css = """
 #generate {
@@ -58,7 +39,7 @@ with gr.Blocks(css=css) as demo:
     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)
+                prompt = gr.Textbox("hogwarts school of witchcraft and wizardry, 8k", placeholder="Prompt", show_label=False, lines=3)
             with gr.Column():
                 text_button = gr.Button("Generate", variant='primary', elem_id="generate")