import os
import sys
import gradio as gr
import subprocess
from fastai.vision.all import *
import torch

# set device to GPU
device = torch.device('cpu')


os.system('git clone https://github.com/iPERDance/iPERCore.git')
os.system('curl -o ./iPERCore/assets/checkpoints.zip http://101.32.75.151:10086/checkpoints.zip')
os.system('curl -o ./iPERCore/assets/samples.zip http://101.32.75.151:12345/iper_plus_plus_latest_samples.zip')

title = "full-body-animation"
description = "A demo for full-body-animation Official Repo"
article = "Official Repo: https://github.com/iPERDance/iPERCore"

def inference(img, video):
    if not os.path.exists('temp'):
        os.system('mkdir temp')
    
    os.system(f"python iPERCore.services.run_imitator --gpu_ids 0 --num_source 2 --src_path {img} --ref_path {video} --output_dir tmp/result.mp4")
    return "tmp/result.mp4"


gr.Interface(
    inference,
    [
        gr.inputs.Image(type="filepath", label="Source (only used if mode is animation)"),
        gr.inputs.Video(type="mp4", label="Driving Video")
    ],
    outputs=gr.outputs.Video(type="mp4", label="Output Video"),
    title=title,
    description=description,
    article=article,
    theme="huggingface",
    allow_flagging=False).launch(debug=True,enable_queue=True