Create app.py

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+import cv2
+import os
+
+# Define the folder path where images will be saved
+dataset_path = "/home/user/app/dataset"  # Adjust path as needed
+
+# Ensure the directory exists
+if not os.path.exists(dataset_path):
+    os.makedirs(dataset_path)
+
+# Function to capture and save images
+def capture_image(image, name):
+    person_folder = os.path.join(dataset_path, name)
+    os.makedirs(person_folder, exist_ok=True)  # Create a folder for each person if not exists
+    
+    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
+    
+    return f"Image saved for {name} at {image_path}"
+
+# Gradio interface for capturing images
+iface = gr.Interface(
+    fn=capture_image,
+    inputs=["webcam", "text"],
+    outputs="text",
+    description="Enter your name and capture your image for the dataset."
+)
+
+iface.launch()