Update app.py

app.py

@@ -8,6 +8,9 @@ import os
 import requests
 import io
 from PIL import Image
+from pptx import Presentation
+from pptx.util import Inches
+import tempfile
 
 API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
 headers = {"Authorization": "Bearer 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"}
@@ -69,17 +72,42 @@ if uploaded_file is not None:
     pdf_data = uploaded_file.read()
 
     paragraphs = extract_paragraphs_by_vertical_spacing(pdf_data)
-    i = 1   
+    i = 1
+
+    # Create a PowerPoint presentation
+    prs = Presentation()
+
     for paragraph in paragraphs:
-    
-        summary = summarizer(paragraph, max_length=(len(paragraph)/2), min_length=10, do_sample=False)
-        st.text(f"Paragraphe {i}: {summary[0]['summary_text']}")  # Affiche le résumé du paragraphe
-        
+        summary = summarizer(paragraph, max_length=(len(paragraph) / 2), min_length=10, do_sample=False)
+
+        # Create a slide
+        slide = prs.slides.add_slide(prs.slide_layouts[5])
+
+        # Add the paragraph to the slide
+        left = top = Inches(1)
+        width = height = Inches(5)
+        txBox = slide.shapes.add_textbox(left, top, width, height)
+        tf = txBox.text_frame
+        p = tf.add_paragraph()
+        p.text = f"Paragraph {i}:"
+        p.space_after = Inches(0.1)
+        p = tf.add_paragraph()
+        p.text = summary[0]['summary_text']
+
+        # Generate and add the image to the slide
         image_bytes = query({
-            "inputs": 'A picture about  :' + summary[0]['summary_text']  # Utilisez le texte du résumé
+            "inputs": 'A picture without text about: ' + summary[0]['summary_text']
         })
         image = Image.open(io.BytesIO(image_bytes))
-        st.image(image)
+        temp_img_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
+        image.save(temp_img_path)
+        slide.shapes.add_picture(temp_img_path, Inches(1), Inches(1.5), Inches(5), Inches(3))
+
+        i += 1
 
-        i = i + 1
+    # Save the PowerPoint presentation
+    presentation_path = tempfile.NamedTemporaryFile(delete=False, suffix=".pptx").name
+    prs.save(presentation_path)
 
+    # Display the link to download the PowerPoint file
+    st.markdown(f"Download the PowerPoint presentation: [PowerPoint File]({presentation_path})")