Create app2.py

app2.py

@@ -0,0 +1,168 @@
+from PyPDF2 import PdfReader
+import re
+import shutil
+import tempfile
+import base64
+from pptx import Presentation
+from pptx.util import Inches, Pt 
+from pptx.enum.text import PP_ALIGN
+import streamlit as st
+import fitz
+from transformers import pipeline
+import os
+import requests
+import io
+from PIL import Image
+
+API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+headers = {"Authorization": "Bearer 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"}
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+def extract_paragraphs_by_vertical_spacing(pdf_data, spacing_threshold=10):
+    paragraphs = []
+    
+    try:
+        pdf_stream = io.BytesIO(pdf_data)
+        pdf_document = fitz.open(stream=pdf_stream, filetype="pdf")
+        
+        for page_number in range(pdf_document.page_count):
+            page = pdf_document.load_page(page_number)
+            blocks = page.get_text("blocks")
+            
+            current_paragraph = ""
+            previous_bottom = None
+            
+            for block in blocks:
+                x0, y0, x1, y1 = block[:4]  # Coordonnées du bloc de texte
+                text = block[4]  # Texte du bloc
+                
+                # Mesurez l'espacement vertical entre les blocs de texte
+                if previous_bottom is not None:
+                    vertical_spacing = y0 - previous_bottom
+                else:
+                    vertical_spacing = 0
+                
+                # Si l'espacement vertical dépasse le seuil, considérez-le comme un nouveau paragraphe
+                if vertical_spacing > spacing_threshold:
+                    if current_paragraph:
+                        paragraphs.append(current_paragraph.strip())
+                    current_paragraph = text
+                else:
+                    current_paragraph += " " + text  # Ajoutez le texte au paragraphe actuel
+                
+                previous_bottom = y1
+            
+            # Ajoutez le dernier paragraphe de la page
+            if current_paragraph:
+                paragraphs.append(current_paragraph.strip())
+        
+        pdf_document.close()
+    except Exception as e:
+        print(f"Erreur lors de l'extraction du PDF : {str(e)}")
+    
+    return paragraphs
+
+st.title("PDF2SLIDE")
+
+uploaded_file = st.file_uploader("Selectionnez un PDF", type=["pdf"])
+
+if uploaded_file is not None:
+    pdf_data = uploaded_file.read()
+
+    paragraphs = extract_paragraphs_by_vertical_spacing(pdf_data)
+    i = 1   
+    prs = Presentation()
+    
+    for paragraph in paragraphs:
+
+        summary = summarizer(paragraph, max_length=(len(paragraph)/4), min_length=10, do_sample=False)
+
+        slide_layout = prs.slide_layouts[5]  # Utilisez le modèle de diapositive approprié (index 5 pour une diapositive de titre et de contenu)
+        slide = prs.slides.add_slide(slide_layout)
+
+        title = slide.shapes.title
+        title.text = f"Paragraphe {i}"
+        title.alignment = PP_ALIGN.CENTER
+
+        slide_width = prs.slide_width
+        slide_height = prs.slide_height
+
+        image_width = slide_width * 0.7  # L'image occupe 70% de la largeur de la slide
+        image_height = slide_height * 0.7  # L'image occupe 70% de la hauteur de la slide
+        left_img = (slide_width - image_width) / 2  # Centrez horizontalement
+        top_img = (slide_height - image_height) * 0.6  # Occupe 15% de la hauteur en haut de la slide
+
+        #left = prs.slide_width * 0.1
+        #top = prs.slide_height * 0.6
+        #width = prs.slide_width * 0.8
+        #height = prs.slide_height * 0.3
+        #txBox = slide.shapes.add_textbox(left, top, width, height)
+        #tf = txBox.text_frame
+        #p = tf.add_paragraph()
+        #p.text = summary[0]['summary_text']
+    
+        #st.text(f"Paragraphe {i}: {summary[0]['summary_text']}")  # Affiche le résumé du paragraphe
+        
+        image_bytes = query({
+            "inputs": 'A picture about  :' + summary[0]['summary_text']  # Utilisez le texte du résumé
+        })
+        image = Image.open(io.BytesIO(image_bytes))
+        #left = top = prs.slide_width * 0.1
+        pic = slide.shapes.add_picture(io.BytesIO(image_bytes), left_img, top_img, image_width, image_height)
+
+        title_width = slide_width * 0.7  # Le titre occupe 70% de la largeur de la slide
+        title_height = slide_height * 0.15  # Le titre occupe 15% de la hauteur de la slide
+        left_title = (slide_width - title_width) / 2  # Centrez horizontalement
+        top_title = (slide_height - title_height) * 0.05  # Occupe 5% de la hauteur en haut de la slide
+        title = slide.shapes.add_textbox(left_title, top_title, title_width, title_height)
+        title_frame = title.text_frame
+
+        #title_p = title_frame.add_paragraph()
+        #title_p.text = "Paragraphe {i}: "
+        #title_p.alignment = PP_ALIGN.CENTER  # Centrez le texte horizontalement
+
+        text_width = slide_width * 0.7  # Le texte occupe 70% de la largeur de la slide
+        text_height = slide_height * 0.15  # Le texte occupe 15% de la hauteur de la slide
+        left_text = (slide_width - text_width) / 2  # Centrez horizontalement
+        top_text = slide_height * 0.85  # Occupe 85% de la hauteur en bas de la slide
+        txBox = slide.shapes.add_textbox(left_text, top_text, text_width, text_height)
+        tf = txBox.text_frame
+        
+        p = tf.add_paragraph()
+        p.text = summary[0]['summary_text']
+
+        # Ajustez la taille de police pour le texte afin qu'il rentre dans le cadre de texte
+        while p.space_after > Pt(0):
+            p.font.size -= Pt(1)  # Réduisez la taille de police de 1 point
+
+        p.space_before = Pt(12)  # Espace avant le paragraphe (12 points)
+        p.space_after = Pt(12)  # Espace après le paragraphe (12 points)
+        p.alignment = PP_ALIGN.CENTER  # Centrez le texte horizontalement
+
+        # Ajustez la largeur du cadre de texte du texte pour éviter le dépassement
+        text_frame_width = Pt(text_width - 2)  # Réduisez la largeur de 0.1 pouce de chaque côté
+        #txBox.width = text_frame_width
+        txBox.text_frame.width = text_frame_width
+        
+       # st.image(image)
+
+        i = i + 1
+
+    #tempfile_name = tempfile.NamedTemporaryFile(delete=False, suffix=".pptx").name
+    pptx_stream = io.BytesIO()
+    prs.save(pptx_stream)
+    pptx_stream.seek(0)
+
+    st.markdown(
+    f'<a href="data:application/vnd.openxmlformats-officedocument.presentationml.presentation;base64,{base64.b64encode(pptx_stream.read()).decode()}" download="presentation.pptx">Télécharger la présentation</a>',
+    unsafe_allow_html=True,
+)
+
+    
+    
+    #st.download_button("Télécharger la présentation", "output_path", key="download_pptx", mime="application/vnd.openxmlformats-officedocument.presentationml.presentation")
+