from depth import MidasDepth
import gradio as gr
import numpy as np
import cv2


depth_estimator = MidasDepth()


def get_depth(rgb):
    depth = depth_estimator.get_depth(rgb)

    return rgb, (depth.clip(0, 64) * 1024).astype("uint16")


starter = gr.Interface(fn=get_depth, inputs=[
    gr.components.Image(label="rgb", type="pil"),
], outputs=[
    gr.components.Image(type="pil", label="image"),
    gr.components.Image(type="numpy", label="depth"),

])

gr.Interface(get_depth).launch(share=True)