#@title #(Optional) Check GPU
#@markdown Some models may not function correctly on a CPU runtime.

#@markdown so you should check your GPU setup before run.
!nvidia-smi

#@title #Installation
#@markdown This cell will install dependencies for Whisper-WebUI!
!git clone https://github.com/jhj0517/Whisper-WebUI.git
%cd Whisper-WebUI
!pip install git+https://github.com/jhj0517/jhj0517-whisper.git
!pip install faster-whisper==1.0.2
!pip install gradio==4.14.0
!pip install pytube
!pip install tokenizers==0.19.1

#@title # (Optional) Configure arguments
#@markdown This section is used to configure some command line arguments.

#@markdown You can simply ignore this section and the default values will be used.

USERNAME = '' #@param {type: "string"}
PASSWORD = '' #@param {type: "string"}
DISABLE_FASTER_WHISPER = False #@param {type: "boolean"}
THEME = '' #@param {type: "string"}

arguments = ""
if USERNAME:
  arguments += f" --username {USERNAME}"
if PASSWORD:
  arguments += f" --password {PASSWORD}"
if THEME:
  arguments += f" --theme {THEME}"
if DISABLE_FASTER_WHISPER:
  arguments += f" --disable_faster_whisper"


#@markdown If you wonder how these arguments are used, you can see the [Wiki](https://github.com/jhj0517/Whisper-WebUI/wiki/Command-Line-Arguments).

#@title #Run
#@markdown Once the installation is complete, you can use public URL that is displayed.
if 'arguments' in locals():
  !python app.py --share --colab{arguments}
else:
    !python app.py --share --colab