test_model.py

# FILEPATH: c:/Users/Acer/MockPitch/Emotion/test_model.py

import pickle
import cv2
import numpy as np
import os
import mediapipe as mp
from utils import get_face_landmarks
import csv
import time

# Get the absolute path to the directory of the current script
script_dir = os.path.dirname(os.path.abspath(__file__))
model_file = os.path.join(script_dir, 'model')

# Load the model
with open(model_file, 'rb') as f:
    model = pickle.load(f)

emotions = ['angry', 'disgust', 'fear', 'happy', 'neutral', 'sad', 'surprise']

# Initialize MediaPipe FaceMesh
face_mesh = mp.solutions.face_mesh.FaceMesh(static_image_mode=False, max_num_faces=1)

cap = cv2.VideoCapture(0)

# Check if the webcam is opened correctly
if not cap.isOpened():
    print("Error: Could not open video stream.")
    print("Please make sure your webcam is not in use by another application.")
    print("If the issue persists, try changing cv2.VideoCapture(0) to cv2.VideoCapture(1).")
    exit()

# Open CSV file for writing
csv_file = open('emotion_output.csv', mode='w', newline='')
csv_writer = csv.writer(csv_file)
csv_writer.writerow(['timestamp', 'emotion'])  # header

while True:
    ret, frame = cap.read()
    if not ret:
        print("Error: Can't receive frame (stream end?). Exiting ...")
        break

    landmarks = get_face_landmarks(frame, face_mesh, draw=False)  # disable drawing landmarks

    if landmarks:
        landmarks = np.array(landmarks).reshape(1, -1)
        emotion_idx = model.predict(landmarks)[0]
        emotion = emotions[int(emotion_idx)]

        # Write timestamp and emotion to CSV and flush immediately
        csv_writer.writerow([time.time(), emotion])
        csv_file.flush()

    # Show clean frame (no text or dots)
    cv2.imshow('Emotion Detection', frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

csv_file.close()
cap.release()
face_mesh.close()
cv2.destroyAllWindows()