paint.py

from tkinter import *
import tkinter.ttk as ttk
from tkinter import colorchooser
from tkinter import filedialog
import PIL
from PIL import Image, ImageDraw, ImageTk
from tkinter import messagebox
import pyscreenshot as ImageGrab

root = Tk()
root.title('Noor_fa_paint')
root.geometry("800x800")
brush_color = "black"

def paint(e):

    # Brush Parameters
    brush_width =  int(my_slider.get()) 
    #brush_color = "green"
    brush_type2 = brush_type.get() # Brush type : BUTT, ROUND, PROJECTING

    #Starting position
    x1 = e.x - 1
    x2 = e.y - 1

    # Ending Position
    x2 = e.x +1
    y2 = e.y +1

    # Draw on tha canvas
    my_canvas.create_line(x1, y2, x2, y2, fill=brush_color, width=brush_width, capstyle=brush_type2, smooth=True)

# Change the size of the brush
def change_brush_size(e):
    slider_label.config(text=int(my_slider.get()))

# Chnage brush color
def change_brush_color():
    global brush_color
    brush_color = "black"
    brush_color = colorchooser.askcolor(color=brush_color)[1]
    # color = Label(root, text=brush_color)
    # color.pack(pady=20)

# Change canvas color
def change_canvas_color():
    global bg_color
    bg_color = "black"
    bg_color = colorchooser.askcolor(color=bg_color)[1]
    my_canvas.config(bg=bg_color)

# Clear Screen
def clear_screen():
    my_canvas.delete(ALL)
    my_canvas.config(bg="white")

# Save Image
def save_as_png():
    result = filedialog.asksaveasfilename(initialdir='c:/Desktop/', filetypes=(("png files", "*.png"), ("all files", "*.* ")))
    if result.endswith(".png"):
        pass
    else:
        result = result + ".png"

    if result:
        x = root.winfo_rootx()+my_canvas.winfo_x()
        y = root.winfo_rooty()+my_canvas.winfo_y()
        x1 = x+my_canvas.winfo_width()
        y1 = y+my_canvas.winfo_height()
        ImageGrab.grab().crop((x,y,x1,y1)).save(result)

        messagebox.showinfo("Image Saved", "Your Image has been saved")

    # result_label = Label(root, text=result)
    # result_label.pack(pady=20)

w = 600
h = 400

my_canvas = Canvas(root, width=w, height=h, bg="white")
my_canvas.pack(pady=20)

#B1- Left button , B2 - Middle button(Scrol button), B3 - RIght button in Mouse
my_canvas.bind('<B1-Motion>', paint)

# Create Brush Options Frame
brush_options_frame = Frame(root)
brush_options_frame.pack(pady=20)

#Brush Size
brush_size_frame = LabelFrame(brush_options_frame, text="Brush Size")
brush_size_frame.grid(row=0, column=0, padx=50)

# Brush Slider
my_slider = ttk.Scale(brush_size_frame, from_=1, to=100, command=change_brush_size, orient=VERTICAL, value=10)
my_slider.pack(pady=10, padx=10)

#Brush Slider label 
slider_label = Label(brush_size_frame, text=my_slider    .get())
slider_label.pack(pady=5)

# Brush Type
brush_type_frame = LabelFrame(brush_options_frame, text="Brush Type", height=400)
brush_type_frame.grid(row=0, column=1, padx=50)

brush_type = StringVar()
brush_type.set("round")

# Create Radio Buttons for brush type
brush_type_radio1 = Radiobutton(brush_type_frame, text="Round", variable=brush_type, value="round")
brush_type_radio2 = Radiobutton(brush_type_frame, text="Slash", variable=brush_type, value="butt")
brush_type_radio3 = Radiobutton(brush_type_frame, text="Diamond", variable=brush_type, value="projecting")

brush_type_radio1.pack(anchor=W)
brush_type_radio2.pack(anchor=W)
brush_type_radio3.pack(anchor=W)

# Change colors
change_colors_frame = LabelFrame(brush_options_frame, text="Change colors")
change_colors_frame.grid(row=0, column=2)

#Change brush color button
brush_color_button = Button(change_colors_frame, text="Brush Color", command=change_brush_color)
brush_color_button.pack(pady=10, padx=10)

#Change Canvas Background color
canvas_color_button = Button(change_colors_frame, text="Canvas Color", command=change_canvas_color)
canvas_color_button.pack(pady=10, padx=10)

#Program Options Frame
options_frame = LabelFrame(brush_options_frame, text="Program Options")
options_frame.grid(row=0, column=3, padx=50)

# clear screen button
clear_button = Button(options_frame, text="Clear Screen", command=clear_screen)
clear_button.pack(padx=10, pady=10)

# Save Image
save_image_button = Button(options_frame, text="Save To PNG", command=save_as_png)
save_image_button.pack(pady=10, padx=10)

root.mainloop()