Update app.py

app.py

@@ -4,7 +4,7 @@ import shutil
 import tempfile
 import base64
 from pptx import Presentation
-from pptx.util import Inches, Pt 
+from pptx.util import Inches, Pt
 from pptx.enum.text import PP_ALIGN
 import streamlit as st
 import fitz
@@ -24,28 +24,28 @@ def query(payload):
 
 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:
@@ -53,17 +53,17 @@ def extract_paragraphs_by_vertical_spacing(pdf_data, spacing_threshold=10):
                     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")
@@ -74,11 +74,11 @@ if uploaded_file is not None:
     pdf_data = uploaded_file.read()
 
     paragraphs = extract_paragraphs_by_vertical_spacing(pdf_data)
-    i = 1   
+    i = 1
     prs = Presentation()
-    
+
     for paragraph in paragraphs:
-        summary = summarizer(paragraph, max_length=(len(paragraph)//4), min_length=10, do_sample=False)
+        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)
@@ -94,7 +94,7 @@ if uploaded_file is not None:
         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
-        
+
         image_bytes = query({
             "inputs": 'A picture about  :' + summary[0]['summary_text']  # Utilisez le texte du résumé
         })
@@ -114,14 +114,18 @@ if uploaded_file is not None:
         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']
 
-        p.space_after = Pt(12)  # Initialize space_after to 12 points
-        while p.space_after > Pt(0):
-          p.font.size -= Pt(1)  # Reduce the font size by 1 point
-          p.space_after = Pt(12)  # Set space_after to 12 points again
+        p.space_after = Pt(0)  # Initialize space_after to 0 points
+
+        while p.space_after != Pt(12):
+            p.space_after = Pt(12)  # Set space_after to 12 points again
+            p.space_after = Pt(0)  # Initialize space_after to 0 points
+            p.space_after = Pt(12)  # Set space_after to 12 points again
+            p.space_after = Pt(0)  # Initialize space_after to 0 points
+            p.font.size -= Pt(1)  # Reduce the font size by 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)
@@ -129,7 +133,7 @@ if uploaded_file is not None:
 
         text_frame_width = Pt(text_width - 2)  # Réduisez la largeur de 0.1 pouce de chaque côté
         txBox.text_frame.width = text_frame_width
-        
+
         i = i + 1
 
     pptx_stream = io.BytesIO()