English Translated GUI

[JD-2006]

Copy below into main.py. If the file changes I will not be updating.

import os import easygui import tkinter as tk from tkinter import filedialog import tkinter.ttk from tkinter.constants import * from tkinter import messagebox import subprocess

import torch import time

import ttkbootstrap as ttk

from ttkbootstrap.constants import *

easygui.fileopenbox()

import webbrowser

def callback(url): webbrowser.open_new(url)

def selectPhotoFolder(): photoPath = easygui.diropenbox("Subtitle storage folder") output_dir.delete(0, 'end') output_dir.insert(0, photoPath)

def selectAudioFile(): paths = filedialog.askopenfilenames() for path in paths: displayAudioFilePath.insert(END, path)

File = easygui.fileopenbox("Choose a sound file file")

# displayAudioFilePath.delete(0, 'end')
# displayAudioFilePath.insert(0, File)

def detectAvailableDevice(): if torch.cuda.is_available(): for i in range(torch.cuda.device_count()): devices.append(torch.cuda.get_device_name(i))

usingDevice['value'] = devices
return

def deviceChange(index, value, op):

print("%s %s %s"%(index, value, op))

dev = devices.index(usingDevice.get())
if dev > 0:
    print("cuda:%s"%(dev-1))
else:
    print("cpu")

def deviceDecode(): dev = devices.index(usingDevice.get()) if dev > 0: return("cuda:%s"%(dev-1)) else: return("cpu")

def process():

File = displayAudioFilePath.get(0)

if displayAudioFilePath.size() == 0:
    messagebox.showerror(title="Mistake", message='No choice')
    return 0

start = time.time()

baseCommandStr = "venv\\Scripts\\whisper"

for i in range(displayAudioFilePath.size()):
    path = displayAudioFilePath.get(i)
    print(path)
    commandStr = baseCommandStr + ' "%s"'%path

    languageInput = usingLanguage.get()

    if languageInput != "Automatic detection":
        commandStr = commandStr + " --language %s "%languageInput

    deviceInput = deviceDecode()
    commandStr = commandStr + " --device %s --fp16 False"%deviceInput

    commandStr = commandStr + " --model %s "%usingModel.get()

    output_dirInput = output_dir.get()

    if outputToTheSamePathAsInputVar.get()=="1":
        output_dirInput = "/".join(path.split("/")[:-1])
        # print(output_dirInput)

    if output_dirInput != "":
        commandStr = commandStr + ' --output_dir "%s" '%output_dirInput

    commandStr = commandStr + " --model_dir %s "%("model")

    if translateToEnglishVar.get() == '1':
        commandStr = commandStr + " --task %s "%("translate")

    # print(os.system("echo %s"%commandStr))
    # print(commandStr)
    out = subprocess.Popen(commandStr)
    (out, err) = out.communicate()

end = time.time()

messagebox.showinfo(title="Message", message="Processing complete \n Spending time %.2F seconds"%(end-start))
#outputPreviewVar.set(out)
#print(out)

heightFix_1 = 60

window = tk.Tk() window.title('WhisperGUI By The Walking Fish') window.geometry('580x290') window.resizable(False, False)

label1 = tk.Label(text='Choice') label1.place(x=0, y=0) displayAudioFilePath = tk.Listbox(width=60, height=5) displayAudioFilePath.place(x=80, y=0) selectAudioFileButton = ttk.Button(text='＋Add to', command=selectAudioFile) selectAudioFileButton.place(x=510, y=10) selectAudioFileButton = ttk.Button(text='－Delete', bootstyle='danger', command=lambda x=displayAudioFilePath: x.delete("active")) selectAudioFileButton.place(x=510, y=40)

label2 = tk.Label(text='Subtitle storage folder') label2.place(x=0, y=30+heightFix_1) output_dir = tk.Entry(width=55) output_dir.place(x=120, y=30+heightFix_1) output_dir.insert(0, os.getcwd() + "\output") selectPhotoPathButton = tk.Button(text='....', command=selectPhotoFolder) selectPhotoPathButton.place(x=500, y=30+heightFix_1)

outputToTheSamePathAsInputVar = tk.StringVar() outputToTheSamePathAsInput = tk.Checkbutton(text="File output to the same position as an input file", variable=outputToTheSamePathAsInputVar, onvalue="1", offvalue="0") outputToTheSamePathAsInput.deselect() outputToTheSamePathAsInput.place(x=300, y=60+heightFix_1)

label_usingModel = tk.Label(text='Use model') label_usingModel.place(x=0, y=60+heightFix_1) var = tk.StringVar() usingModel = tkinter.ttk.Combobox(window, textvariable=var) usingModel['value'] = ('tiny', 'base', 'small', 'medium', 'large', 'large-v1', 'large-v2') usingModel.current(2) usingModel.place(x=60, y=60+heightFix_1)

label_usingDevice = tk.Label(text='Use device') label_usingDevice.place(x=0, y=90+heightFix_1) deviceVar = tk.StringVar() deviceVar.trace("w", deviceChange) usingDevice = tkinter.ttk.Combobox(window, textvariable=deviceVar) devices = ['cpu'] usingDevice['value'] = devices detectAvailableDevice() usingDevice.current(0) usingDevice.place(x=60, y=90+heightFix_1)

label_language = tk.Label(text='Recognize language') label_language.place(x=0, y=120+heightFix_1) languageVar = tk.StringVar() usingLanguage = tkinter.ttk.Combobox(window, textvariable=languageVar) languages = ['Automatic detection', "Afrikaans","Albanian","Amharic","Arabic","Armenian","Assamese","Azerbaijani","Bashkir","Basque","Belarusian","Bengali","Bosnian","Breton","Bulgarian","Burmese","Castilian","Catalan","Chinese","Croatian","Czech","Danish","Dutch","English","Estonian","Faroese","Finnish","Flemish","French","Galician","Georgian","German","Greek","Gujarati","Haitian","Haitian Creole","Hausa","Hawaiian","Hebrew","Hindi","Hungarian","Icelandic","Indonesian","Italian","Japanese","Javanese","Kannada","Kazakh","Khmer","Korean","Lao","Latin","Latvian","Letzeburgesch","Lingala","Lithuanian","Luxembourgish","Macedonian","Malagasy","Malay","Malayalam","Maltese","Maori","Marathi","Moldavian","Moldovan","Mongolian","Myanmar","Nepali","Norwegian","Nynorsk","Occitan","Panjabi","Pashto","Persian","Polish","Portuguese","Punjabi","Pushto","Romanian","Russian","Sanskrit","Serbian","Shona","Sindhi","Sinhala","Sinhalese","Slovak","Slovenian","Somali","Spanish","Sundanese","Swahili","Swedish","Tagalog","Tajik","Tamil","Tatar","Telugu","Thai","Tibetan","Turkish","Turkmen","Ukrainian","Urdu","Uzbek","Valencian","Vietnamese","Welsh","Yiddish","Yoruba"] usingLanguage['value'] = languages usingLanguage.current(0) usingLanguage.place(x=60, y=120+heightFix_1)

translateToEnglishVar = tk.StringVar() translateToEnglish = tk.Checkbutton(text="Translate the output subtitles into English", variable=translateToEnglishVar, onvalue="1", offvalue="0") translateToEnglish.deselect() translateToEnglish.place(x=0, y=150+heightFix_1)

label_copyright = tk.Label(text='MIT License') label_copyright.place(x=0, y=180+heightFix_1) label_author = tk.Label(text='Make: The Walking Fish') label_author.bind("", lambda e: callback("https://www.youtube.com/@the_walking_fish")) label_author.place(x=0, y=200+heightFix_1)

