google-workspace-mcp 1.0.5__py3-none-any.whl → 1.2.0__py3-none-any.whl

google_workspace_mcp/services/slides.py

@@ -37,7 +37,11 @@ class SlidesService(BaseGoogleService):
             Presentation data dictionary or error information
         """
         try:
-            return self.service.presentations().get(presentationId=presentation_id).execute()
+            return (
+                self.service.presentations()
+                .get(presentationId=presentation_id)
+                .execute()
+            )
         except Exception as e:
             return self.handle_api_error("get_presentation", e)
 
@@ -57,7 +61,9 @@ class SlidesService(BaseGoogleService):
         except Exception as e:
             return self.handle_api_error("create_presentation", e)
 
-    def create_slide(self, presentation_id: str, layout: str = "TITLE_AND_BODY") -> dict[str, Any]:
+    def create_slide(
+        self, presentation_id: str, layout: str = "TITLE_AND_BODY"
+    ) -> dict[str, Any]:
         """
         Add a new slide to an existing presentation.
 
@@ -80,11 +86,17 @@ class SlidesService(BaseGoogleService):
                 }
             ]
 
-            logger.info(f"Sending API request to create slide: {json.dumps(requests[0], indent=2)}")
+            logger.info(
+                f"Sending API request to create slide: {json.dumps(requests[0], indent=2)}"
+            )
 
             # Execute the request
             response = (
-                self.service.presentations().batchUpdate(presentationId=presentation_id, body={"requests": requests}).execute()
+                self.service.presentations()
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": requests}
+                )
+                .execute()
             )
 
             logger.info(f"API response: {json.dumps(response, indent=2)}")
@@ -161,12 +173,20 @@ class SlidesService(BaseGoogleService):
                 },
             ]
 
-            logger.info(f"Sending API request to create shape: {json.dumps(requests[0], indent=2)}")
-            logger.info(f"Sending API request to insert text: {json.dumps(requests[1], indent=2)}")
+            logger.info(
+                f"Sending API request to create shape: {json.dumps(requests[0], indent=2)}"
+            )
+            logger.info(
+                f"Sending API request to insert text: {json.dumps(requests[1], indent=2)}"
+            )
 
             # Execute the request
             response = (
-                self.service.presentations().batchUpdate(presentationId=presentation_id, body={"requests": requests}).execute()
+                self.service.presentations()
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": requests}
+                )
+                .execute()
             )
 
             logger.info(f"API response: {json.dumps(response, indent=2)}")
@@ -205,7 +225,9 @@ class SlidesService(BaseGoogleService):
             Response data or error information
         """
         try:
-            logger.info(f"Adding formatted text to slide {slide_id}, position={position}, size={size}")
+            logger.info(
+                f"Adding formatted text to slide {slide_id}, position={position}, size={size}"
+            )
             logger.info(f"Text content: '{formatted_text[:100]}...'")
             logger.info(
                 f"Checking for formatting: bold={'**' in formatted_text}, italic={'*' in formatted_text}, code={'`' in formatted_text}"
@@ -240,17 +262,23 @@ class SlidesService(BaseGoogleService):
             ]
 
             # Log the shape creation request
-            logger.info(f"Sending API request to create shape: {json.dumps(create_requests[0], indent=2)}")
+            logger.info(
+                f"Sending API request to create shape: {json.dumps(create_requests[0], indent=2)}"
+            )
 
             # Execute creation request
             creation_response = (
                 self.service.presentations()
-                .batchUpdate(presentationId=presentation_id, body={"requests": create_requests})
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": create_requests}
+                )
                 .execute()
             )
 
             # Log the response
-            logger.info(f"API response for shape creation: {json.dumps(creation_response, indent=2)}")
+            logger.info(
+                f"API response for shape creation: {json.dumps(creation_response, indent=2)}"
+            )
 
             # Process the formatted text
             # First, remove formatting markers to get plain text
@@ -274,7 +302,9 @@ class SlidesService(BaseGoogleService):
             ]
 
             # Log the text insertion request
-            logger.info(f"Sending API request to insert plain text: {json.dumps(text_request[0], indent=2)}")
+            logger.info(
+                f"Sending API request to insert plain text: {json.dumps(text_request[0], indent=2)}"
+            )
 
             # Execute text insertion
             text_response = (
@@ -287,7 +317,9 @@ class SlidesService(BaseGoogleService):
             )
 
             # Log the response
-            logger.info(f"API response for plain text insertion: {json.dumps(text_response, indent=2)}")
+            logger.info(
+                f"API response for plain text insertion: {json.dumps(text_response, indent=2)}"
+            )
 
             # Now generate style requests if there's formatting to apply
             if "**" in formatted_text or "*" in formatted_text:
@@ -364,9 +396,13 @@ class SlidesService(BaseGoogleService):
                 if style_requests:
                     try:
                         # Log the style requests
-                        logger.info(f"Sending API request to apply text styling with {len(style_requests)} style requests")
+                        logger.info(
+                            f"Sending API request to apply text styling with {len(style_requests)} style requests"
+                        )
                         for i, req in enumerate(style_requests):
-                            logger.info(f"Style request {i + 1}: {json.dumps(req, indent=2)}")
+                            logger.info(
+                                f"Style request {i + 1}: {json.dumps(req, indent=2)}"
+                            )
 
                         # Execute style requests
                         style_response = (
@@ -379,9 +415,13 @@ class SlidesService(BaseGoogleService):
                         )
 
                         # Log the response
-                        logger.info(f"API response for text styling: {json.dumps(style_response, indent=2)}")
+                        logger.info(
+                            f"API response for text styling: {json.dumps(style_response, indent=2)}"
+                        )
                     except Exception as style_error:
-                        logger.warning(f"Failed to apply text styles: {str(style_error)}")
+                        logger.warning(
+                            f"Failed to apply text styles: {str(style_error)}"
+                        )
                         logger.exception("Style application error details")
 
             return {
@@ -440,7 +480,9 @@ class SlidesService(BaseGoogleService):
                     },
                 }
             }
-            logger.info(f"Sending API request to create shape for bullet list: {json.dumps(log_data, indent=2)}")
+            logger.info(
+                f"Sending API request to create shape for bullet list: {json.dumps(log_data, indent=2)}"
+            )
 
             # Create requests
             requests = [
@@ -476,15 +518,23 @@ class SlidesService(BaseGoogleService):
             ]
 
             # Log the text insertion
-            logger.info(f"Sending API request to insert bullet text: {json.dumps(requests[1], indent=2)}")
+            logger.info(
+                f"Sending API request to insert bullet text: {json.dumps(requests[1], indent=2)}"
+            )
 
             # Execute the request to create shape and insert text
             response = (
-                self.service.presentations().batchUpdate(presentationId=presentation_id, body={"requests": requests}).execute()
+                self.service.presentations()
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": requests}
+                )
+                .execute()
             )
 
             # Log the response
-            logger.info(f"API response for bullet list creation: {json.dumps(response, indent=2)}")
+            logger.info(
+                f"API response for bullet list creation: {json.dumps(response, indent=2)}"
+            )
 
             # Now add bullet formatting
             try:
@@ -493,14 +543,18 @@ class SlidesService(BaseGoogleService):
                     {
                         "createParagraphBullets": {
                             "objectId": element_id,
-                            "textRange": {"type": "ALL"},  # Apply to all text in the shape
+                            "textRange": {
+                                "type": "ALL"
+                            },  # Apply to all text in the shape
                             "bulletPreset": "BULLET_DISC_CIRCLE_SQUARE",
                         }
                     }
                 ]
 
                 # Log the bullet formatting request
-                logger.info(f"Sending API request to apply bullet formatting: {json.dumps(bullet_request[0], indent=2)}")
+                logger.info(
+                    f"Sending API request to apply bullet formatting: {json.dumps(bullet_request[0], indent=2)}"
+                )
 
                 bullet_response = (
                     self.service.presentations()
@@ -512,9 +566,13 @@ class SlidesService(BaseGoogleService):
                 )
 
                 # Log the response
-                logger.info(f"API response for bullet formatting: {json.dumps(bullet_response, indent=2)}")
+                logger.info(
+                    f"API response for bullet formatting: {json.dumps(bullet_response, indent=2)}"
+                )
             except Exception as bullet_error:
-                logger.warning(f"Failed to apply bullet formatting: {str(bullet_error)}")
+                logger.warning(
+                    f"Failed to apply bullet formatting: {str(bullet_error)}"
+                )
                 # No fallback here - the text is already added, just without bullets
 
             return {
@@ -527,7 +585,9 @@ class SlidesService(BaseGoogleService):
         except Exception as e:
             return self.handle_api_error("add_bulleted_list", e)
 
-    def create_presentation_from_markdown(self, title: str, markdown_content: str) -> dict[str, Any]:
+    def create_presentation_from_markdown(
+        self, title: str, markdown_content: str
+    ) -> dict[str, Any]:
         """
         Create a Google Slides presentation from Markdown content using markdowndeck.
 
