Update app.py

app.py

@@ -6,10 +6,11 @@ from simple_salesforce import Salesforce
 from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseDownload
+import tempfile
 
 # Google Drive Authentication Setup
 SCOPES = ['https://www.googleapis.com/auth/drive']
-SERVICE_ACCOUNT_FILE = r"C:\Users\bharg\Documents\Work\Sathkrutha\service_account.json"
+SERVICE_ACCOUNT_FILE = 'service_account.json'  # Ensure this file is uploaded to the Hugging Face space
 credentials = service_account.Credentials.from_service_account_file(
     SERVICE_ACCOUNT_FILE, scopes=SCOPES)
 drive_service = build('drive', 'v3', credentials=credentials)
@@ -22,13 +23,11 @@ sf = Salesforce(
     domain='login'
 )
 
-# Local folder to save retrieved files
-LOCAL_FOLDER = r"C:\Users\bharg\Documents\Work\Sathkrutha\Retrieval"
-if not os.path.exists(LOCAL_FOLDER):
-    os.makedirs(LOCAL_FOLDER)
+# Using a temporary directory to save retrieved files
+temp_dir = tempfile.mkdtemp()
 
+# Function to retrieve all files from the Salesforce custom object
 def get_all_salesforce_files(custom_object_api_name):
-    custom_object_api_name = 'MyCustomer__c'
     # Step 1: Query all record IDs from the custom object
     query_all_records = f"SELECT Id FROM {custom_object_api_name}"
     all_records = sf.query(query_all_records)
@@ -46,9 +45,10 @@ def get_all_salesforce_files(custom_object_api_name):
     
     return all_files
 
+# Function to download a file from Google Drive and save it to a temporary location
 def download_google_drive_file(file_id, file_name):
     request = drive_service.files().get_media(fileId=file_id)
-    file_path = os.path.join(LOCAL_FOLDER, file_name)
+    file_path = os.path.join(temp_dir, file_name)
     with open(file_path, 'wb') as f:
         downloader = MediaIoBaseDownload(f, request)
         done = False
@@ -58,6 +58,7 @@ def download_google_drive_file(file_id, file_name):
                 print(f"Download progress: {int(status.progress() * 100)}%")
     return file_path
 
+# Function to save all files linked to records from the Salesforce custom object
 def save_salesforce_files(custom_object_api_name):
     files = get_all_salesforce_files(custom_object_api_name)
     for file in files: