import os
import gradio as gr
os.system("pip3 install cython_bbox gdown")
os.system("git clone https://github.com/Megvii-BaseDetection/YOLOX.git")
os.chdir("YOLOX")
os.system("pip3 install -r requirements.txt")
os.system("pip3 install -v -e .")
os.chdir("..")
os.system("gdown https://drive.google.com/uc?id=1RE5vaKHYt_TL0cXNn6FmjO7aZx3sVfVA")
from pathlib import Path


def inference(video):
  os.system("python3 deploy/ONNXRuntime/onnx_inference.py -i "+video)
  vidname = Path(video).stem
  return "demo_output/"+vidname+".mp4"
  

title = "Anime2Sketch"
description = "demo for Anime2Sketch. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2104.05703'>Adversarial Open Domain Adaption for Sketch-to-Photo Synthesis</a> | <a href='https://github.com/Mukosame/Anime2Sketch'>Github Repo</a></p>"

gr.Interface(
    inference, 
    "video", 
    "playable_video",
    title=title,
    description=description,
    article=article
    ).launch(debug=True)