director tools and send jump

app.py

@@ -8,20 +8,14 @@ from huggingface_hub import HfApi, snapshot_download
 
 from pnginfo import read_info_from_image, send_paras
 from images_history import img_history_ui
+from director_tools import director_ui, send_outputs
 from utils import set_token, generate_novelai_image, image_from_bytes, get_remain_anlas, calculate_cost
 
 client_config = toml.load("config.toml")['client']
 today_count = 0
-anlas = -1
 today = datetime.date.today().strftime('%Y-%m-%d')
 api = HfApi()
 
-def refresh_anlas():
-    global anlas
-    set_token(os.environ.get('token'))
-    anlas = get_remain_anlas()
-    return anlas
-
 def get_count():
     global today_count, today
     now = datetime.date.today().strftime('%Y-%m-%d')
@@ -61,7 +55,7 @@ def control_ui():
         height = gr.Slider(label="高度", value=1024, minimum=64, maximum=2048, step=64)
     with gr.Row():
         with gr.Column():
-            with gr.Accordion('风格迁移', open=False):
+            with gr.Accordion('风格迁移', open=False) as vibe_tab:
                 ref_images = gr.Gallery(label="上传单(多)张图片", format="png", value=None, interactive=True, type="pil", show_share_button=False)
                 info_extracts = gr.State([])
                 ref_strs = gr.State([])
@@ -85,13 +79,13 @@ def control_ui():
                             s.change(fn=dynparas, inputs=strs, outputs=ref_strs)
                 ref_images.change(lambda l: ([], []) if l is None else ([1]*len(l), [0.6]*len(l)), inputs=ref_images, outputs=[info_extracts, ref_strs])
                 ref_images.upload(lambda l: ([], []) if l is None else ([1]*len(l), [0.6]*len(l)), inputs=ref_images, outputs=[info_extracts, ref_strs])
-            with gr.Accordion('附加输入', open=False):
-                with gr.Tab('图生图') as i2i:
+            with gr.Accordion('附加输入', open=False, elem_id="i2i_tab") as i2i_tab:
+                with gr.Tab('图生图', elem_id="i2i_block") as i2i:
                     i2i_image = gr.Image(label="上传图片", value=None, sources=["upload", "clipboard", "webcam"], interactive=True, type="pil", show_share_button=False)
                     i2i_str = gr.Slider(label='去噪强度', value=0.7, minimum=0, maximum=0.99, step=0.01)
                     i2i_noise = gr.Slider(label='噪声', value=0, minimum=0, maximum=1, step=0.1)
                     reuse_img_i2i = gr.Button(value='使用上一次生成的图片')
-                with gr.Tab('局部重绘') as inp:
+                with gr.Tab('局部重绘', elem_id="inp_block") as inp:
                     overlay = gr.Checkbox(label='覆盖原图', value=True)
                     inp_img = gr.ImageMask(label="上传图片", value=None, sources=["upload", "clipboard", "webcam"], interactive=True, type="pil", eraser=False, transforms=None, brush=gr.Brush(colors=['#FFFFFF'], color_mode='fixed'), layers=False, show_share_button=False)
                     reuse_img_inp = gr.Button(value='使用上一次生成的图片')
@@ -120,7 +114,7 @@ def control_ui():
     rand_seed.click(fn=lambda: -1, inputs=None, outputs=seed)
     i2i.select(lambda: 'i2i', inputs=None, outputs=selection)
     inp.select(lambda: 'inp', inputs=None, outputs=selection)
-    return gen_btn, stop_btn, [prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection], [save, rand_seed, reuse_seed, reuse_img_i2i, reuse_img_inp]
+    return gen_btn, stop_btn, [prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection], [save, rand_seed, reuse_seed, reuse_img_i2i, reuse_img_inp, vibe_tab, i2i_tab]
 
 def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, ref_images, info_extracts, ref_strs, i2i_image, i2i_str, i2i_noise, overlay, inp_img, selection):
     global today_count
@@ -140,8 +134,10 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
 def preview_ui():
     with gr.Blocks(css='#preview_image { height: 100%;}'):
         image = gr.Image(format='png', elem_id='preview_image', interactive=False, type='filepath', show_share_button=False)
