NN + DT connection

Signed-off-by: Neerka <kuba.markil0220@gmail.com>

classes/Trash.py

@@ -1,11 +1,14 @@
 from random import choice as rd
-
-import numpy as np
-
 from classes.trashFunctions import generatePicList
 
 
 class Trash:
+    tdic = {"paper": 1,
+            "bio_waste": 0,
+            "mixed": 4,
+            "glass": 3,
+            "metals_and_plastics": 2}
+
     def __init__(self):
         self.tree_type = 0
         self.ttype = "Trash"
@@ -24,7 +27,7 @@ class Trash:
         return self.tree_type
 
     def setTreetype(self, treetype):
-        self.tree_type = id
+        self.tree_type = treetype
         return self
 
     def getTtype(self):
@@ -32,6 +35,7 @@ class Trash:
 
     def setTtype(self, ttype):
         self.ttype = ttype
+        self.setTreetype(self.tdic[ttype])
         return self
 
     def getWaga(self):