prompt output

app.py

@@ -95,9 +95,21 @@ def simple_desc(img, prompt):
   total = ""
   for resp in gen:
     print(total := total + resp)
-  return total
+  img.resize((102,192)).save("tmp.png")
+  bts = False
+  with open("tmp.png", "rb") as f:
+    bts = f.read()
+  if bts:
+    os.remove("tmp.png")
+  res = {
+    'image_b64': base64.b64encode(bts).decode('utf-8'),
+    'description': total,
+  }
+  return total, res
+
+ifc_imgprompt2text = gr.Interface(simple_desc, inputs=[gr.Image(label="input", type="pil"), gr.Textbox(label="prompt")], outputs=[gr.Textbox(label="description"), gr.JSON(label="json")])
+
 
-ifc_imgprompt2text = gr.Interface(simple_desc, inputs=[gr.Image(label="input", type="pil"), gr.Textbox(label="prompt")], outputs=[gr.Textbox(label="prompt")])
 
 """
 with gr.Blocks() as demo: