Update app.py

app.py

@@ -0,0 +1,67 @@
+import streamlit as st
+from PIL import Image
+from deepface import DeepFace
+import os
+from googleapiclient.discovery import build
+from google.oauth2 import service_account
+import io
+
+# Google Drive API setup
+SCOPES = ['https://www.googleapis.com/auth/drive']
+SERVICE_ACCOUNT_FILE = 'service_account.json'
+
+credentials = service_account.Credentials.from_service_account_file(
+    SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+drive_service = build('drive', 'v3', credentials=credentials)
+REFERENCE_FOLDER_ID = 'your_google_drive_folder_id'
+
+# Helper function to download reference files
+def list_reference_files(folder_id):
+    query = f"'{folder_id}' in parents and trashed=false"
+    results = drive_service.files().list(q=query, fields="files(id, name)").execute()
+    return results.get("files", [])
+
+def download_file(file_id, file_name):
+    request = drive_service.files().get_media(fileId=file_id)
+    file_io = io.BytesIO()
+    downloader = MediaIoBaseDownload(file_io, request)
+    done = False
+    while not done:
+        _, done = downloader.next_chunk()
+    os.makedirs("reference_images", exist_ok=True)
+    file_path = os.path.join("reference_images", file_name)
+    with open(file_path, "wb") as f:
+        f.write(file_io.getvalue())
+    return file_path
+
+# Streamlit app
+st.title("Face Authentication Demo")
+st.write("Upload a selfie to authenticate.")
+
+uploaded_file = st.file_uploader("Choose a selfie", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    selfie_path = os.path.join("static", "uploads", uploaded_file.name)
+    os.makedirs(os.path.dirname(selfie_path), exist_ok=True)
+    with open(selfie_path, "wb") as f:
+        f.write(uploaded_file.getbuffer())
+
+    # Download reference images from Google Drive
+    st.write("Downloading reference images...")
+    reference_files = list_reference_files(REFERENCE_FOLDER_ID)
+    reference_paths = [download_file(f['id'], f['name']) for f in reference_files]
+
+    # Compare selfie with reference images
+    st.write("Authenticating...")
+    authenticated = False
+    for ref_path in reference_paths:
+        if DeepFace.verify(img1_path=ref_path, img2_path=selfie_path,
+                           model_name="ArcFace", enforce_detection=False)['verified']:
+            authenticated = True
+            break
+
+    if authenticated:
+        st.success("Authentication successful!")
+    else:
+        st.error("Authentication failed. ID not verified.")
+