+        send_dtool = gr.Button(value="发送到引导工具", visible=False)
+        image.change(lambda i: gr.Button(visible=False) if i is None else gr.Button(visible=True), inputs=image, outputs=send_dtool)
         info = gr.JSON(value={}, label="生成信息")
-    return image, info
+    return image, info, send_dtool
 
 def rename_save_img(path, payload, save):
     if path is None:
@@ -168,38 +164,39 @@ def update_btn_cost(w, h, s, sm, dyn, i2i_img, i2i_str, inp_img, selection):
         cost = calculate_cost(w, h, s, False, False)
     else:
         cost = calculate_cost(w, h, s, sm, dyn)
-    return gr.Button(value=f"生成(消耗{cost}点数)")
+    return gr.Button(value=f"生成(预计消耗{cost}点数)")
 
 def main_ui():
-    with gr.Blocks() as page:
+    with gr.Blocks():
         with gr.Row(variant="panel"):
             with gr.Column():
                 gen_btn, stop_btn, paras, others = control_ui()
             with gr.Column():
-                image, info = preview_ui()
+                image, info, send_dtool = preview_ui()
     cost_list = paras[5:8] + paras[10:12] + paras[17:19] + paras[21:23]
     for component in cost_list:
         component.change(update_btn_cost, inputs=cost_list, outputs=gen_btn)
     gen_btn.click(lambda: (gr.Button(visible=False), gr.Button(visible=True)), inputs=None, outputs=[gen_btn, stop_btn])
-    gen = gen_btn.click(generate, paras, [image, info], concurrency_limit=1)
+    gen = gen_btn.click(generate, paras, [image, info], concurrency_limit=1, concurrency_id="generate")
     others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
     others[3].click(lambda i: i, inputs=image, outputs=paras[17])
     others[4].click(lambda i: gr.ImageEditor(value=i), inputs=image, outputs=paras[21])
     ren = image.change(rename_save_img, inputs=[image, info, others[0]], outputs=image, trigger_mode="once")
     image.change(lambda: (gr.Button(visible=True), gr.Button(visible=False)), inputs=None, outputs=[gen_btn, stop_btn])
     stop_btn.click(lambda: (gr.Button(visible=True), gr.Button(visible=False)), inputs=None, outputs=[gen_btn, stop_btn], cancels=[gen, ren])
-    return page, paras[:14]
+    return image, paras, others, send_dtool
 
 def util_ui():
-    with gr.Blocks(analytics_enabled=False) as page:
+    with gr.Blocks():
         with gr.Row(equal_height=False):
             with gr.Column(variant='panel'):
                 image = gr.Image(label="上传图片", sources=["upload"], interactive=True, type="pil")
             with gr.Column(variant='panel'):
-                info = gr.HTML()
+                info = gr.HTML('')
                 items = gr.JSON(value=None, visible=False)
-                png2main = gr.Button('参数发送到文生图')
-    return page, png2main, items, info, image
+                png2main = gr.Button('参数发送到文生图', visible=False)
+                items.change(lambda i: gr.Button(visible=False) if i is None else gr.Button(visible=True), inputs=items, outputs=png2main)
+    return png2main, items, info, image
     
 def load_javascript():
     head = ''
@@ -215,22 +212,40 @@ def load_javascript():
         return res
 
     gr.routes.templates.TemplateResponse = template_response
-    
+
+def send_and_jump(component, fn, inputs, outputs, tab, trigger):
+    component.click(fn, inputs=inputs, outputs=outputs)
+    component.click(fn=None,
+                    js="(x) => { if (x !== null && x != 0) document.getElementById('" + tab + "-button').click(); return null; }",
+                    inputs=trigger)
+
+def send_jump_select(component, fn, inputs, outputs, tab, accordin, trigger, secondary=None):
+    component.click(fn, inputs=inputs, outputs=outputs)
+    extra_js = ''
+    if secondary is not None:
+        extra_js = "document.getElementById('" + secondary + "-button').click();"
+    component.click(fn=None,
+                    js="(x) => { if (x !== null && x != 0) document.getElementById('" + tab + "-button').click();" + extra_js + " return null; }",
+                    inputs=trigger)
+    component.click(lambda x: gr.Accordion(open=True) if x else gr.Accordion(), inputs=trigger, outputs=accordin)
+
 def ui():
     load_javascript()
-    refresh_anlas()
-    with gr.Blocks(title="NAI Client", theme=gr.themes.Base(), js="() => {document.body.classList.toggle('dark');}") as website:
+    set_token(os.environ.get('token'))
+    with gr.Blocks(title="NAI Client", theme=gr.themes.Base(), js="() => {document.body.classList.toggle('dark', true);}") as website:
         with gr.Row():
-            remain_anlas = gr.Text(label="剩余点数", value=anlas)
+            remain_anlas = gr.Textbox(label="剩余点数", value=get_remain_anlas, scale=3)
             refresh = gr.Button("🔄")
-            refresh.click(refresh_anlas, inputs=None, outputs=remain_anlas)
+            refresh.click(get_remain_anlas, inputs=None, outputs=remain_anlas)
         with gr.Tabs():
             with gr.TabItem("图片生成", elem_id="client_ui_main"):
-                _, paras = main_ui()
+                image, paras, others, send_dtool = main_ui()
+            with gr.TabItem("引导工具", elem_id="client_ui_dtool"):
+                from_t2i, send_i2i, send_inp, send_vib, in_image, out_image, d_index = director_ui()
             with gr.TabItem("图片信息读取"):
-                _, png2main, png_items, info, image = util_ui()
+                png2main, png_items, info, read_image = util_ui()
             with gr.TabItem("云端图片浏览") as tab:
-                gal2main, gal_items = img_history_ui(tab)
+                gallery, h_index, gal2main, gal_items, history2ref, history2i2i, history2inp, history2dtl = img_history_ui(tab)
             with gr.TabItem("设置"):
                 switchLightDark = gr.Button(value="切换浅色/深色模式")
                 switchLightDark.click(fn=None, js="() => {document.body.classList.toggle('dark');}")
@@ -238,19 +253,18 @@ def ui():
                 loadTagComplete.click(fn=None, js="() => {run();}")
                 clearTagCache = gr.Button(value="清除tag补全缓存")
                 clearTagCache.click(fn=None, js="() => {localStorage.clear();}")
-            png2main.click(fn=send_paras,
-                           inputs=[png_items] + paras,
-                           outputs=paras)
-            png2main.click(fn=None,
-                           js="(x) => { if (x !== null) document.getElementById('client_ui_main-button').click(); return null; }",
-                           inputs=image)
-            gal2main.click(fn=send_paras,
-                           inputs=[gal_items] + paras,
-                           outputs=paras)
-            gal2main.click(fn=None,
-                           js="(x) => { if (x !== null) document.getElementById('client_ui_main-button').click(); return null; }",
-                           inputs=gal_items)
-            image.change(read_info_from_image, inputs=image, outputs=[info, png_items])
+            send_and_jump(png2main, send_paras, [png_items] + paras[:14], paras[:14], "client_ui_main", read_image)
+            send_and_jump(gal2main, send_paras, [gal_items] + paras[:14], paras[:14], "client_ui_main", gal_items)
+            from_t2i.click(lambda x: x, inputs=image, outputs=in_image)
+            send_jump_select(send_i2i, send_outputs, [out_image, d_index], paras[17], "client_ui_main", others[6], out_image, "i2i_block")
+            send_jump_select(send_inp, send_outputs, [out_image, d_index], paras[21], "client_ui_main", others[6], out_image, "inp_block")
+            send_jump_select(send_vib, (lambda l, i: l if i == -1 else [l[i]]), [out_image, d_index], paras[14], "client_ui_main", others[5], out_image)
+            send_and_jump(send_dtool, (lambda x: x), image, in_image, "client_ui_dtool", image)
+            send_jump_select(history2i2i, send_outputs, [gallery, h_index], paras[17], "client_ui_main", others[6], gal_items, "i2i_block")
+            send_jump_select(history2inp, send_outputs, [gallery, h_index], paras[21], "client_ui_main", others[6], gal_items, "inp_block")
+            send_jump_select(history2ref, (lambda l, i: None if i == -1 else [l[i]]), [gallery, h_index], paras[14], "client_ui_main", others[5], gal_items)
+            send_and_jump(history2dtl, send_outputs, [gallery, h_index], in_image, "client_ui_dtool", gal_items)
+            read_image.change(read_info_from_image, inputs=read_image, outputs=[info, png_items])
     return website
 
 

