---
- name: Bringing-Old-Photos-Back-to-Life
  hosts: all
  gather_facts: no

# Succesfully tested on Ubuntu 18.04\20.04 and Debian 10 

  pre_tasks: 
  - name: install packages
    package:
      name:
        - python3
        - python3-pip
        - python3-venv
        - git
        - unzip
        - tar
        - lbzip2
        - build-essential
        - cmake
        - ffmpeg
        - libsm6
        - libxext6
        - libgl1-mesa-glx
      state: latest
    become: yes

  tasks:
  - name: git clone repo
    git:
      repo: 'https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life.git'
      dest: Bringing-Old-Photos-Back-to-Life
      clone: yes

  - name: requirements setup
    pip:
      requirements: "~/Bringing-Old-Photos-Back-to-Life/requirements.txt"
      virtualenv: "~/Bringing-Old-Photos-Back-to-Life/.venv"
      virtualenv_command: /usr/bin/python3 -m venv .venv

  - name: additional pip packages #requirements lack some packs
    pip:
      name: 
          - setuptools 
          - wheel
          - scikit-build
      virtualenv: "~/Bringing-Old-Photos-Back-to-Life/.venv"
      virtualenv_command: /usr/bin/python3 -m venv .venv

  - name: git clone batchnorm-pytorch
    git:
      repo: 'https://github.com/vacancy/Synchronized-BatchNorm-PyTorch'
      dest: Synchronized-BatchNorm-PyTorch
      clone: yes

  - name: copy sync_batchnorm to face_enhancement
    copy:
      src: Synchronized-BatchNorm-PyTorch/sync_batchnorm
      dest: Bringing-Old-Photos-Back-to-Life/Face_Enhancement/models/networks/
      remote_src: yes

  - name: copy sync_batchnorm to global
    copy:
      src: Synchronized-BatchNorm-PyTorch/sync_batchnorm
      dest: Bringing-Old-Photos-Back-to-Life/Global/detection_models
      remote_src: yes

  - name: check if shape_predictor_68_face_landmarks.dat
    stat:
      path: Bringing-Old-Photos-Back-to-Life/Face_Detection/shape_predictor_68_face_landmarks.dat
    register: p

  - name: get shape_predictor_68_face_landmarks.dat.bz2
    get_url:
      url: http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2
      dest: Bringing-Old-Photos-Back-to-Life/Face_Detection/
    when: p.stat.exists == False

  - name: unarchive shape_predictor_68_face_landmarks.dat.bz2
    shell: 'bzip2 -d Bringing-Old-Photos-Back-to-Life/Face_Detection/shape_predictor_68_face_landmarks.dat.bz2'
    when: p.stat.exists == False

  - name: check if face_enhancement
    stat:
      path: Bringing-Old-Photos-Back-to-Life/Face_Enhancement/checkpoints/Setting_9_epoch_100/latest_net_G.pth
    register: fc

  - name: unarchive Face_Enhancement/checkpoints.zip
    unarchive:
      src: https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Face_Enhancement/checkpoints.zip
      dest: Bringing-Old-Photos-Back-to-Life/Face_Enhancement/
      remote_src: yes
    when: fc.stat.exists == False

  - name: check if global
    stat:
      path: Bringing-Old-Photos-Back-to-Life/Global/checkpoints/detection/FT_Epoch_latest.pt
    register: gc

  - name: unarchive Global/checkpoints.zip
    unarchive:
      src: https://facevc.blob.core.windows.net/zhanbo/old_photo/pretrain/Global/checkpoints.zip
      dest: Bringing-Old-Photos-Back-to-Life/Global/
      remote_src: yes
    when: gc.stat.exists == False

# Do not forget to execute 'source .venv/bin/activate' inside Bringing-Old-Photos-Back-to-Life before starting run.py