import os
os.system('pip install --upgrade --no-cache-dir gdown')
os.system('gdown -O ./output/ctw/model_ctw.pth 1Ajslu_9WisuZ2nJGzE6qbD87aK6_ozzA')
os.system('pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"')
os.system('python setup.py build develop --user')
#os.system('pip install git+https://github.com/Cyril-Sterling/ContourNet')

import cv2
import pandas as pd
import gradio as gr
from det_demo import DetDemo
from maskrcnn_benchmark.config import cfg


def infer(filepath):
    cfg.merge_from_file('./configs/det/r50_baseline.yaml')
    # manual override some options
    cfg.merge_from_list(["MODEL.DEVICE", "cpu"])

    det_demo = DetDemo(
        cfg,
        min_image_size=800,
        confidence_threshold=0.7,
        output_polygon=True
    )
    image = cv2.imread(filepath)
    result_polygons, result_masks, result_boxes = det_demo.run_on_opencv_image(image)
    visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes)
    cv2.imwrite('result.jpg', visual_image)
    return 'result.jpg'#, pd.DataFrame(result_words)


iface = gr.Interface(
    fn=infer,
    title="张博强毕设展示",
    description="毕设题目：自然场景中任意形状文字的检测与识别\n目前进度：检测",
    inputs=[gr.inputs.Image(label="image", type="filepath")],
    outputs=[gr.outputs.Image()], #, gr.outputs.Dataframe(headers=['word'])],
    #examples=['example1.jpg', 'example2.jpg', 'example3.jpg'],
    #article="<a href=\"https://github.com/MhLiao/MaskTextSpotterV3\">GitHub Repo</a>",
).launch(enable_queue=True)