vinay-text-editor.py

from tkinter import *
import tkinter.filedialog
import os
import tkinter.messagebox as tmb
from pytesseract import image_to_string 
import cv2

PROGRAM_NAME="VINAY TEXT EDITOR"
file_name=None
root=Tk()
root.geometry('350x350')
root.title(PROGRAM_NAME)

def new_file():
	root.title("Untitled")
	global file_name
	file_name=None
	content_text.delete(1.0,END)
def open_file(event=None):
	input_file_name=tkinter.filedialog.askopenfilename(defaultextension=".txt",filetypes=[("All Files","*.*"),("Text Documents","*.txt")])
	if input_file_name:
		global file_name
		file_name=input_file_name
		root.title('{} - {}'.format(os.path.basename(file_name),PROGRAM_NAME))
		content_text.delete(1.0,END)
		with open(file_name) as _file:
			content_text.insert(1.0,_file.read())
def save_file(event=None):
	global file_name
	if not file_name:
		save_as_file()
	else:
		write_to_file(file_name)
	return "break"
def save_as_file():
	input_file_name=tkinter.filedialog.asksaveasfilename(defaultextension=".txt",filetypes=[("All Files","*.*"),("Text Documents","*.txt")])
	if input_file_name:
		global file_name
		file_name=input_file_name
		write_to_file(file_name)
		root.title('{} - {}'.format(os.path.basename(file_name),PROGRAM_NAME))
	return "break"
def write_to_file(file_name):
	try:
		content=content_text.get(1.0,'end')
		with open(file_name,'w') as the_file:
			the_file.write(content)
	except IOError:
		pass
def exit_editor():
	if tkinter.messagebox.askokcancel("Quit?","Are You Sure?"):
		root.destroy()
def undo():
	content_text.event_generate("<<Undo>>")
	return "break"
def redo(event=None):
	content_text.event_generate("<<Redo>>")
	return "break"
def cut():
	content_text.event_generate("<<Cut>>")
	return "break"
def copy():
	content_text.event_generate("<<Copy>>")
	return "break"
def paste():
	content_text.event_generate("<<Paste>>")
	return "break"
def find_text(event=None):
	search_toplevel=Toplevel(root)
	search_toplevel.title("Find Text")
	search_toplevel.transient(root)
	search_toplevel.resizable(False,False)
	Label(search_toplevel,text="Find All:").grid(row=0,column=0,sticky='e')
	search_entry_widget=Entry(search_toplevel,width=25)
	search_entry_widget.grid(row=0,column=1,padx=2,pady=2,sticky='we')
	search_entry_widget.focus_set()
	ignore_case_value=IntVar()
	Checkbutton(search_toplevel,text='Ignore Case',variable=ignore_case_value).grid(row=1,column=1,sticky='e',padx=2,pady=2)
	Button(search_toplevel,text="Find All",underline=0,command=lambda:search_output(search_entry_widget.get(),ignore_case_value.get(),content_text,search_toplevel,search_entry_widget)).grid(row=0,column=2,sticky='e'+'w',padx=2,pady=2)
	def close_search_window():
		content_text.tag_remove('match','1.0',END)
		search_toplevel.destroy()
		search_toplevel.protocol('WM_DELETE_WINDOW',close_search_window)
		return "break"
def search_output(needle,if_ignore_case,content_text,search_toplevel,search_box):
	content_text.tag_remove('match','1.0',END)
	matches_found=0
	if needle:
		start_pos='1.0'
		while True:
			start_pos=content_text.search(needle,start_pos,nocase=if_ignore_case,stopindex=END)
			if not start_pos:
				break
			end_pos='{}+{}c'.format(start_pos,len(needle))
			content_text.tag_add('match',start_pos,end_pos)
			matches_found+=1
			start_pos=end_pos
		content_text.tag_config('match',foreground='red',background='yellow')
	search_box.focus_set()
	search_toplevel.title('{} matches found'.format(matches_found))

def select_all(event=None):
	content_text.tag_add("sel","1.0","end")
	return "break"
def about():
	tkinter.messagebox.showinfo("About","{} {}".format(PROGRAM_NAME,"\nCoolest text editor you have ever used"))
def Help():
	tkinter.messagebox.showinfo("Help","For any query\n Feel Free to write email\n vinay070717@gmail.com\n Thank You",icon="question")
def show_line_no():
	pass
def show_cursor_info():
	pass
