Compare commits

2 Commits

Author SHA1 Message Date
Kristofer Söderström 7d50d5f4cf QOL improvements 2023-11-06 09:57:44 +01:00
Kristofer Söderström 7799d03960 bug fixes 2023-11-06 09:31:53 +01:00
3 changed files with 12 additions and 15 deletions
+11 -12
View File
@@ -5,9 +5,10 @@ from tkinter import messagebox
from src._LocalTranscribe import transcribe, get_path
import customtkinter
import threading
from colorama import Back, Fore
from colorama import Back
import colorama
colorama.init(autoreset=True)
import os
@@ -41,7 +42,8 @@ class App:
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.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
@@ -72,7 +74,8 @@ class App:
# Helper functions
# Browsing
def browse(self):
folder_path = filedialog.askdirectory()
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)
# Start transcription
@@ -85,22 +88,18 @@ class App:
def transcribe_thread(self):
path = self.path_entry.get()
model = self.model_combobox.get()
language = self.language_entry.get() or None
language = self.language_entry.get()
# Check if the language field has the default text or is empty
if language == self.default_language_text or not language.strip():
language = None # This is the same as passing nothing
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 = 'Continue?'
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
messagebox.showinfo("Message", "Starting transcription!")
# 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:
@@ -1,4 +1,2 @@
Armstrong_Small_Step
[0:00:00 --> 0:00:07]: And they're still brought to land now.
[0:00:07 --> 0:00:18]: It's one small step for man.
[0:00:18 --> 0:00:23]: One by a fleet for man time.
[0:00:00 --> 0:00:29.360000]: alumnfeldaguyrjarna om det nya skirprå kızım om det där föddarna hatt splittar, do nackrott,
Binary file not shown.