Update app.py

app.py

@@ -47,6 +47,24 @@ except Exception as e:
 # Using a temporary directory to save retrieved files
 temp_dir = tempfile.mkdtemp()
 
+def get_all_salesforce_files(custom_object_api_name):
+    if sf is None:
+        print("Salesforce connection is not initialized.")
+        return []
+
+    # Step 1: Query all record IDs from the custom object, including the Google Drive File ID field
+    query_all_records = f"SELECT Id, Google_Fille_ID__c FROM {custom_object_api_name} WHERE Google_Fille_ID__c != NULL"
+    try:
+        all_records = sf.query(query_all_records)
+        if 'records' not in all_records or not all_records['records']:
+            print("No records found or records list is empty.")
+            return []
+        return all_records['records']
+    except Exception as e:
+        print(f"Error querying Salesforce: {e}")
+        return []
+
+
 # Function to retrieve all files from the Salesforce custom object
 def get_all_salesforce_files(custom_object_api_name):
     if sf is None: