Update to 20-Emotion Classifier with Optimized Models

- Expanded from 10 to 20 emotions
- Optimized Word2Vec model
- Updated neural network
- Enhanced UI with all 20 emotions

.gitattributes

@@ -1,37 +1,4 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.keras filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
 *.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
-models/best_model.keras filter=lfs diff=lfs merge=lfs -text
-models/final_model.keras filter=lfs diff=lfs merge=lfs -text
+*.csv filter=lfs diff=lfs merge=lfs -text

app.py

@@ -28,8 +28,8 @@ app = Flask(__name__)
 print("Loading emotion classification models...")
 
 try:
-    # Load Word2Vec model
-    w2v_model = Word2Vec.load('models/word2vec.model')
+    # Load Word2Vec model (optimized version)
+    w2v_model = Word2Vec.load('models/word2vec_optimized.model')
 
     # Load neural network classifier (without compile to avoid optimizer issues)
     classifier = keras.models.load_model('models/best_model.keras', compile=False)
@@ -49,7 +49,7 @@ except Exception as e:
     print("Please run the training scripts first (02-06.py files)")
     MODELS_LOADED = False
 
-# Emotion colors and emojis for UI
+# Emotion colors and emojis for UI (20 emotions)
 EMOTION_CONFIG = {
     'happiness': {
         'color': '#FFD700',
@@ -66,31 +66,21 @@ EMOTION_CONFIG = {
         'emoji': 'O_O',
         'gradient': 'linear-gradient(135deg, #9370DB, #8A2BE2)'
     },
-    'embarrassment': {
-        'color': '#FF69B4',
-        'emoji': '>///<',
-        'gradient': 'linear-gradient(135deg, #FF69B4, #FF1493)'
+    'anger': {
+        'color': '#DC143C',
+        'emoji': '>:(',
+        'gradient': 'linear-gradient(135deg, #DC143C, #B22222)'
     },
     'disgust': {
         'color': '#8B4513',
         'emoji': 'X_X',
         'gradient': 'linear-gradient(135deg, #8B4513, #A0522D)'
     },
-    'drive': {
-        'color': '#FF4500',
-        'emoji': '>:)',
-        'gradient': 'linear-gradient(135deg, #FF4500, #FF6347)'
-    },
     'surprise': {
         'color': '#FFD700',
         'emoji': ':O',
         'gradient': 'linear-gradient(135deg, #FFD700, #FFFF00)'
     },
-    'loneliness': {
-        'color': '#708090',
-        'emoji': '...',
-        'gradient': 'linear-gradient(135deg, #708090, #778899)'
-    },
     'love': {
         'color': '#FF1493',
         'emoji': '<3',
@@ -101,6 +91,66 @@ EMOTION_CONFIG = {
         'emoji': '!!!',
         'gradient': 'linear-gradient(135deg, #FF6347, #FF4500)'
     },
+    'embarrassment': {
+        'color': '#FF69B4',
+        'emoji': '>///<',
+        'gradient': 'linear-gradient(135deg, #FF69B4, #FF1493)'
+    },
+    'loneliness': {
+        'color': '#708090',
+        'emoji': '...',
+        'gradient': 'linear-gradient(135deg, #708090, #778899)'
+    },
+    'anxiety': {
+        'color': '#9370DB',
+        'emoji': ':S',
+        'gradient': 'linear-gradient(135deg, #9370DB, #7B68EE)'
+    },
+    'frustration': {
+        'color': '#FF8C00',
+        'emoji': '>:|',
+        'gradient': 'linear-gradient(135deg, #FF8C00, #FFA500)'
+    },
+    'guilt': {
+        'color': '#696969',
+        'emoji': ':/',
+        'gradient': 'linear-gradient(135deg, #696969, #808080)'
+    },
+    'disappointment': {
+        'color': '#6A5ACD',
+        'emoji': ':-/',
+        'gradient': 'linear-gradient(135deg, #6A5ACD, #7B68EE)'
+    },
+    'jealousy': {
+        'color': '#2E8B57',
+        'emoji': ':-|',
+        'gradient': 'linear-gradient(135deg, #2E8B57, #3CB371)'
+    },
+    'gratitude': {
+        'color': '#98FB98',
+        'emoji': '^_^',
+        'gradient': 'linear-gradient(135deg, #98FB98, #90EE90)'
+    },
+    'pride': {
+        'color': '#FFD700',
+        'emoji': 'B)',
+        'gradient': 'linear-gradient(135deg, #FFD700, #DAA520)'
+    },
+    'relief': {
+        'color': '#87CEEB',
+        'emoji': '*phew*',
+        'gradient': 'linear-gradient(135deg, #87CEEB, #87CEFA)'
+    },
+    'hope': {
+        'color': '#FFA07A',
+        'emoji': ':)',
+        'gradient': 'linear-gradient(135deg, #FFA07A, #FA8072)'
+    },
+    'confusion': {
+        'color': '#D3D3D3',
+        'emoji': '???',
+        'gradient': 'linear-gradient(135deg, #D3D3D3, #C0C0C0)'
+    },
 }
 
 # Preprocessing function

models/best_model.keras

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0661374cf8f18da22e8e0daed3b239139caacddac487d9238d2712117b8b8f37
-size 162337
+oid sha256:49ccf76f4dceab7271800864c9c2ac654bf6530f42d47160febfa5ffb9888c0a
+size 113737

models/final_model.keras

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0661374cf8f18da22e8e0daed3b239139caacddac487d9238d2712117b8b8f37
-size 162337
+oid sha256:49ccf76f4dceab7271800864c9c2ac654bf6530f42d47160febfa5ffb9888c0a
+size 113737

models/label_encoder.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c1ba42b38db8974d15055e9675cbe0581632315fdc8d8b9f53ba6ad56353349e
-size 349
+oid sha256:9b46030c0fef80c7b2a8d9a7d5325a0a0b1f5e187f62ca0658cbf59f4b5d031e
+size 457

models/word2vec_optimized.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e8b89ec7042ba06a86fda6d1ea164a8e883ac62d12053e0b0a6b8811e27dd0
+size 3048421