director_tools.py

@@ -0,0 +1,98 @@
+import gradio as gr
+from utils import calculate_cost, augment_image
+
+def update_btn_cost(w, h, req_type, selection, l=''):
+    if selection == 'scale':
+        after = l.split()[-1]
+        w, h = map(int, after.split('x'))
+    cost = calculate_cost(w, h, rmbg=(req_type == "移除背景"))
+    return gr.Button(value=f"生成(预计消耗{cost}点数)")
+
+def getWH(img, w, h, f):
+    if img is None:
+        return w, h, ''
+    w, h = img.width, img.height
+    l = f"{w}x{h} -> {int(w * f)}x{int(h * f)}"
+    return w, h, l
+
+def refresh_hint(l, f):
+    if len(l) == 0:
+        return ''
+    before = l.split()[0]
+    w, h = map(int, before.split('x'))
+    l = f"{w}x{h} -> {int(w * f)}x{int(h * f)}"
+    return l
+
+def set_index(evt: gr.SelectData):
+    return evt.index
+
+def send_outputs(imgs, index):
+    if imgs is None or len(imgs) == 0:
+        return None
+    elif len(imgs) == 1:
+        return imgs[0][0]
+    elif index >= 0:
+        return imgs[index][0]
+    else:
+        return None
+    
+def director_ui():
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="上传图片", value=None, sources=["upload", "clipboard", "webcam"], interactive=True, type="pil", show_share_button=False)
+            with gr.Row():
+                from_t2i = gr.Button("使用文生图上一次生成的图片")
+                from_out = gr.Button("使用右侧生成的图片")
+        with gr.Column():
+            output_image = gr.Gallery(format='png', interactive=False, preview=True, show_share_button=False)
+            selected_index = gr.Number(value=-1, visible=False)
+            with gr.Row():
+                send_i2i = gr.Button("发送到图生图")
+                send_inp = gr.Button("发送到局部重绘")
+                send_vib = gr.Button("发送到风格迁移")
+    req_type = gr.Radio(["移除背景", "素描", "线稿", "上色", "更改表情", "去聊天框"], value="线稿", label="功能选择")
+    with gr.Tab("自由调整") as resize:
+        width = gr.Slider(label="调整宽度", value=1024, minimum=1, maximum=2048, step=1)
+        height = gr.Slider(label="调整高度", value=1024, minimum=1, maximum=2048, step=1)
+    with gr.Tab("等比例缩放") as scale:
+        hint = gr.HTML('', show_label=False)
+        factor = gr.Slider(label="缩放比例", minimum=0.1, maximum=2, step=0.1)
+    defry = gr.State(0)
+    prompt = gr.State('')
+    selection = gr.Radio(["resize", "scale"], visible=False)
+    resize.select(lambda: 'resize', inputs=None, outputs=selection)
+    scale.select(lambda: 'scale', inputs=None, outputs=selection)
+    @gr.render(inputs=req_type)
+    def extra_paras(r):
+        if r == "上色":
+            with gr.Row():
+                strength = gr.Radio(choices=range(6), value=0, label="强度")
+                text = gr.Textbox(label="提示词(可选)", value='')
+                strength.change(lambda x: x, inputs=strength, outputs=defry)
+                text.change(lambda x: x, inputs=text, outputs=prompt)
+        elif r == "更改表情":
+            with gr.Row():
+                emotion = gr.Dropdown(choices=["neutral", "happy", "sad", "angry", "scared", "surprised", "tired", "excited", "nervous", "thinking", "confused", "shy", "disgusted", "smug", "bored", "laughing", "irritated", "aroused", "embarrassed", "worried", "love", "determined", "hurt", "playful"], value="neutral", label="表情")
+                strength = gr.Dropdown(choices=["正常", "偏弱", "弱", "更弱", "很弱", "最弱"], value="正常", label="强度")
+                text = gr.Textbox(label="提示词(可选)", value='')
+                emotion.change(lambda e, t: f"{e};;{t}", inputs=[emotion, text], outputs=prompt)
+                text.change(lambda e, t: f"{e};;{t}", inputs=[emotion, text], outputs=prompt)
+                strength.change(lambda s: {"正常": 0, "偏弱": 1, "弱": 2, "更弱": 3, "很弱": 4, "最弱": 5}[s], inputs=strength, outputs=defry)
+        else:
+            return
+    output_image.select(set_index, inputs=None, outputs=selected_index)
+    from_out.click(send_outputs, inputs=[output_image, selected_index], outputs=input_image)
+    req_type.change(lambda r: "neutral;;" if r == "更改表情" else '', inputs=req_type, outputs=prompt)
+    input_image.change(getWH, inputs=[input_image, width, height, factor], outputs=[width, height, hint])
+    factor.change(refresh_hint, inputs=[hint, factor], outputs=[hint])
+    tool_btn = gr.Button(value="生成", variant="primary")
+    dtool_stop = gr.Button(value="取消", variant="stop", visible=False)
+    tool_btn.click(lambda: (gr.Button(visible=False), gr.Button(visible=True)), inputs=None, outputs=[tool_btn, dtool_stop])
+    dtool_gen = tool_btn.click(augment_image, inputs=[input_image, width, height, req_type, selection, factor, defry, prompt], outputs=output_image, concurrency_id="generate")
+    dtool_stop.click(lambda: (gr.Button(visible=True), gr.Button(visible=False)), inputs=None, outputs=[tool_btn, dtool_stop], cancels=[dtool_gen])
+    width.change(update_btn_cost, inputs=[width, height, req_type, selection, hint], outputs=tool_btn)
+    height.change(update_btn_cost, inputs=[width, height, req_type, selection, hint], outputs=tool_btn)
+    req_type.change(update_btn_cost, inputs=[width, height, req_type, selection, hint], outputs=tool_btn)
+    factor.change(update_btn_cost, inputs=[width, height, req_type, selection, hint], outputs=tool_btn)
+    selection.change(update_btn_cost, inputs=[width, height, req_type, selection, hint], outputs=tool_btn)
+    return from_t2i, send_i2i, send_inp, send_vib, input_image, output_image, selected_index

images_history.py

@@ -61,14 +61,18 @@ def img_history_ui(tab):
                 page_index = gr.Number(value=1, label="页码", precision=0)
                 next_page = gr.Button('下一页')
                 end_page = gr.Button('末尾页')
-            history_gallery = gr.Gallery(show_label=False, columns=6, height=1024, show_share_button=False)
+            history_gallery = gr.Gallery(show_label=False, format='png', columns=6, height=1024, interactive=False, show_share_button=False)
         with gr.Column(scale=1):
             info = gr.HTML()
             items = gr.JSON(value=None, visible=False)
             selected_index = gr.Number(value=-1, visible=False, precision=0)
             with gr.Row():
-                send_btn = gr.Button('参数发送到文生图')
-                del_btn = gr.Button('删除')
+                send_btn = gr.Button('参数发送到文生图', visible=False)
+                history2ref = gr.Button("图片发送到风格迁移", visible=False)
+                history2i2i = gr.Button("图片发送到图生图", visible=False)
+                history2inp = gr.Button("图片发送到局部重绘", visible=False)
+                history2dtl = gr.Button("图片发送到引导工具", visible=False)
+                del_btn = gr.Button('删除', visible=False)
     
     img_dir_path.change(lambda path: get_img_list(path, 1) + ('', None), inputs=img_dir_path, outputs=[history_gallery, page_index, info, items])
     img_dir_path.blur(update_img_dir, inputs=None, outputs=img_dir_path)
@@ -80,5 +84,5 @@ def img_history_ui(tab):
     history_gallery.select(get_img_info, inputs=None, outputs=[selected_index, info, items])
     history_gallery.change(lambda: -1, outputs=selected_index)
     del_btn.click(del_img, inputs=[selected_index, img_dir_path, page_index], outputs=history_gallery)
