monkey-storage / app.py
Bart6114's picture
init
a1f30f8
raw
history blame
558 Bytes
import gradio as gr
import monkey
import tempfile
def encode_decode(data, input_image):
if len(data) > 0:
fp =tempfile.NamedTemporaryFile(suffix = '.jpg').name
img = monkey.encode(data)
img.save(fp)
return img, fp, None
elif input_image is not None:
print(input_image)
return None, None, monkey.decode(input_image)
else:
raise Exception("please provide data OR input_image")
iface = gr.Interface(fn=encode_decode, inputs=["text", "pil"], outputs=["image", "file", "text"])
iface.launch()