processButton = tk.Button(text='Execute', width=20, command=process) processButton.place(anchor='center', x=290, y=210+heightFix_1)

window.mainloop()

[JD-2006]

English UI version main.py. Updated v1.8

import os
import easygui
import tkinter as tk
from tkinter import filedialog
import tkinter.ttk
from tkinter.constants import *
from tkinter import messagebox
import subprocess
import json

import torch
import time
import urllib.request

import ttkbootstrap as ttk
# from ttkbootstrap.constants import *

#easygui.fileopenbox()

import webbrowser

def callback(url):
    webbrowser.open_new(url)

def selectPhotoFolder():
    outputDir = easygui.diropenbox("Subtitle storage folder")
    output_dir.delete(0, 'end')
    output_dir.insert(0, outputDir)
    config["outputDir"] = outputDir
    saveConfig("outputDir", outputDir)
    # with open('config.json', 'w', encoding='utf8') as f:
    #     json.dump(config, f)

def selectAudioFile():
    paths = filedialog.askopenfilenames()
    for path in paths:
        displayAudioFilePath.insert(END, path)
    # File = easygui.fileopenbox("Choose a sound file file")
    # displayAudioFilePath.delete(0, 'end')
    # displayAudioFilePath.insert(0, File)

def detectAvailableDevice():
    if torch.cuda.is_available():
        for i in range(torch.cuda.device_count()):
            devices.append(torch.cuda.get_device_name(i))

    usingDevice['value'] = devices
    return

def deviceChange(index, value, op):
    #print("%s %s %s"%(index, value, op))
    dev = devices.index(usingDevice.get())
    config["usingDevice"] = dev
    saveConfig("usingDevice", dev)
    # with open('config.json', 'w', encoding='utf8') as f:
    #     json.dump(config, f)

    if dev > 0:
        print("cuda:%s"%(dev-1))
    else:
        print("cpu")

def deviceDecode():
    dev = devices.index(usingDevice.get())
    if dev > 0:
        return("cuda:%s"%(dev-1))
    else:
        return("cpu")

def languageChange(index, value, op):
    transcribeLanguage = languages.index(usingLanguage.get())
    config["transcribeLanguage"] = transcribeLanguage
    saveConfig("transcribeLanguage", transcribeLanguage)
    # with open('config.json', 'w', encoding='utf8') as f:
    #     json.dump(config, f)

def modelChange(index, value, op):
    model = models.index(usingModel.get())
    config["usingModel"] = model
    saveConfig("usingModel", model)
    # with open('config.json', 'w', encoding='utf8') as f:
    #     json.dump(config, f)

def versionCheck(ori): # fetch app version
    try:
        url = 'https://raw.githubusercontent.com/ADT109119/WhisperGUI/main/version.txt'
        response = urllib.request.urlopen(url)
        fetchVersion = response.read().decode('utf-8')
        if fetchVersion != "ver 1.8":
            checkVisitGithub = messagebox.askquestion(title="Have a new version", message="The latest version is%s\nDo you want to go to GitHub to download the latest version"%(fetchVersion))
            if checkVisitGithub == 'yes':
                callback("https://github.com/ADT109119/WhisperGUI")
        elif ori == 1:
            messagebox.showinfo(title="Message", message="The current version is the latest")

    except:
        print("Unable to obtain version information")

def saveConfig(key, value):
    global config
    config[key] = value
    with open('config.json', 'w', encoding='utf8') as f:
        json.dump(config, f)

