Update app.py

app.py

@@ -29,6 +29,16 @@ for cls in myList:
 st.write(classnames)
 
 
+json_file = open("facialemotionmodel.json", "r")
+model_json = json_file.read()
+json_file.close()
+model = model_from_json(model_json)
+
+model.load_weights("facialemotionmodel.h5")
+haar_file=cv2.data.haarcascades + 'haarcascade_frontalface_default.xml'
+face_cascade=cv2.CascadeClassifier(haar_file)
+
+
 def findEncodings(Images):
     encodeList = []
     for img in Images:
@@ -37,6 +47,12 @@ def findEncodings(Images):
         encodeList.append(encode)
     return encodeList
 
+def extract_features(image):
+    feature = np.array(image)
+    feature = feature.reshape(1,48,48,1)
+    return feature/255.0
+
+labels = {0 : 'angry', 1 : 'disgust', 2 : 'fear', 3 : 'happy', 4 : 'neutral', 5 : 'sad', 6 : 'surprise'}
 
 encodeListknown = findEncodings(Images)
 st.write('Encoding Complete')
@@ -69,12 +85,29 @@ if img_file_buffer is not None:
             cv2.rectangle(image,(x1,y1),(x2,y2),(0,255,0),2)
             cv2.rectangle(image,(x1,y2-35),(x2,y2),(0,255,0),cv2.FILLED)
             cv2.putText(image,name,(x1+6,y2-6),cv2.FONT_HERSHEY_COMPLEX,1,(255, 255, 255),2)
+
+            faces=face_cascade.detectMultiScale(image,1.3,5)
+            st.write("close")
+            try: 
+                for (p,q,r,s) in faces:
+                    
+                    cv2.rectangle(im,(p,q),(p+r,q+s),(255,0,0),2)
+                    image = cv2.resize(image,(48,48))
+                    img = extract_features(image)
+                    pred = model.predict(img)
+                    prediction_label = labels[pred.argmax()]
+                    # print("Predicted Output:", prediction_label)
+                    # cv2.putText(im,prediction_label)
+                    cv2.putText(image, '% s' %(prediction_label), (p-10, q-10),cv2.FONT_HERSHEY_COMPLEX_SMALL,2, (0,0,255))
+                    st.write("success")
+            
+             
            
-            ##############
-            url = "https://kiwi-whispering-plier.glitch.me/update" 
+            # ##############
+            # url = "https://kiwi-whispering-plier.glitch.me/update" 
             
-            data = {
-                'name': name,
-            }
-            else:
-                st.write("Please smile")
+            # data = {
+            #     'name': name,
+            # }
+            # else:
+            #     st.write("Please smile")