Update google_drive_upload.py

google_drive_upload.py

@@ -9,22 +9,22 @@ SCOPES = ['https://www.googleapis.com/auth/drive.file']
 
 def authenticate_google_drive():
     """Autentica usando una cuenta de servicio."""
-    # Cargar las credenciales desde el secreto
     service_account_info = json.loads(os.getenv('GOOGLE_SERVICE_ACCOUNT', '{}'))
     if not service_account_info:
         raise ValueError("No se encontró la información de la cuenta de servicio.")
-
-    # Crear credenciales
     creds = Credentials.from_service_account_info(service_account_info, scopes=SCOPES)
     return creds
 
-def upload_to_google_drive(file_path):
+def upload_to_google_drive(file_path, folder_id=None):
     """Sube un archivo a Google Drive."""
     try:
         creds = authenticate_google_drive()
         service = build('drive', 'v3', credentials=creds)
 
         file_metadata = {'name': os.path.basename(file_path)}
+        if folder_id:
+            file_metadata['parents'] = [folder_id]  # Especifica la carpeta de destino
+
         media = MediaFileUpload(file_path, resumable=True)
         file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
         print(f"Archivo subido con ID: {file.get('id')}")