import gradio as gr
from pptx import Presentation
from pptx.util import Pt, Inches
from pptx.shapes.group import GroupShape
from pptx.dml.color import RGBColor
from pptx.enum.text import PP_ALIGN
from pptx.shapes.picture import Picture
import json
import os
from PIL import Image
import io
from pptx2png import pptx_to_images, render_images_with_skia

def print_json(item):
    item_json = json.dumps(item, ensure_ascii=False, indent=4)
    return item_json

def transfer_textbox_content_in_group(group_shape):
    """Edit the content of text boxes within a 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):
                original_run = paragraph.runs[0]
                paragraph_item = {}
                paragraph_item['text'] = paragraph.text
                paragraph_item['align'] = paragraph.alignment
                font_item = {}
                font_item['name'] = original_run.font.name
                font_item['bold'] = original_run.font.bold
                font_item['italic'] = original_run.font.italic
                font_item['underline'] = original_run.font.underline
                font_item['color'] = original_run.font.color.rgb
                font_item['language_id'] = original_run.font.language_id
                paragraph_item['font'] = font_item
                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 = []
    
    # Iterate through each slide in the presentation
    for i, slide in enumerate(prs.slides):
        # Iterate through each shape in the slide
        slide_item = {}
        for j, shape in enumerate(slide.shapes):
            # If the shape is a group, process its shapes
            shape_item = {}
            if shape.has_text_frame:
                text_frame = shape.text_frame
                for paragraph in text_frame.paragraphs:
                    # Clear the existing text but keep the paragraph
                    paragraph.clear()
                    # Add a new run with the new content and copy font style
                    run = paragraph.add_run()
                    run.text = new_content
                    if paragraph.runs:
                        original_run = paragraph.runs[0]
                        copy_font_style(original_run, run)
            elif isinstance(shape, GroupShape):
                shape_item['type'] = "group"
                group_shape_item = transfer_textbox_content_in_group(shape)
                shape_item['group_content'] = group_shape_item
                pass
            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 # width, height
                shape_item['dpi'] = shape.image.dpi # (horz_dpi, vert_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)
                pass
            slide_item[f"shape_{j}"] = shape_item
        item[f"slide_{i}"] = slide_item
                
    return print_json(item), image_path_list

def copy_font_style(original_run, new_run):
    new_run.font.name = original_run.font.name
    new_run.font.bold = original_run.font.bold
    new_run.font.italic = original_run.font.italic
    new_run.font.underline = original_run.font.underline
    new_run.font.color.rgb = original_run.font.color.rgb
    new_run.font.language_id = original_run.font.language_id

def process_pptx(pptx_file):
    images_dir_path = "images"
    output_dir = "rendered_png"  # Directory to save the rendered images
    if not os.path.exists(images_dir_path):
        os.makedirs(images_dir_path)
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    json_output, image_paths = transfer_to_structure(pptx_file, images_dir_path)

    images, slide_dimensions = pptx_to_images(pptx_file)
    rendered_image_path = render_images_with_skia(images, slide_dimensions, output_dir)
    print(image_paths, rendered_image_path)
    return json_output, image_paths, rendered_image_path

# Gradio interface
iface = gr.Interface(
    fn=process_pptx,
    inputs=gr.File(label="Upload PowerPoint File"),
    outputs=[gr.Textbox(label="JSON Output"), gr.Gallery(label="Extracted Images"), gr.Image(type="filepath", label="PPT Preview")],
    title="PowerPoint to JSON Converter",
    description="Upload a PowerPoint file to convert its structure to JSON and display extracted images."
)

iface.launch()