-
-    return send_btn, items
+    items.change(lambda s: [gr.Button(visible=False) for _ in range(6)] if s is None else [gr.Button(visible=True) for _ in range(6)], inputs=items, outputs=[send_btn, history2ref, history2i2i, history2inp, history2dtl, del_btn])
+    return history_gallery, selected_index, send_btn, items, history2ref, history2i2i, history2inp, history2dtl

utils.py

@@ -5,15 +5,14 @@ import requests
 import json
 import base64
 import math
+import gradio as gr
 
 from PIL import Image
 
-
 jwt_token = ''
 url = "https://image.novelai.net/ai/generate-image"
 headers = {}
 
-
 def set_token(token):
     global jwt_token, headers
     if jwt_token == token:
@@ -34,9 +33,9 @@ def get_remain_anlas():
     except:
         return '获取失败，err:' + str(data)
 
-def calculate_cost(width, height, steps, sm, dyn, strength=1, rmbg=False):
+def calculate_cost(width, height, steps=28, sm=False, dyn=False, strength=1, rmbg=False):
     pixels = width * height
-    if pixels <= 1048576 and steps <= 28:
+    if pixels <= 1048576 and steps <= 28 and not rmbg:
         return 0
     dyn = sm and dyn
     L = math.ceil(2951823174884865e-21 * pixels + 5.753298233447344e-7 * pixels * steps)
@@ -122,7 +121,7 @@ def generate_novelai_image(
     try:
         response = requests.post(url, json=payload, headers=headers, timeout=180)
     except:
-        return None, {'message': 'NAI response timeout'}
+        raise gr.Error('NAI response timeout')
 
     # Process the response
     if response.headers.get('Content-Type') == 'binary/octet-stream':
@@ -133,9 +132,11 @@ def generate_novelai_image(
                 with zip_ref.open(file_names[0]) as file:
                     return file.read(), payload
             else:
-                return "NAI doesn't return any images", json.loads(response.content)
+                messages = json.loads(response.content)
+                raise gr.Error(messages["statusCode"] + ": " + messages["message"])
     else:
-        return "Generation failed", json.loads(response.content)
+        messages = json.loads(response.content)
+        raise gr.Error(messages["statusCode"] + ": " + messages["message"])
 
 def image_from_bytes(data):
     img_file = io.BytesIO(data)
@@ -147,4 +148,38 @@ def image2base64(img):
     img.save(output_buffer, format='PNG' if img.mode=='RGBA' else 'JPEG')
     byte_data = output_buffer.getvalue()
     base64_str = base64.b64encode(byte_data).decode()
-    return base64_str
+    return base64_str
+
+def augment_image(image, width, height, req_type, selection, factor=1, defry=0, prompt=''):
+    if selection == "scale":
+        width = int(width * factor)
+        height = int(height * factor)
+    image = image.resize((width, height))
+    req_type = {"移除背景": "bg-removal", "素描": "sketch", "线稿": "lineart", "上色": "colorize", "更改表情": "emotion", "去聊天框": "declutter"}[req_type]
+    base64img = image2base64(image)
+    payload = {"image": base64img, "width": width, "height": height, "req_type": req_type}
+    if req_type == "colorize" or req_type == "emotion":
+        payload["defry"] = defry
+        payload["prompt"] = prompt
+    
+    try:
+        response = requests.post("https://image.novelai.net/ai/augment-image", json=payload, headers=headers, timeout=60)
+    except:
+        raise gr.Error('NAI response timeout')
+
+    # Process the response
+    if response.headers.get('Content-Type') == 'binary/octet-stream':
+        zipfile_in_memory = io.BytesIO(response.content)
+        with zipfile.ZipFile(zipfile_in_memory, 'r') as zip_ref:
+            if len(zip_ref.namelist()):
+                files = []
+                for file_name in zip_ref.namelist():
+                    with zip_ref.open(file_name) as file:
+                        files.append(file.read())
+                return files
+            else:
+                messages = json.loads(response.content)
+                raise gr.Error(messages["statusCode"] + ": " + messages["message"])
+    else:
+        messages = json.loads(response.content)
+        raise gr.Error(messages["statusCode"] + ": " + messages["message"])