@@ -545,9 +605,13 @@ class SlidesService(BaseGoogleService):
             credentials = gauth.get_credentials()
 
             # Use markdowndeck to create the presentation
-            result = create_presentation(markdown=markdown_content, title=title, credentials=credentials)
+            result = create_presentation(
+                markdown=markdown_content, title=title, credentials=credentials
+            )
 
-            logger.info(f"Successfully created presentation with ID: {result.get('presentationId')}")
+            logger.info(
+                f"Successfully created presentation with ID: {result.get('presentationId')}"
+            )
 
             # The presentation data is already in the expected format from markdowndeck
             return result
@@ -568,7 +632,11 @@ class SlidesService(BaseGoogleService):
         """
         try:
             # Get the presentation with slide details
-            presentation = self.service.presentations().get(presentationId=presentation_id).execute()
+            presentation = (
+                self.service.presentations()
+                .get(presentationId=presentation_id)
+                .execute()
+            )
 
             # Extract slide information
             slides = []
@@ -587,9 +655,13 @@ class SlidesService(BaseGoogleService):
                         if "textElements" in element["shape"]["text"]:
                             # Extract text content
                             text_parts = []
-                            for text_element in element["shape"]["text"]["textElements"]:
+                            for text_element in element["shape"]["text"][
+                                "textElements"
+                            ]:
                                 if "textRun" in text_element:
-                                    text_parts.append(text_element["textRun"].get("content", ""))
+                                    text_parts.append(
+                                        text_element["textRun"].get("content", "")
+                                    )
                             element_content = "".join(text_parts)
                     elif "image" in element:
                         element_type = "image"
@@ -597,9 +669,7 @@ class SlidesService(BaseGoogleService):
                             element_content = element["image"]["contentUrl"]
                     elif "table" in element:
                         element_type = "table"
-                        element_content = (
-                            f"Table with {element['table'].get('rows', 0)} rows, {element['table'].get('columns', 0)} columns"
-                        )
+                        element_content = f"Table with {element['table'].get('rows', 0)} rows, {element['table'].get('columns', 0)} columns"
 
                     # Add to elements if we found content
                     if element_type and element_content:
@@ -613,7 +683,10 @@ class SlidesService(BaseGoogleService):
 
                 # Get speaker notes if present
                 notes = ""
-                if "slideProperties" in slide and "notesPage" in slide["slideProperties"]:
+                if (
+                    "slideProperties" in slide
+                    and "notesPage" in slide["slideProperties"]
+                ):
                     notes_page = slide["slideProperties"]["notesPage"]
                     if "pageElements" in notes_page:
                         for element in notes_page["pageElements"]:
@@ -623,9 +696,13 @@ class SlidesService(BaseGoogleService):
                                 and "textElements" in element["shape"]["text"]
                             ):
                                 note_parts = []
-                                for text_element in element["shape"]["text"]["textElements"]:
+                                for text_element in element["shape"]["text"][
+                                    "textElements"
+                                ]:
                                     if "textRun" in text_element:
-                                        note_parts.append(text_element["textRun"].get("content", ""))
+                                        note_parts.append(
+                                            text_element["textRun"].get("content", "")
+                                        )
                                 if note_parts:
                                     notes = "".join(note_parts)
 
@@ -657,14 +734,22 @@ class SlidesService(BaseGoogleService):
             # Define the delete request
             requests = [{"deleteObject": {"objectId": slide_id}}]
 
-            logger.info(f"Sending API request to delete slide: {json.dumps(requests[0], indent=2)}")
+            logger.info(
+                f"Sending API request to delete slide: {json.dumps(requests[0], indent=2)}"
+            )
 
             # Execute the request
             response = (
-                self.service.presentations().batchUpdate(presentationId=presentation_id, body={"requests": requests}).execute()
+                self.service.presentations()
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": requests}
+                )
+                .execute()
             )
 
-            logger.info(f"API response for slide deletion: {json.dumps(response, indent=2)}")
+            logger.info(
+                f"API response for slide deletion: {json.dumps(response, indent=2)}"
+            )
 
             return {
                 "presentationId": presentation_id,
@@ -724,7 +809,9 @@ class SlidesService(BaseGoogleService):
                     "height": {"magnitude": size[1], "unit": "PT"},
                 }
 
-            logger.info(f"Sending API request to create image: {json.dumps(create_image_request, indent=2)}")
+            logger.info(
+                f"Sending API request to create image: {json.dumps(create_image_request, indent=2)}"
+            )
 
             # Execute the request
             response = (
@@ -739,7 +826,9 @@ class SlidesService(BaseGoogleService):
             # Extract the image ID from the response
             if "replies" in response and len(response["replies"]) > 0:
                 image_id = response["replies"][0].get("createImage", {}).get("objectId")
-                logger.info(f"API response for image creation: {json.dumps(response, indent=2)}")
+                logger.info(
+                    f"API response for image creation: {json.dumps(response, indent=2)}"
+                )
                 return {
                     "presentationId": presentation_id,
                     "slideId": slide_id,
@@ -804,7 +893,9 @@ class SlidesService(BaseGoogleService):
                 }
             }
 
-            logger.info(f"Sending API request to create table: {json.dumps(create_table_request, indent=2)}")
+            logger.info(
+                f"Sending API request to create table: {json.dumps(create_table_request, indent=2)}"
+            )
 
             # Execute table creation
             response = (
@@ -816,7 +907,9 @@ class SlidesService(BaseGoogleService):
                 .execute()
             )
 
-            logger.info(f"API response for table creation: {json.dumps(response, indent=2)}")
+            logger.info(
+                f"API response for table creation: {json.dumps(response, indent=2)}"
+            )
 
             # Populate the table if data is provided
             if data:
@@ -841,7 +934,9 @@ class SlidesService(BaseGoogleService):
                             )
 
                 if text_requests:
-                    logger.info(f"Sending API request to populate table with {len(text_requests)} cell entries")
+                    logger.info(
+                        f"Sending API request to populate table with {len(text_requests)} cell entries"
+                    )
                     table_text_response = (
                         self.service.presentations()
                         .batchUpdate(
@@ -850,7 +945,9 @@ class SlidesService(BaseGoogleService):
                         )
                         .execute()
                     )
-                    logger.info(f"API response for table population: {json.dumps(table_text_response, indent=2)}")
+                    logger.info(
+                        f"API response for table population: {json.dumps(table_text_response, indent=2)}"
+                    )
 
             return {
                 "presentationId": presentation_id,
@@ -891,14 +988,22 @@ class SlidesService(BaseGoogleService):
                 }
             ]
 
-            logger.info(f"Sending API request to add slide notes: {json.dumps(requests[0], indent=2)}")
+            logger.info(
+                f"Sending API request to add slide notes: {json.dumps(requests[0], indent=2)}"
+            )
 
             # Execute the request
             response = (
-                self.service.presentations().batchUpdate(presentationId=presentation_id, body={"requests": requests}).execute()
+                self.service.presentations()
+                .batchUpdate(
+                    presentationId=presentation_id, body={"requests": requests}
+                )
+                .execute()
             )
 
-            logger.info(f"API response for slide notes: {json.dumps(response, indent=2)}")
+            logger.info(
+                f"API response for slide notes: {json.dumps(response, indent=2)}"
+            )
 
             return {
                 "presentationId": presentation_id,
@@ -909,7 +1014,9 @@ class SlidesService(BaseGoogleService):
         except Exception as e:
             return self.handle_api_error("add_slide_notes", e)
 
-    def duplicate_slide(self, presentation_id: str, slide_id: str, insert_at_index: int | None = None) -> dict[str, Any]:
+    def duplicate_slide(
+        self, presentation_id: str, slide_id: str, insert_at_index: int | None = None
+    ) -> dict[str, Any]:
         """
         Duplicate a slide in a presentation.
 
@@ -929,7 +1036,9 @@ class SlidesService(BaseGoogleService):
             if insert_at_index is not None:
                 duplicate_request["duplicateObject"]["insertionIndex"] = insert_at_index
 
-            logger.info(f"Sending API request to duplicate slide: {json.dumps(duplicate_request, indent=2)}")
+            logger.info(
+                f"Sending API request to duplicate slide: {json.dumps(duplicate_request, indent=2)}"
+            )
 
             # Execute the duplicate request
             response = (
@@ -941,12 +1050,16 @@ class SlidesService(BaseGoogleService):
                 .execute()
             )
 
-            logger.info(f"API response for slide duplication: {json.dumps(response, indent=2)}")
+            logger.info(
+                f"API response for slide duplication: {json.dumps(response, indent=2)}"
+            )
 
             # Extract the duplicated slide ID
             new_slide_id = None
             if "replies" in response and len(response["replies"]) > 0:
-                new_slide_id = response["replies"][0].get("duplicateObject", {}).get("objectId")
+                new_slide_id = (
+                    response["replies"][0].get("duplicateObject", {}).get("objectId")
+                )
 
             return {
                 "presentationId": presentation_id,
@@ -957,3 +1070,79 @@ class SlidesService(BaseGoogleService):
             }
         except Exception as e:
             return self.handle_api_error("duplicate_slide", e)
