3-gui-creation.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 33,
   "id": "7dbdfa48",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "අ\n",
      "අ\n",
      "අ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n",
      "උ\n"
     ]
    }
   ],
   "source": [
    "import tkinter as tk\n",
    "from PIL import ImageTk,Image,ImageDraw\n",
    "import numpy as np\n",
    "import cv2\n",
    "import os\n",
    "import joblib\n",
    "\n",
    "model = joblib.load('sinhala-character-knn.sav')\n",
    "\n",
    "width = 500\n",
    "height = 500\n",
    "\n",
    "font_btn = 'Helvetica 20 bold'\n",
    "font_label = 'Helvetica 22 bold'\n",
    "\n",
    "count = 0\n",
    "\n",
    "label_dict={0:'අ',1:'එ',2:'ඉ',3:'උ'}\n",
    "\n",
    "win = tk.Tk() # create main window\n",
    "\n",
    "def event_function(event):\n",
    "    \n",
    "    x = event.x\n",
    "    y = event.y\n",
    "    \n",
    "    x1=x-30\n",
    "    x2=x+30\n",
    "    \n",
    "    y1=y-30\n",
    "    y2=y+30\n",
    "    \n",
    "    canvas.create_oval((x1,y1,x2,y2),fill='black') # drow oval into canvas\n",
    "    img_drow.ellipse((x1,y1,x2,y2),fill='black') # drow oval into image\n",
    "    \n",
    "def save():\n",
    "    global count\n",
    "    \n",
    "    img_array = np.array(img) # convert image to numpy array\n",
    "    path=os.path.join('data',str(count)+'.jpg')\n",
    "    cv2.imwrite(path,img_array) # save the image in the data folder\n",
    "    count=count+1\n",
    "\n",
    "def clear():\n",
    "    global img,img_draw\n",
    "    \n",
    "    canvas.delete('all') # delete canvas charactor\n",
    "    img = Image.new('RGB',(width,height),(255,255,255)) # create new image\n",
    "    img_draw = ImageDraw.Draw(img)\n",
    "    \n",
    "    label_status.config(text='PREDICTED CHARACTER : NONE')\n",
    "\n",
    "def predict():\n",
    "    img_array = np.array(img)\n",
    "    img_array = cv2.cvtColor(img_array,cv2.COLOR_BGR2GRAY) # converting into a gray image\n",
    "    img_array = cv2.resize(img_array,(8,8)) # resizing into 8x8\n",
    "    img_array = np.reshape(img_array,(1,64)) # reshaping into 1x64 (flattening / 1D)\n",
    "     \n",
    "    result = model.predict(img_array)[0] # get result\n",
    "    print(label_dict[result])\n",
    "    label = label_dict[result]\n",
    "    \n",
    "    label_status.config(text='PREDICTED CHARACTER :'+label)\n",
    "    \n",
    "canvas = tk.Canvas(win,width=width,height=height,bg='white') # create canvas\n",
    "canvas.grid(row=0,column=0,columnspan=4) # place canvas\n",
    "\n",
    "button_save = tk.Button(win,text='SAVE',border=0,cursor='hand2',bg='green',fg='white',font=font_btn,command=save) # create button \n",
    "button_save.grid(row=1,column=0)\n",
    "\n",
    "button_save = tk.Button(win,text='PREDICT',border=0,cursor='hand2',bg='blue',fg='white',font=font_btn,command=predict)\n",
    "button_save.grid(row=1,column=1)\n",
    "\n",
    "button_save = tk.Button(win,text='CLEAR',border=0,cursor='hand2',bg='yellow',fg='white',font=font_btn,command=clear)\n",
    "button_save.grid(row=1,column=2)\n",
    "\n",
    "button_save = tk.Button(win,text='EXIT',border=0,cursor='hand2',bg='red',fg='white',font=font_btn,command=win.destroy)\n",
    "button_save.grid(row=1,column=3)\n",
    "\n",
    "label_status = tk.Button(win,text='PREDICTED CHARACTER : NONE',border=0,bg='white',font=font_label) # create label \n",
    "label_status.grid(row=2,column=0,columnspan=4)\n",
    "\n",
    "canvas.bind('<B1-Motion>',event_function) # apply mouse event\n",
    "\n",
    "img = Image.new('RGB',(width,height),(255,255,255)) # create image\n",
    "img_drow = ImageDraw.Draw(img)\n",
    "\n",
    "win.mainloop() # run window"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "75a33e33",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}