Open bdytx5 opened 5 months ago
heres a version of app.py with the option to download a youtube video from a link
`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 import colorama import youtube_dl import os import re
colorama.init(autoreset=True)
customtkinter.set_appearance_mode("System") customtkinter.set_default_color_theme("blue") # Themes: blue (default), dark-blue, green firstclick = True
def sanitize_folder_name(name):
return re.sub(r'[^a-zA-Z0-9_\s]', '', name)
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
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)
# YouTube URL
youtube_frame = customtkinter.CTkFrame(master)
youtube_frame.pack(fill=tk.BOTH, padx=10, pady=10)
customtkinter.CTkLabel(youtube_frame, text="YouTube URL:", font=font).pack(side=tk.LEFT, padx=5)
self.youtube_entry = customtkinter.CTkEntry(youtube_frame, width=50, font=font_b)
self.youtube_entry.pack(side=tk.LEFT, fill=tk.X, expand=True)
customtkinter.CTkButton(youtube_frame, text="Download", command=self.download_youtube_video, font=font).pack(side=tk.LEFT, padx=5)
# Language frame
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.default_language_text = "Enter language (or ignore to auto-detect)"
self.language_entry.insert(0, self.default_language_text)
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)
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
def browse(self):
initial_dir = os.getcwd()
folder_path = filedialog.askdirectory(initialdir=initial_dir)
self.path_entry.delete(0, tk.END)
self.path_entry.insert(0, folder_path)
def download_youtube_video(self):
url = self.youtube_entry.get()
if not url:
messagebox.showerror("Error", "Please enter a YouTube URL.")
return
ydl_opts = {
'format': 'bestaudio/best',
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}],
'outtmpl': '%(title)s.%(ext)s'
}
try:
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=False)
title = sanitize_folder_name(info_dict.get('title', ''))
output_path = self.path_entry.get() or os.getcwd()
output_dir = os.path.join(output_path, title)
if not os.path.exists(output_dir):
os.makedirs(output_dir)
ydl_opts['outtmpl'] = os.path.join(output_dir, '%(title)s.%(ext)s')
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
ydl.download([url])
messagebox.showinfo("Success", f"Downloaded to {output_dir}")
except Exception as e:
messagebox.showerror("Error", f"Failed to download video: {e}")
def start_transcription(self):
self.transcribe_button.configure(state=tk.DISABLED)
threading.Thread(target=self.transcribe_thread).start()
def transcribe_thread(self):
path = self.path_entry.get()
model = self.model_combobox.get()
language = self.language_entry.get()
if language == self.default_language_text or not language.strip():
language = None
verbose = self.verbose_var.get()
self.progress_bar.pack(fill=tk.X, padx=5, pady=5)
self.progress_bar.start()
glob_file = get_path(path)
try:
output_text = transcribe(path, glob_file, model, language, verbose)
except UnboundLocalError:
messagebox.showinfo("Files not found error!", 'Nothing found, choose another folder.')
except ValueError:
messagebox.showinfo("Invalid language name, you might have to clear the default text to continue!")
self.progress_bar.stop()
self.progress_bar.pack_forget()
self.transcribe_button.configure(state=tk.NORMAL)
try:
messagebox.showinfo("Finished!", output_text)
except UnboundLocalError:
pass
if name == "main": root = customtkinter.CTk() root.title("Local Transcribe with Whisper") width, height = 450, 275 root.geometry('{}x{}'.format(width, height)) root.iconbitmap('images/icon.ico') app = App(root) root.mainloop()`
this thing just works, salute to you. thank you