Spaces:
Sleeping
Sleeping
| import json | |
| import os | |
| from pptx import Presentation | |
| from pptx.util import Inches | |
| from pptx.shapes.group import GroupShape | |
| from pptx.shapes.picture import Picture | |
| from PIL import Image | |
| import io | |
| def print_json(item): | |
| return json.dumps(item, ensure_ascii=False, indent=4) | |
| def transfer_textbox_content_in_group(group_shape): | |
| group_shape_item = {} | |
| for l, shape in enumerate(group_shape.shapes): | |
| shape_item = {} | |
| if shape.has_text_frame: | |
| shape_item['type'] = "text" | |
| shape_item['location'] = (shape.left, shape.top) | |
| text_frame = shape.text_frame | |
| for r, paragraph in enumerate(text_frame.paragraphs): | |
| if paragraph.runs: | |
| original_run = paragraph.runs[0] | |
| paragraph_item = { | |
| 'text': paragraph.text, | |
| 'align': paragraph.alignment, | |
| 'font': { | |
| 'name': original_run.font.name, | |
| 'bold': original_run.font.bold, | |
| 'italic': original_run.font.italic, | |
| 'underline': original_run.font.underline, | |
| 'color': str(original_run.font.color.rgb), | |
| 'language_id': original_run.font.language_id, | |
| } | |
| } | |
| shape_item[f'paragraph_{r}'] = paragraph_item | |
| group_shape_item[f"shape_{l}"] = shape_item | |
| return group_shape_item | |
| def transfer_to_structure(pptx_file, images_dir_path): | |
| item = {} | |
| prs = Presentation(pptx_file) | |
| image_path_list = [] | |
| for i, slide in enumerate(prs.slides): | |
| slide_item = {} | |
| for j, shape in enumerate(slide.shapes): | |
| shape_item = {} | |
| # Case 1: Normal text box | |
| if shape.has_text_frame: | |
| shape_item['type'] = "text" | |
| text_frame = shape.text_frame | |
| for r, paragraph in enumerate(text_frame.paragraphs): | |
| if paragraph.runs: | |
| original_run = paragraph.runs[0] | |
| paragraph_item = { | |
| 'text': paragraph.text, | |
| 'align': paragraph.alignment, | |
| 'font': { | |
| 'name': original_run.font.name, | |
| 'bold': original_run.font.bold, | |
| 'italic': original_run.font.italic, | |
| 'underline': original_run.font.underline, | |
| 'color': str(original_run.font.color.rgb), | |
| 'language_id': original_run.font.language_id, | |
| } | |
| } | |
| shape_item[f'paragraph_{r}'] = paragraph_item | |
| # Case 2: Grouped shapes | |
| elif isinstance(shape, GroupShape): | |
| shape_item['type'] = "group" | |
| shape_item['group_content'] = transfer_textbox_content_in_group(shape) | |
| # Case 3: Picture | |
| elif isinstance(shape, Picture): | |
| shape_item['type'] = "picture" | |
| image_path = os.path.join(images_dir_path, f"picture_{j}.png") | |
| image_path_list.append(image_path) | |
| shape_item['image_path'] = image_path | |
| shape_item['size'] = shape.image.size | |
| shape_item['dpi'] = shape.image.dpi | |
| shape_item['location'] = (shape.left, shape.top) | |
| shape_item['location_inches'] = (Inches(shape.left).inches, Inches(shape.top).inches) | |
| image_stream = io.BytesIO(shape.image.blob) | |
| shape_image = Image.open(image_stream) | |
| shape_image.save(image_path) | |
| slide_item[f"shape_{j}"] = shape_item | |
| item[f"slide_{i}"] = slide_item | |
| return print_json(item), image_path_list |