GuiMultipleCameras.py

from main import execute
from WebsiteVidCapture import run
from tkinter import *
from tkinter.ttk import *
import os
import time

""" 
    purpose: Creates GUI for Traffic analyzer. Passes arguments to main.py

    arguments
    ----------
    file_name: type = String, optimal: must be a video type that cv2.VideoCaputure accepts
    result_name: type = String, optimal: file should not exist, otherwise it overwrites the file
    up_down: type = Integer, optimal: 0 = count cars going down, 1 = count cars going
    up, 2 = count cars going both up and down
    ----------
"""
window = Tk()

window.title("Traffic Analyzer")
 
window.geometry('1024x768')
up_down = 0
up = 0
down = 0

lbl = Label(window, text="Please type websites which you want to process")
lbl1 = Label(window, text="What would you like to name the results?")
lbl2 = Label(window, text="Would up like to display cars going down only, up only, or both?")
lbl3 = Label(window, text="Live Feed")

lbl4 = Label(window, text="Up: ")
lbl5 = Label(window, text="Down: ")
lbl6 = Label(window, text="Up: ")
lbl7 = Label(window, text="Down: ")

up1 = Label(window, text="0")
down1 = Label(window, text="0")
up2 = Label(window, text="0")
down2 = Label(window, text="0")

lbl.grid(column=0, row=0)
lbl1.grid(column=0, row=3)
lbl2.grid(column=0, row=4)

lbl3.grid(column=0, row=5)
lbl4.grid(column=0, row=6)
lbl5.grid(column=1, row=6)
up1.grid(column=0, row=7)
down1.grid(column=1, row=7)

lbl6.grid(column=0, row=8)
lbl7.grid(column=1, row=8)
up2.grid(column=0, row=9)
down2.grid(column=1, row=9)

txt = Entry(window,width=20)
txt1 = Entry(window,width=20)
#txt2 = Entry(window,width=20)
txt3 = Entry(window,width=10)

txt.grid(column=1, row=0)
txt1.grid(column=1, row=1)
#txt2.grid(column=1, row=2)
txt3.grid(column=1, row=3)
 


def clicked():    
    global up
    global down

    result_name = "" + txt3.get()

    strTime = run("" + txt.get())
    os.chdir('/Users/paulyp123/Desktop/vision-master/'+strTime)
    #execute('/Users/paulyp123/Desktop/vision-master/'+strTime, "video.avi", result_name, up_down)

    tup = execute('/Users/paulyp123/Desktop/vision-master/'+strTime, "video.avi", result_name, up_down)
    if(up_down == 2):
        up += tup[0]
        down += tup[-1]
    elif(up_down == 1):
        up += tup[0]
    else:
        down += tup[-1]
    up1.configure(text=str(up))
    down1.configure(text=str(down))
    window.update()
    clicked()


def clicked1():
 
    #purpose: sets up_down to 1, which means program will only calculate 
    #cars going up

    global up_down
    up_down = 1

    

def clicked2():

    #purpose: sets up_down to 2, which means program will calculate 
    #cars going up and down

    global up_down
    up_down = 2

def clicked3():
 
    #purpose: sets up_down to 0, which means program will only calculate 
    #cars going down
    
    global up_down
    up_down = 0

    

rad1 = Radiobutton(window,text='Down', value=1, command=clicked3)
rad2 = Radiobutton(window,text='Up', value=2, command=clicked1)
rad3 = Radiobutton(window,text='Both', value=3, command=clicked2)

rad1.grid(column=1, row=4)
rad2.grid(column=2, row=4)
rad3.grid(column=3, row=4)

btn = Button(window, text="Run", command=clicked) 
btn.grid(column=1, row=10)

 
window.mainloop()