Update app.py

app.py

@@ -1,67 +1,54 @@
-import streamlit as st
-from PIL import Image
-from deepface import DeepFace
+import requests
 import os
 from googleapiclient.discovery import build
-from google.oauth2 import service_account
-import io
+from google.oauth2.service_account import Credentials
 
-# Google Drive API setup
+# Define the 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 = '1DGeRqRbCPcfLDdEgP0h5fyX-MF8EQ8AH'
-
-# 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!")
+# URL of the service account JSON file on Hugging Face
+HUGGINGFACE_JSON_URL = 'https://huggingface.co/username/repository-name/resolve/main/service_account.json'
+
+def download_service_account_file():
+    """Download the service account JSON file from Hugging Face."""
+    if not os.path.exists(SERVICE_ACCOUNT_FILE):  # Check if file already exists
+        print("Downloading service account file from Hugging Face...")
+        response = requests.get(HUGGINGFACE_JSON_URL)
+        if response.status_code == 200:
+            with open(SERVICE_ACCOUNT_FILE, 'wb') as f:
+                f.write(response.content)
+            print(f"Service account file downloaded: {SERVICE_ACCOUNT_FILE}")
+        else:
+            raise Exception(f"Failed to download service account file: {response.status_code}, {response.text}")
+    else:
+        print(f"Service account file already exists: {SERVICE_ACCOUNT_FILE}")
+
+# Download the service account file
+download_service_account_file()
+
+# Authenticate with the Google API
+def authenticate_google_drive():
+    """Authenticate and return the Google Drive service object."""
+    credentials = Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+    drive_service = build('drive', 'v3', credentials=credentials)
+    print("Google Drive API authenticated successfully!")
+    return drive_service
+
+# Example usage of the authenticated service
+drive_service = authenticate_google_drive()
+
+# Example API call: List files in Google Drive
+def list_drive_files():
+    """List files in Google Drive."""
+    results = drive_service.files().list(pageSize=10).execute()
+    items = results.get('files', [])
+    if not items:
+        print("No files found.")
     else:
-        st.error("Authentication failed. ID not verified.")
+        print("Files:")
+        for item in items:
+            print(f"{item['name']} ({item['id']})")
 
+# Call the function to list files
+list_drive_files()