Update app.py

app.py

@@ -9,14 +9,13 @@ from PIL import Image
 KNOWN_FACES_DIR = "known_faces"
 os.makedirs(KNOWN_FACES_DIR, exist_ok=True)
 
-# Retrieve the token from the environment
-hf_token = os.getenv("token")  # Ensure the key matches the name in Hugging Face secrets
+# Retrieve the Hugging Face token from the environment
+hf_token = os.getenv("token")  # Ensure the key matches the secret name
 if not hf_token:
     raise ValueError("Hugging Face token not found. Ensure it's set as a secret in the Hugging Face Space.")
 
 # Initialize the Hugging Face API
 api = HfApi()
-api.set_access_token(hf_token)
 
 # Database setup
 DATABASE = "students.db"
@@ -63,7 +62,8 @@ def save_image_to_hugging_face(image, roll_no):
             path_or_fileobj=local_path,
             path_in_repo=filename,
             repo_id=repo_name,
-            repo_type="dataset"
+            repo_type="dataset",
+            token=hf_token  # Pass the token directly
         )
         st.success(f"Image uploaded to Hugging Face: {filename}")
     except Exception as e:
@@ -85,7 +85,11 @@ roll_no = st.text_input("Enter your roll number")
 capture_mode = st.radio("Choose an option to upload your image", ["Use Webcam", "Upload File"])
 
 if capture_mode == "Use Webcam":
-    picture = st.camera_input("Take a picture")
+    try:
+        picture = st.camera_input("Take a picture")
+    except Exception as e:
+        st.error(f"Error accessing webcam: {e}")
+        picture = None
 
 elif capture_mode == "Upload File":
     picture = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
@@ -98,9 +102,9 @@ if st.button("Register"):
         st.error("Please upload or capture an image.")
     else:
         try:
-            if capture_mode == "Use Webcam":
+            if capture_mode == "Use Webcam" and picture:
                 image = Image.open(picture)
-            else:
+            elif capture_mode == "Upload File" and picture:
                 image = Image.open(picture)
             
             # Save locally and upload to Hugging Face