Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from pptx import Presentation
 from pptx.util import Inches, Pt
-from pptx.enum.text import MSO_ANCHOR, MSO_AUTO_SIZE
+from pptx.enum.text import MSO_ANCHOR, MSO_AUTO_SIZE, PP_ALIGN # IMPORTACIÓN CORREGIDA AQUÍ
 import requests
 from PIL import Image
 from io import BytesIO
@@ -65,12 +65,12 @@ def parse_gemini_response_for_slides(response_text, expected_slides_key="slides"
     except json.JSONDecodeError as e:
         raise ValueError(f"Error al decodificar JSON: {e}. JSON intentado: {json_str[:500]}...")
 
-    if is_single_slide: # Para el título creativo
+    if is_single_slide: 
         if not (isinstance(data, dict) and 'title' in data and 'content' in data):
              raise ValueError("La estructura del JSON para la diapositiva de título es incorrecta. Se esperaba 'title' y 'content'.")
         if not (isinstance(data['title'], str) and isinstance(data['content'], str)):
             raise ValueError("El título y el contenido de la diapositiva de título deben ser cadenas.")
-        return [data] # Devolver como lista de una diapositiva
+        return [data] 
 
     if not isinstance(data, dict) or expected_slides_key not in data:
         raise ValueError(f"La estructura del JSON es incorrecta. Falta la clave '{expected_slides_key}'.")
@@ -90,7 +90,6 @@ def parse_gemini_response_for_slides(response_text, expected_slides_key="slides"
     return slides
 
 def generate_creative_title_slide(topic, client, max_retries=3):
-    """Genera una única diapositiva de título creativa."""
     prompt = f"""Genera un título MUY llamativo y creativo para una presentación sobre "{topic}".
     Además, proporciona una breve frase o subtítulo de impacto para acompañar el título.
     Es CRUCIAL que tu respuesta sea ÚNICAMENTE un JSON válido con la siguiente estructura exacta, sin texto adicional antes o después:
@@ -109,7 +108,7 @@ def generate_creative_title_slide(topic, client, max_retries=3):
             if not response_obj.parts: raise ValueError("Respuesta del modelo vacía (título creativo).")
             last_response_text = response_obj.text
             title_slide_data = parse_gemini_response_for_slides(last_response_text, is_single_slide=True)
-            return title_slide_data # Es una lista con un solo diccionario de slide
+            return title_slide_data 
         except (ValueError, json.JSONDecodeError) as e:
             last_error = e
             st.warning(f"Intento {attempt + 1}/{max_retries} (título creativo) fallido: {e}")
@@ -155,7 +154,6 @@ def generate_presentation_content(topic, client, max_retries=3):
     return None
 
 def generate_conclusion_slides_from_text(text_to_summarize, num_conclusion_slides, topic, client, max_retries=3):
-    """Genera diapositivas de conclusión basadas en el texto y el tema general."""
     if not text_to_summarize: return []
     max_len = 30000 
     if len(text_to_summarize) > max_len:
@@ -225,16 +223,14 @@ def add_slide_content_and_image(slide, slide_data, prs_slide_width, prs_slide_he
         slide.shapes.title.text = slide_data['title']
     
     content_placeholder = None
-    # Para la diapositiva de título, el placeholder de contenido puede ser el primero si el layout es de título solo
     if is_title_slide and slide.placeholders and slide.placeholders[0] == slide.shapes.title and len(slide.placeholders) > 1:
-        content_placeholder = slide.placeholders[1] # Subtitle placeholder
+        content_placeholder = slide.placeholders[1] 
     elif not is_title_slide and slide.placeholders and len(slide.placeholders) > 1 and slide.placeholders[1]:
-         content_placeholder = slide.placeholders[1] # Body placeholder
+         content_placeholder = slide.placeholders[1]
 
     margin_emu = Inches(0.3)
-    # Si es diapositiva de título, el contenido (subtítulo) puede estar más arriba
     common_top_emu = Inches(0.5) if is_title_slide and slide.shapes.title else Inches(1.6)
-    if is_title_slide and content_placeholder: # Ajustar top del subtitulo
+    if is_title_slide and content_placeholder: 
          common_top_emu = content_placeholder.top 
     
     content_bottom_emu = prs_slide_height - margin_emu
@@ -243,18 +239,17 @@ def add_slide_content_and_image(slide, slide_data, prs_slide_width, prs_slide_he
     text_area_left_emu = margin_emu
     
     img_area_width_float = 0
-    if add_image and not is_title_slide: # No añadir imagen a la diapositiva de título principal
+    if add_image and not is_title_slide: 
         img_area_width_float = prs_slide_width * 0.40 
-        text_area_width_float = prs_slide_width - img_area_width_float - (3 * margin_emu) # +margen entre texto e img
-
+        text_area_width_float = prs_slide_width - img_area_width_float - (3 * margin_emu)
 
     common_height_emu = content_bottom_emu - common_top_emu
-    if is_title_slide and content_placeholder: # Ajustar altura del subtitulo
+    if is_title_slide and content_placeholder: 
         common_height_emu = content_placeholder.height
     
     if content_placeholder:
         content_placeholder.left = int(text_area_left_emu)
-        if not is_title_slide: # No sobreescribir el top del subtítulo si ya está bien
+        if not is_title_slide: 
             content_placeholder.top = int(common_top_emu)
         content_placeholder.width = int(text_area_width_float)
         content_placeholder.height = int(common_height_emu)
@@ -263,23 +258,23 @@ def add_slide_content_and_image(slide, slide_data, prs_slide_width, prs_slide_he
         p = tf.add_paragraph(); p.text = slide_data['content']
         tf.word_wrap = True
         tf.auto_size = MSO_AUTO_SIZE.TEXT_TO_FIT_SHAPE if not is_title_slide else MSO_AUTO_SIZE.NONE
-        if is_title_slide: # Centrar subtítulo
-            p.alignment = PP_ALIGN.CENTER # Necesitarías: from pptx.enum.text import PP_ALIGN
+        if is_title_slide: 
+            p.alignment = PP_ALIGN.CENTER 
             tf.vertical_anchor = MSO_ANCHOR.MIDDLE
     else: 
         txBox_top = common_top_emu
-        if is_title_slide and slide.shapes.title: # Colocar subtítulo debajo del título principal
+        if is_title_slide and slide.shapes.title: 
             txBox_top = slide.shapes.title.top + slide.shapes.title.height + Inches(0.2)
 
         txBox = slide.shapes.add_textbox(
             int(text_area_left_emu), int(txBox_top), 
-            int(text_area_width_float), int(common_height_emu / (1.5 if is_title_slide else 1) ) # Menos altura para subtitulo
+            int(text_area_width_float), int(common_height_emu / (1.5 if is_title_slide else 1) ) 
         )
         tf = txBox.text_frame; tf.text = slide_data['content']
         tf.word_wrap = True
         tf.auto_size = MSO_AUTO_SIZE.SHAPE_TO_FIT_TEXT if is_title_slide else MSO_AUTO_SIZE.TEXT_TO_FIT_SHAPE
-        if is_title_slide: # Centrar subtítulo
-            p = tf.paragraphs[0]; p.alignment = PP_ALIGN.CENTER # Necesitarías: from pptx.enum.text import PP_ALIGN
+        if is_title_slide: 
+            p = tf.paragraphs[0]; p.alignment = PP_ALIGN.CENTER 
             tf.vertical_anchor = MSO_ANCHOR.TOP
 
 
@@ -327,29 +322,34 @@ def create_powerpoint(title_slide_content, main_slides_content, template_path, c
     prs = Presentation(template_path)
     slide_width_emu = prs.slide_width
     slide_height_emu = prs.slide_height
-    from pptx.enum.text import PP_ALIGN # Importar aquí para evitar error si no se usa add_slide_content_and_image
 
-    # 0. Diapositiva de Título Creativa (usar layout de título, ej. 0 o 5)
-    if title_slide_content: # es una lista con 1 slide_data
-        # Intentar usar un layout de solo título si está disponible, sino el de título y contenido.
-        title_layout = prs.slide_layouts[0] if prs.slide_layouts else prs.slide_layouts[5] if len(prs.slide_layouts) > 5 else prs.slide_layouts[1]
+    if title_slide_content: 
+        title_layout_index = 0 
+        if len(prs.slide_layouts) > 5: 
+            title_layout_index = 5
+        elif not prs.slide_layouts: 
+             prs.slide_master.slide_layouts.add_slide_layout()
+             title_layout_index = 0
+
+        if title_layout_index >= len(prs.slide_layouts) :
+            title_layout_index = 0 
+        
+        title_layout = prs.slide_layouts[title_layout_index] if prs.slide_layouts else prs.slide_layouts[0] # Fallback final
+            
         slide = prs.slides.add_slide(title_layout)
         add_slide_content_and_image(slide, title_slide_content[0], slide_width_emu, slide_height_emu, add_image=False, is_title_slide=True)
 
-    # 1. Diapositivas principales (con imágenes)
     for slide_data in main_slides_content:
-        slide_layout = prs.slide_layouts[1] # Título y Contenido
+        slide_layout = prs.slide_layouts[1] 
         slide = prs.slides.add_slide(slide_layout)
         add_slide_content_and_image(slide, slide_data, slide_width_emu, slide_height_emu, add_image=True)
 
-    # 2. Diapositivas de Conclusión (sin imágenes)
     if conclusion_slides_content:
         for slide_data in conclusion_slides_content:
             slide_layout = prs.slide_layouts[1] 
             slide = prs.slides.add_slide(slide_layout)
             add_slide_content_and_image(slide, slide_data, slide_width_emu, slide_height_emu, add_image=False)
 
-    # 3. Diapositiva final de agradecimiento
     final_slide_layout = prs.slide_layouts[1] 
     final_slide = prs.slides.add_slide(final_slide_layout)
     if final_slide.shapes.title: final_slide.shapes.title.text = "¡Gracias por su atención!"
@@ -374,7 +374,6 @@ def main():
     topic = st.sidebar.text_input("📝 Tema principal de la presentación:", placeholder="Ej: El futuro de la IA")
 
     plantillas_dir = "PLANTILLAS"
-    # ... (resto de la lógica de plantillas sin cambios, solo movida a sidebar si no estaba ya) ...
     if not os.path.exists(plantillas_dir):
         try: os.makedirs(plantillas_dir); st.sidebar.info(f"Directorio '{plantillas_dir}' creado.")
         except OSError as e: st.sidebar.error(f"No se pudo crear dir '{plantillas_dir}': {e}"); st.stop()
@@ -386,13 +385,26 @@ def main():
     if not available_templates:
         st.sidebar.warning(f"No hay plantillas .pptx en '{plantillas_dir}'.")
         default_prs = Presentation()
-        if not default_prs.slide_layouts: default_prs.slide_master.slide_layouts.add_slide_layout()
-        if default_prs.slide_layouts: default_prs.slides.add_slide(default_prs.slide_layouts[0])
+        # Asegurar que hay al menos un layout usable
+        if not default_prs.slide_layouts: 
+            try:
+                default_prs.slide_master.slide_layouts.add_slide_layout() # Intenta añadir un layout maestro
+            except Exception: # Si falla, al menos intenta con los layouts por defecto
+                pass
+        if default_prs.slide_layouts: 
+            try:
+                default_prs.slides.add_slide(default_prs.slide_layouts[0])
+            except IndexError: # Si el layout 0 no existe tras añadir
+                 if default_prs.slide_layouts: # Intenta con el primero que encuentre
+                     default_prs.slides.add_slide(default_prs.slide_layouts[0])
+
+
         default_template_path = os.path.join(plantillas_dir, "default.pptx")
         try: default_prs.save(default_template_path); st.sidebar.info(f"Plantilla 'default.pptx' creada."); available_templates.append("default.pptx")
         except Exception as e: st.sidebar.error(f"No se pudo crear plantilla por defecto: {e}")
+            
     template_options = {os.path.splitext(t)[0]: os.path.join(plantillas_dir, t) for t in available_templates}
-    if not template_options: st.sidebar.error("CRÍTICO: No hay plantillas."); st.stop()
+    if not template_options: st.sidebar.error("CRÍTICO: No hay plantillas. Cree 'PLANTILLAS' y añada un .pptx."); st.stop()
     selected_template_name = st.sidebar.selectbox("🎨 Seleccione una plantilla:", list(template_options.keys()))
 
 
@@ -416,25 +428,20 @@ def main():
         overall_progress = st.progress(0)
         status_text = st.empty()
 
-        # 0. Generar título creativo
         status_text.info("🎨 Creando un título impactante...")
         title_slide_content = generate_creative_title_slide(topic, client)
         if not title_slide_content:
             status_text.warning("⚠️ No se pudo generar el título creativo, continuando sin él.")
-            # No detenemos la ejecución, solo no habrá diapositiva de título.
         else:
             status_text.success("👍 Título creativo generado.")
         overall_progress.progress(10)
 
-
-        # 1. Generar contenido principal
         status_text.info("🧠 Generando contenido principal con Gemini...")
         main_slides_content = generate_presentation_content(topic, client)
         if not main_slides_content:
             status_text.error("❌ Falló la generación del contenido principal."); overall_progress.progress(100); return
         status_text.success("👍 Contenido principal generado."); overall_progress.progress(50)
 
-        # 2. Procesar PDF para conclusiones
         if uploaded_pdf and num_conclusion_slides > 0:
             status_text.info(f"📄 Procesando PDF '{uploaded_pdf.name}' para conclusiones...")
             pdf_text = extract_text_from_pdf(uploaded_pdf); overall_progress.progress(60)
@@ -446,11 +453,8 @@ def main():
             else: status_text.warning("⚠️ No se pudo extraer texto del PDF.")
         overall_progress.progress(80)
 
-        # 3. Crear archivo PowerPoint
         status_text.info("🛠️ Creando archivo PowerPoint...")
         try:
-            # Necesitamos PP_ALIGN para centrar el subtítulo de la diapositiva de título
-            from pptx.enum.text import PP_ALIGN 
             pptx_buffer = create_powerpoint(title_slide_content, main_slides_content, template_path, conclusion_slides_content)
             overall_progress.progress(100)
             status_text.success("🎉 ¡Presentación generada con éxito!")