whisper-local-transcribe/app.py

import tkinter as tk
from tkinter import ttk
from tkinter import filedialog
from tkinter import messagebox
from src._LocalTranscribe import transcribe, get_path
import customtkinter
import threading
from colorama import Back, Fore
import colorama
colorama.init(autoreset=True)



customtkinter.set_appearance_mode("System")
customtkinter.set_default_color_theme("blue")  # Themes: blue (default), dark-blue, green
firstclick = True

class App:
    def __init__(self, master):
        print(Back.CYAN + "Welcome to Local Transcribe with Whisper!\U0001f600\nCheck back here to see some output from your transcriptions.\nDon't worry, they will also be saved on the computer!\U0001f64f")
        self.master = master
        # Change font
        font = ('Roboto', 13, 'bold')  # Change the font and size here
        font_b = ('Roboto', 12)  # Change the font and size here
        # Folder Path
        path_frame = customtkinter.CTkFrame(master)
        path_frame.pack(fill=tk.BOTH, padx=10, pady=10)
        customtkinter.CTkLabel(path_frame, text="Folder:", font=font).pack(side=tk.LEFT, padx=5)
        self.path_entry = customtkinter.CTkEntry(path_frame, width=50, font=font_b)
        self.path_entry.pack(side=tk.LEFT, fill=tk.X, expand=True)
        customtkinter.CTkButton(path_frame, text="Browse", command=self.browse, font=font).pack(side=tk.LEFT, padx=5)
        # Language frame        
        #thanks to pommicket from Stackoverflow for this fix
        def on_entry_click(event):
            """function that gets called whenever entry is clicked"""        
            global firstclick
            if firstclick: # if this is the first time they clicked it
                firstclick = False
                self.language_entry.delete(0, "end") # delete all the text in the entry
        language_frame = customtkinter.CTkFrame(master)
        language_frame.pack(fill=tk.BOTH, padx=10, pady=10)
        customtkinter.CTkLabel(language_frame, text="Language:", font=font).pack(side=tk.LEFT, padx=5)
        self.language_entry = customtkinter.CTkEntry(language_frame, width=50, font=('Roboto', 12, 'italic'))
        self.language_entry.insert(0, 'Select language or clear to detect automatically')
        self.language_entry.bind('<FocusIn>', on_entry_click)
        self.language_entry.pack(side=tk.LEFT, fill=tk.X, expand=True)
        # Model frame
        models = ['base.en', 'base', 'small.en',
                  'small', 'medium.en', 'medium', 'large']
        model_frame = customtkinter.CTkFrame(master)
        model_frame.pack(fill=tk.BOTH, padx=10, pady=10)
        customtkinter.CTkLabel(model_frame, text="Model:", font=font).pack(side=tk.LEFT, padx=5)
        # ComboBox frame
        self.model_combobox = customtkinter.CTkComboBox(
            model_frame, width=50, state="readonly",
            values=models, font=font_b)
        self.model_combobox.set(models[1])  # Set the default value
        self.model_combobox.pack(side=tk.LEFT, fill=tk.X, expand=True)
        # Verbose frame
        verbose_frame = customtkinter.CTkFrame(master)
        verbose_frame.pack(fill=tk.BOTH, padx=10, pady=10)
        self.verbose_var = tk.BooleanVar()
        customtkinter.CTkCheckBox(verbose_frame, text="Output transcription to terminal", variable=self.verbose_var, font=font).pack(side=tk.LEFT, padx=5)
        # Progress Bar
        self.progress_bar = ttk.Progressbar(master, length=200, mode='indeterminate')
        # Button actions frame
        button_frame = customtkinter.CTkFrame(master)
        button_frame.pack(fill=tk.BOTH, padx=10, pady=10)
        self.transcribe_button = customtkinter.CTkButton(button_frame, text="Transcribe", command=self.start_transcription, font=font)
        self.transcribe_button.pack(side=tk.LEFT, padx=5, pady=10, fill=tk.X, expand=True)
        customtkinter.CTkButton(button_frame, text="Quit", command=master.quit, font=font).pack(side=tk.RIGHT, padx=5, pady=10, fill=tk.X, expand=True)
    # Helper functions
    # Browsing
    def browse(self):
        folder_path = filedialog.askdirectory()
        self.path_entry.delete(0, tk.END)
        self.path_entry.insert(0, folder_path)
    # Start transcription
    def start_transcription(self):
        # Disable transcribe button
        self.transcribe_button.configure(state=tk.DISABLED)
        # Start a new thread for the transcription process
        threading.Thread(target=self.transcribe_thread).start()
    # Threading
    def transcribe_thread(self):
        path = self.path_entry.get()
        model = self.model_combobox.get()
        language = self.language_entry.get() or None
        verbose = self.verbose_var.get()
        # Show progress bar
        self.progress_bar.pack(fill=tk.X, padx=5, pady=5)
        self.progress_bar.start()
        # Setting path and files
        glob_file = get_path(path)
        info_path = 'I will transcribe all eligible audio/video files in the path: {}\n\nContinue?'.format(path)
        answer = messagebox.askyesno("Confirmation", info_path)
        if not answer:
            self.progress_bar.stop()
            self.progress_bar.pack_forget()
            self.transcribe_button.configure(state=tk.NORMAL)
            return
        # Start transcription
        error_language = 'https://github.com/openai/whisper#available-models-and-languages'
        try:
            output_text = transcribe(path, glob_file, model, language, verbose)
        except UnboundLocalError:
            messagebox.showinfo("Files not found error!", 'Nothing found, choose another folder.')
            pass
        except ValueError:
            messagebox.showinfo("Language error!", 'See {} for supported languages'.format(error_language))
        # Hide progress bar
        self.progress_bar.stop()
        self.progress_bar.pack_forget()
        # Enable transcribe button
        self.transcribe_button.configure(state=tk.NORMAL)
        # Recover output text
        try:
            messagebox.showinfo("Finished!", output_text)
        except UnboundLocalError:
            pass

if __name__ == "__main__":
    # Setting custom themes
    root = customtkinter.CTk()
    root.title("Local Transcribe with Whisper")
    # Geometry
    width,height = 450,275
    root.geometry('{}x{}'.format(width,height))
    # Icon 
    root.iconbitmap('images/icon.ico')
    # Run
    app = App(root)
    root.mainloop()