Update app.py

app.py

@@ -1,31 +1,44 @@
 import gradio as gr
 import cv2
 import os
+import numpy as np
+from deepface import DeepFace
 
 # Define the folder path where images will be saved
-dataset_path = "/home/user/app/dataset"  # Adjust path as needed
-
-# Ensure the directory exists
+dataset_path = "/content/dataset"  # For Colab or your local environment
 if not os.path.exists(dataset_path):
     os.makedirs(dataset_path)
 
-# Function to capture and save images
-def capture_image(image, name):
+# Function to predict emotion and save the image
+def capture_and_recognize(image, name):
+    # Convert Gradio image (PIL format) to OpenCV image
+    img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+
+    # Analyze the emotion using DeepFace
+    result = DeepFace.analyze(img, actions=['emotion'])
+    
+    # Get the dominant emotion
+    dominant_emotion = result[0]['dominant_emotion']
+    
+    # Create a folder for each person if not exists
     person_folder = os.path.join(dataset_path, name)
-    os.makedirs(person_folder, exist_ok=True)  # Create a folder for each person if not exists
+    os.makedirs(person_folder, exist_ok=True)
     
+    # Save the image in the person's folder
     image_count = len(os.listdir(person_folder))
     image_path = os.path.join(person_folder, f"{image_count + 1}.jpg")
-    cv2.imwrite(image_path, cv2.cvtColor(image, cv2.COLOR_RGB2BGR))  # Save image in BGR format for OpenCV
+    cv2.imwrite(image_path, cv2.cvtColor(img, cv2.COLOR_RGB2BGR))  # Save in BGR format for OpenCV
     
-    return f"Image saved for {name} at {image_path}"
+    return f"Image saved for {name} with emotion: {dominant_emotion} at {image_path}"
 
-# Gradio interface for capturing images
+# Define the Gradio interface
 iface = gr.Interface(
-    fn=capture_image,
-    inputs=["webcam", "text"],
-    outputs="text",
-    description="Enter your name and capture your image for the dataset."
+    fn=capture_and_recognize,
+    inputs=[gr.Image(type="pil", source="webcam"), gr.Textbox()],  # Webcam input and Name input
+    outputs="text",  # Text output for saved image path and emotion
+    title="Attendance Image Capture with Emotion Recognition",
+    description="Capture an image via webcam, enter your name, and the image will be saved with emotion recognition."
 )
 
+# Launch the Gradio app
 iface.launch()