+
+    def embed_sheets_chart(
+        self,
+        presentation_id: str,
+        slide_id: str,
+        spreadsheet_id: str,
+        chart_id: int,
+        position: tuple[float, float],
+        size: tuple[float, float],
+    ) -> dict[str, Any]:
+        """
+        Embeds a chart from Google Sheets into a Google Slides presentation.
+
+        Args:
+            presentation_id: The ID of the presentation.
+            slide_id: The ID of the slide to add the chart to.
+            spreadsheet_id: The ID of the Google Sheet containing the chart.
+            chart_id: The ID of the chart within the sheet.
+            position: Tuple of (x, y) coordinates for position in PT.
+            size: Tuple of (width, height) for the chart size in PT.
+
+        Returns:
+            The API response from the batchUpdate call.
+        """
+        try:
+            element_id = f"chart_{chart_id}_{int(__import__('time').time() * 1000)}"
+            logger.info(
+                f"Embedding chart {chart_id} from sheet {spreadsheet_id} into slide {slide_id}"
+            )
+
+            requests = [
+                {
+                    "createSheetsChart": {
+                        "objectId": element_id,
+                        "spreadsheetId": spreadsheet_id,
+                        "chartId": chart_id,
+                        "linkingMode": "LINKED",
+                        "elementProperties": {
+                            "pageObjectId": slide_id,
+                            "size": {
+                                "width": {"magnitude": size[0], "unit": "PT"},
+                                "height": {"magnitude": size[1], "unit": "PT"},
+                            },
+                            "transform": {
+                                "scaleX": 1,
+                                "scaleY": 1,
+                                "translateX": position[0],
+                                "translateY": position[1],
+                                "unit": "PT",
+                            },
+                        },
+                    }
+                }
+            ]
+
+            response = self.batch_update(presentation_id, requests)
+            if response.get("error"):
+                raise ValueError(
+                    response.get("message", "Batch update for chart embedding failed")
+                )
+
+            created_element_id = (
+                response.get("replies", [{}])[0]
+                .get("createSheetsChart", {})
+                .get("objectId")
+            )
+
+            return {
+                "success": True,
+                "presentation_id": presentation_id,
+                "slide_id": slide_id,
+                "element_id": created_element_id or element_id,
+            }
+
+        except Exception as e:
+            return self.handle_api_error("embed_sheets_chart", e)