def process():
    # File = displayAudioFilePath.get(0)
    if displayAudioFilePath.size() == 0:
        messagebox.showerror(title="Mistake", message='No choice')
        return 0

    start = time.time()

    baseCommandStr = "venv\\Scripts\\whisper"

    for i in range(displayAudioFilePath.size()):
        path = displayAudioFilePath.get(i)
        print(path)
        commandStr = baseCommandStr + ' "%s"'%path

        languageInput = usingLanguage.get()

        if languageInput != "Automatic detection":
            commandStr = commandStr + " --language %s "%languageInput

        deviceInput = deviceDecode()
        commandStr = commandStr + " --device %s --fp16 False"%deviceInput

        commandStr = commandStr + " --model %s "%usingModel.get()

        output_dirInput = output_dir.get()

        if outputToTheSamePathAsInputVar.get()=="1":
            output_dirInput = "/".join(path.split("/")[:-1])
            # print(output_dirInput)

        if output_dirInput != "":
            commandStr = commandStr + ' --output_dir "%s" '%output_dirInput

        commandStr = commandStr + " --model_dir %s "%("model")

        if translateToEnglishVar.get() == '1':
            commandStr = commandStr + " --task %s "%("translate")

        if initial_prompt.get() != "":
            commandStr = commandStr + ' --initial_prompt "%s" '%initial_prompt.get()

        # print(os.system("echo %s"%commandStr))
        # print(commandStr)
        out = subprocess.Popen(commandStr)
        (out, err) = out.communicate()

    end = time.time()

    messagebox.showinfo(title="Message", message="Processing complete \n Spending time %.2F seconds"%(end-start))
    #outputPreviewVar.set(out)
    #print(out)

# cinfig
config = {
    "outputDir": os.getcwd() + "\\output",
    "usingModel": 2,
    "usingDevice": 0,
    "transcribeLanguage": 0,
    "autoCheckVersion": True
}

if not os.path.exists(".\\config.json"):
    with open('config.json', 'w', encoding='utf8') as f:
        json.dump(config, f)
else:
    with open('config.json', 'r', encoding='utf8') as f:
        config = json.load(f)

if config["autoCheckVersion"] == True:
    versionCheck(0)

heightFix_1 = 70

window = tk.Tk()
window.title('WhisperGUI By The Walking Fish')
window.geometry('580x330')
window.resizable(False, False)

label1 = tk.Label(text='Choice')
label1.place(x=0, y=10)
displayAudioFilePath = tk.Listbox(width=60, height=5)
displayAudioFilePath.place(x=80, y=10)
selectAudioFileButton = ttk.Button(text='＋Add to', command=selectAudioFile)
selectAudioFileButton.place(x=510, y=20)
selectAudioFileButton = ttk.Button(text='－Delete', bootstyle='danger', command=lambda x=displayAudioFilePath: x.delete("active"))
selectAudioFileButton.place(x=510, y=50)

label2 = tk.Label(text='Subtitle storage folder')
label2.place(x=0, y=30+heightFix_1)
output_dir = tk.Entry(width=55)
output_dir.place(x=120, y=30+heightFix_1)
output_dir.insert(0, config["outputDir"])
selectPhotoPathButton = tk.Button(text='....', command=selectPhotoFolder)
selectPhotoPathButton.place(x=500, y=30+heightFix_1)

outputToTheSamePathAsInputVar = tk.StringVar()
outputToTheSamePathAsInput = tk.Checkbutton(text="File output to the same position as an input file", variable=outputToTheSamePathAsInputVar, onvalue="1", offvalue="0")
outputToTheSamePathAsInput.deselect()
outputToTheSamePathAsInput.place(x=300, y=60+heightFix_1)

label_usingModel = tk.Label(text='Use model')
label_usingModel.place(x=0, y=60+heightFix_1)
var = tk.StringVar()
var.trace("w", modelChange)
usingModel = tkinter.ttk.Combobox(window, textvariable=var)
models = ['tiny', 'base', 'small', 'medium', 'large', 'large-v1', 'large-v2', 'large-v3']
usingModel['value'] = models
usingModel.current(config["usingModel"])
usingModel.place(x=60, y=60+heightFix_1)

label_usingDevice = tk.Label(text='Use device')
label_usingDevice.place(x=0, y=90+heightFix_1)
deviceVar = tk.StringVar()
deviceVar.trace("w", deviceChange)
usingDevice = tkinter.ttk.Combobox(window, textvariable=deviceVar)
devices = ['cpu']
usingDevice['value'] = devices
detectAvailableDevice()
usingDevice.current(config["usingDevice"])
usingDevice.place(x=60, y=90+heightFix_1)

label_language = tk.Label(text='Recognize language')
label_language.place(x=0, y=120+heightFix_1)
languageVar = tk.StringVar()
languageVar.trace("w", languageChange)
usingLanguage = tkinter.ttk.Combobox(window, textvariable=languageVar)
languages = ['Automatic detection', "Afrikaans","Albanian","Amharic","Arabic","Armenian","Assamese","Azerbaijani","Bashkir","Basque","Belarusian","Bengali","Bosnian","Breton","Bulgarian","Burmese","Castilian","Catalan","Chinese","Croatian","Czech","Danish","Dutch","English","Estonian","Faroese","Finnish","Flemish","French","Galician","Georgian","German","Greek","Gujarati","Haitian","Haitian Creole","Hausa","Hawaiian","Hebrew","Hindi","Hungarian","Icelandic","Indonesian","Italian","Japanese","Javanese","Kannada","Kazakh","Khmer","Korean","Lao","Latin","Latvian","Letzeburgesch","Lingala","Lithuanian","Luxembourgish","Macedonian","Malagasy","Malay","Malayalam","Maltese","Maori","Marathi","Moldavian","Moldovan","Mongolian","Myanmar","Nepali","Norwegian","Nynorsk","Occitan","Panjabi","Pashto","Persian","Polish","Portuguese","Punjabi","Pushto","Romanian","Russian","Sanskrit","Serbian","Shona","Sindhi","Sinhala","Sinhalese","Slovak","Slovenian","Somali","Spanish","Sundanese","Swahili","Swedish","Tagalog","Tajik","Tamil","Tatar","Telugu","Thai","Tibetan","Turkish","Turkmen","Ukrainian","Urdu","Uzbek","Valencian","Vietnamese","Welsh","Yiddish","Yoruba"]
usingLanguage['value'] = languages
usingLanguage.current(config["transcribeLanguage"])
usingLanguage.place(x=60, y=120+heightFix_1)

translateToEnglishVar = tk.StringVar()
translateToEnglish = tk.Checkbutton(text="Translate the output subtitles into English", variable=translateToEnglishVar, onvalue="1", offvalue="0")
translateToEnglish.deselect()
translateToEnglish.place(x=0, y=150+heightFix_1)

initial_prompt_label = tk.Label(text='Content reminder')
initial_prompt_label.place(x=0, y=180+heightFix_1)
initial_prompt = tk.Entry(width=55)
initial_prompt.place(x=80, y=180+heightFix_1)

label_copyright = tk.Label(text='MIT License')
label_copyright.place(x=0, y=210+heightFix_1)
label_author = tk.Label(text='Made by: The Walking Fish')
label_author.bind("<Button-1>", lambda e: callback("https://www.youtube.com/@the_walking_fish"))
label_author.place(x=0, y=230+heightFix_1)

processButton = tk.Button(text='Implement', width=20, command=process)
processButton.place(anchor='center', x=290, y=240+heightFix_1)

# menu
menu = tk.Menu(window)
settingMenu = tk.Menu(menu)
autoCheckVar = tk.BooleanVar()
autoCheckVar.trace("w", lambda index, value, op: saveConfig("autoCheckVersion", autoCheckVar.get()))
autoCheckVar.set(config["autoCheckVersion"])
settingMenu.add_checkbutton(label="Automatic check version", variable=autoCheckVar)
settingMenu.add_separator()
settingMenu.add_command(label="Check version", command=lambda: versionCheck(1))
menu.add_cascade(label="setting", menu=settingMenu)
window.config(menu=menu)

window.mainloop()