Update pipeline.py

pipeline.py

@@ -49,17 +49,43 @@ def get_or_create_drive_folder(name, parent_id=None):
         file_metadata["parents"] = [parent_id]
     file = drive_service.files().create(body=file_metadata, fields="id").execute()
     return file["id"]
+# def find_drive_file(filename, parent_id):
+#     """
+#     Checks if a file with the given name exists inside the specified Google Drive folder.
+#     Returns the file ID if found, else None.
+#     """
+#     query = f"'{parent_id}' in parents and name = '{filename}' and trashed = false"
+#     results = drive_service.files().list(q=query, spaces='drive', fields='files(id, name)', pageSize=1).execute()
+#     files = results.get('files', [])
+#     if files:
+#         return files[0]["id"]
+#     return None
+
 def find_drive_file(filename, parent_id):
     """
     Checks if a file with the given name exists inside the specified Google Drive folder.
     Returns the file ID if found, else None.
     """
-    query = f"'{parent_id}' in parents and name = '{filename}' and trashed = false"
-    results = drive_service.files().list(q=query, spaces='drive', fields='files(id, name)', pageSize=1).execute()
-    files = results.get('files', [])
-    if files:
-        return files[0]["id"]
-    return None
+    try:
+        print(f"🔍 Searching for '{filename}' in folder: {parent_id}")
+        query = f"'{parent_id}' in parents and name = '{filename}' and trashed = false"
+        results = drive_service.files().list(
+            q=query,
+            spaces='drive',
+            fields='files(id, name)',
+            pageSize=1
+        ).execute()
+        files = results.get('files', [])
+        if files:
+            print(f"✅ Found file: {files[0]['name']} with ID: {files[0]['id']}")
+            return files[0]["id"]
+        else:
+            print("⚠️ File not found.")
+            return None
+    except Exception as e:
+        print(f"❌ Error during find_drive_file: {e}")
+        return None
+
 
 
 # def upload_file_to_drive(local_path, remote_name, folder_id):