def on_content_changed(event=None):
	update_line_numbers()
	update_cursor_info_bar()
def get_line_numbers():
	output=''
	if(show_line_number.get()):
		row,col=content_text.index('end').split('.')
		for i in range(1,int(row)):
			output+=str(i)+'\n'
	return output
def update_line_numbers(event=None):
	line_number=get_line_numbers()
	line_number_bar.config(state='normal')
	line_number_bar.delete('1.0',END)
	line_number_bar.insert('1.0',line_number)
	line_number_bar.config(state='disabled')
def highlight_line(interval=100):
	content_text.tag_remove("activa_line",1.0,END)
	content_text.tag_add("active_line","insert linestart","insert lineend+1c")
	content_text.after(interval,toggle_highlight)
def undo_highlight():
	content_text.tag_remove("active_line",1.0,"end")
def toggle_highlight(event=None):
	if(to_highlight_line.get()):
		highlight_line()
	else:
		undo_highlight()
def show_cursor_info_bar():
	show_cursor_info_checked=show_cursor_info.get()
	if(show_cursor_info_checked):
		cursor_info_bar.pack(expand=NO,fill=None,side='right',anchor='se')
	else:
		cursor_info_bar.pack_forget()
def update_cursor_info_bar(event=None):
	row,col=content_text.index(INSERT).split('.')
	line_num,col_num=str(int(row)),str(int(col)+1)
	infotext="Line: {0} | Column: {1}".format(line_num,col_num)
	cursor_info_bar.config(text=infotext)
def change_theme(event=None):
	selected_theme=theme_choice.get()
	fg_bg_colors=color_schemes.get(selected_theme)
	foreground_color,background_color=fg_bg_colors.split('.')
	content_text.config(background=background_color,fg=foreground_color)

def read_image_file():
	input_file_name=tkinter.filedialog.askopenfilename(defaultextension=".png",filetypes=[("Png files","*.png"),("Jpeg Files","*.jpg")])
	if input_file_name:
		global file_name
		file_name=input_file_name
		root.title('{} - {}'.format(os.path.basename(file_name),PROGRAM_NAME))
		content_text.delete(1.0,END)
		# with open(file_name) as _file:
		print(file_name)
		im = cv2.imread(file_name,cv2.IMREAD_COLOR)
		config = ('-l eng --oem 1 --psm 3')
		text =image_to_string(im,config=config)
		content_text.insert(1.0,text)


new_file_icon = PhotoImage(file='icons/new_file.gif')
open_file_icon = PhotoImage(file='icons/open_file.gif')
save_file_icon = PhotoImage(file='icons/save.gif')
cut_icon = PhotoImage(file='icons/cut.gif')
copy_icon = PhotoImage(file='icons/copy.gif')
paste_icon = PhotoImage(file='icons/paste.gif')
undo_icon = PhotoImage(file='icons/undo.gif')
redo_icon = PhotoImage(file='icons/redo.gif')

menu_bar=Menu(root)
file_menu=Menu(menu_bar,tearoff=0)
file_menu.add_command(label='New',accelerator='Ctrl+N',compound='left',image=new_file_icon,underline=0,command=new_file)
file_menu.add_command(label='Open',accelerator='Ctrl+O',compound='left',image=open_file_icon,underline=0,command=open_file)
file_menu.add_command(label='Open image->text',compound='left',underline=0,command=read_image_file)
file_menu.add_command(label='Save',accelerator='Ctrl+S',compound='left',image=save_file_icon,underline=0,command=save_file)
file_menu.add_command(label='Save as',accelerator='Shift+Ctrl+N',compound='left',underline=0,command=save_as_file)
file_menu.add_separator()
file_menu.add_command(label='Exit',accelerator='Alt+F4',compound='left',underline=0,command=exit_editor)

menu_bar.add_cascade(label="File",menu=file_menu)
root.config(menu=menu_bar)

edit_menu=Menu(menu_bar,tearoff=0)
edit_menu.add_command(label="Undo",accelerator="Ctrl+Z",compound='left',image=undo_icon,underline=0,command=undo)
edit_menu.add_command(label="Redo",accelerator="Ctrl+Y",compound='left',image=redo_icon,underline=0,command=redo)
edit_menu.add_separator()
edit_menu.add_command(label="Cut",accelerator="Ctrl+X",compound='left',image=cut_icon,underline=0,command=cut)
edit_menu.add_command(label="Copy",accelerator="Ctrl+C",compound='left',image=copy_icon,underline=0,command=copy)
edit_menu.add_command(label="Paste",accelerator="Ctrl+V",compound='left',image=paste_icon,underline=0,command=paste)
edit_menu.add_separator()
edit_menu.add_command(label="Find",accelerator="Ctrl+F",compound='left',underline=0,command=find_text)
edit_menu.add_separator()
edit_menu.add_command(label="Select All",accelerator="Ctrl+A",compound='left',underline=0,command=select_all)


menu_bar.add_cascade(label='Edit',menu=edit_menu)
root.config(menu=menu_bar)
view_menu=Menu(menu_bar,tearoff=0)
show_line_number=IntVar()
show_line_number.set(1)
view_menu.add_checkbutton(label="Show Line Number",variable=show_line_number)
show_cursor_info=IntVar()
show_cursor_info.set(1)
view_menu.add_checkbutton(label='Show Cursor Location at Bottom', variable=show_cursor_info,command=show_cursor_info_bar)
to_highlight_line=BooleanVar()
view_menu.add_checkbutton(label='Highlight Current Line', onvalue=1, offvalue=0, variable=to_highlight_line,command=toggle_highlight)
themes_menu = Menu(menu_bar, tearoff=0)
view_menu.add_cascade(label='Themes', menu=themes_menu)
color_schemes = {
    'Default': '#000000.#FFFFFF',
    'Greygarious': '#83406A.#D1D4D1',
    'Aquamarine': '#5B8340.#D1E7E0',
    'Bold Beige': '#4B4620.#FFF0E1',
    'Cobalt Blue': '#ffffBB.#3333aa',
    'Olive Green': '#D1E7E0.#5B8340',
    'Night Mode': '#FFFFFF.#000000',
}

theme_choice = StringVar()
theme_choice.set('Default')
for k in sorted(color_schemes):
    themes_menu.add_radiobutton(label=k, variable=theme_choice,command=change_theme)
menu_bar.add_cascade(label='View',menu=view_menu)

root.config(menu=menu_bar)
about_menu=Menu(menu_bar,tearoff=0)
about_menu.add_command(label="About",compound='left',underline=0,command=about)
about_menu.add_command(label="Help",compound='left',underline=0,command=Help)

menu_bar.add_cascade(label='About',menu=about_menu)
root.config(menu=menu_bar)
shortcut_bar=Frame(root,height=25,background='light sea green')
shortcut_bar.pack(expand='no',fill='x')
icons=("new_file","open_file","cut","copy","paste","undo","redo","find_text")
for i,icon in enumerate(icons):
	tool_bar_icon=PhotoImage(file='icons/{}.gif'.format(icon))
	cmd = eval(icon)
	tool_bar=Button(shortcut_bar,image=tool_bar_icon,command=cmd)
	tool_bar.image=tool_bar_icon
	tool_bar.pack(side='left')
line_number_bar=Text(root,width=4,padx=3,takefocus=0,border=0,background='khaki',state='disabled',wrap='none')
line_number_bar.pack(side='left',fill='y')

content_text=Text(root,wrap='word',undo=1)
content_text.pack(expand='yes',fill='both')
content_text.bind("<Control-y>",redo)
content_text.bind("<Control-A>",select_all)
content_text.bind("<Control-a>",select_all)
content_text.bind("<Control-f>",find_text)
content_text.bind("<Control-F>",find_text)
content_text.bind("<Control-o>",open_file)
content_text.bind("<Control-O>",open_file)
content_text.bind("<Control-s>",save_file)
content_text.bind("<Control-S>",save_file)
content_text.bind("<Control-n>",new_file)
content_text.bind("<Control-N>",new_file)
content_text.bind('<KeyPress-F1>',help)
content_text.bind('<Any-KeyPress>',on_content_changed)
content_text.tag_configure('active_line',background='ivory2')

scroll_bar=Scrollbar(content_text)
content_text.configure(yscrollcommand=scroll_bar.set)
scroll_bar.config(command=content_text.yview)
scroll_bar.pack(side='right',fill='y')
cursor_info_bar=Label(content_text,text='Line: 1 | Column: 1')
cursor_info_bar.pack(expand=NO,fill=None,side=RIGHT,anchor='se')
root.protocol('WM_DELETE_WINDOW',exit_editor)
root.mainloop()