Update app.py

app.py

@@ -96,16 +96,41 @@ def process_pptx(pptx_file):
     images_dir_path = "images"
     if not os.path.exists(images_dir_path):
         os.makedirs(images_dir_path)
+    
+    # Convert the first slide to an image for preview
+    prs = Presentation(pptx_file.name)
+    slide = prs.slides[0]
+    slide_image_path = os.path.join(images_dir_path, "preview.png")
+    save_slide_as_image(slide, slide_image_path)
+    
     json_output, image_paths = transfer_to_structure(pptx_file.name, images_dir_path)
-    return json_output, image_paths
+    return json_output, image_paths, slide_image_path
+
+def save_slide_as_image(slide, image_path):
+    # Convert slide to image using PIL
+    slide_image = slide.shapes.title.text_frame.paragraphs[0].runs[0].font.color.rgb
+    slide_image.save(image_path)
 
 # Gradio interface
+def upload_pptx(pptx_file):
+    images_dir_path = "images"
+    if not os.path.exists(images_dir_path):
+        os.makedirs(images_dir_path)
+    
+    # Convert the first slide to an image for preview
+    prs = Presentation(pptx_file.name)
+    slide = prs.slides[0]
+    slide_image_path = os.path.join(images_dir_path, "preview.png")
+    save_slide_as_image(slide, slide_image_path)
+    
+    return slide_image_path
+
 iface = gr.Interface(
     fn=process_pptx,
     inputs=gr.File(label="Upload PowerPoint File"),
-    outputs=[gr.Textbox(label="JSON Output"), gr.Gallery(label="Extracted Images")],
+    outputs=[gr.Textbox(label="JSON Output"), gr.Gallery(label="Extracted Images"), gr.Image(label="PPT Preview")],
     title="PowerPoint to JSON Converter",
-    description="Upload a PowerPoint file to convert its structure to JSON and display extracted images."
+    description="Upload a PowerPoint file to convert its structure to JSON and display extracted images and preview."
 )
 
 iface.launch()