Update app.py

app.py

@@ -75,24 +75,38 @@ def save_to_database(name, roll_no, image_path):
 
 def save_image_to_hugging_face(image, name, roll_no):
     """ Saves the image locally to the KNOWN_FACES_DIR and uploads it to Hugging Face. """
+    # Ensure the directory exists
+    if not os.path.exists(KNOWN_FACES_DIR):
+        os.makedirs(KNOWN_FACES_DIR)
+
+    # Construct the local file path
     filename = f"{name}_{roll_no}.jpg"
-    local_path = os.path.join(KNOWN_FACES_DIR, filename)  # Ensure it's saved in the known_faces directory
+    local_path = os.path.join(KNOWN_FACES_DIR, filename)
     
     try:
-        # Convert image to RGB if not already in RGB mode (PIL handling)
+        # Convert image to RGB if necessary
         if image.mode != "RGB":
             image = image.convert("RGB")
-        image.save(local_path)  # Save image locally to known_faces directory
         
-        # Upload the file to Hugging Face
-        api.upload_file(path_or_fileobj=local_path, path_in_repo=filename, repo_id=REPO_ID, repo_type="space", token=hf_token)
-        st.success(f"Image uploaded to Hugging Face: {filename}")
+        # Save the image to the known_faces directory
+        image.save(local_path)
+        
+        # Upload the saved file to Hugging Face
+        api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=filename,
+            repo_id=REPO_ID,
+            repo_type="space",
+            token=hf_token,
+        )
+        st.success(f"Image saved to {KNOWN_FACES_DIR} and uploaded to Hugging Face as {filename}.")
     except Exception as e:
-        st.error(f"Error uploading image to Hugging Face: {e}")
+        st.error(f"Error saving or uploading image: {e}")
     
     return local_path
 
 
+
 # Initialize the database when the app starts
 initialize_database()