import gradio as gr
import json
import requests


def test():
    print('testing backend interface...')
    url = 'http://81.70.158.155:5005/predict'
    with open('./2680.jpg', 'rb') as f:
        try:
            r = requests.post(url, files = {'image' : f}, timeout=10)
            ret = json.loads(r.text)
        except Exception as e:
            print('backend test failed')
            return

    # image level cls score
    pred_cls_score = ret['cls_score']
    print('--------', pred_cls_score)
    print('backend test success')

def predict(img):
    url = 'http://81.70.158.155:5005/predict'
    with open(img, 'rb') as f:
        r = requests.post(url, files = {'image' : f})
        ret = json.loads(r.text)

    # image level cls score
    pred_cls_score = ret['cls_score']
    pred_seg = json.loads(ret['result'])
    pred_seg = np.array(pred_seg, dtype=np.uint8)
    
    return pred_seg
  
        
iface = gr.Interface(
    predict,
    inputs=gr.inputs.Image(label="Upload image to detect", type="filepath"),
    outputs='image',
    title="Forged? Or Not?",
)

test()

iface.launch()