Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from pptx import Presentation
 from pptx.util import Inches, Pt
+from pptx.enum.shapes import MSO_SHAPE
+from pptx.dml.color import RGBColor
 import io
 from huggingface_hub import InferenceClient
 import json
@@ -11,21 +13,24 @@ def get_inference_client():
     return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def extract_and_clean_json(text):
-    # Intenta encontrar un JSON válido en el texto
     json_match = re.search(r'\{.*\}', text, re.DOTALL)
     if json_match:
         json_str = json_match.group()
-        # Limpia el JSON de caracteres no válidos
         json_str = re.sub(r'[\n\t\r]', '', json_str)
-        # Intenta corregir problemas comunes de formato
-        json_str = re.sub(r',\s*}', '}', json_str)  # Elimina comas finales extras
-        json_str = re.sub(r',\s*]', ']', json_str)  # Elimina comas finales extras en arrays
+        json_str = re.sub(r',\s*}', '}', json_str)
+        json_str = re.sub(r',\s*]', ']', json_str)
         return json_str
     return None
 
 def generate_presentation_content(topic, client):
     prompt = f"""Crea una presentación de PowerPoint sobre el tema: {topic}.
-    Genera exactamente 5 diapositivas. Cada diapositiva debe tener un título y contenido.
+    Genera exactamente 8 diapositivas con la siguiente estructura:
+    1. Título de la presentación
+    2. Introducción
+    3. Índice
+    4-7. Contenido principal (4 diapositivas)
+    8. Conclusiones
+    
     Formatea la salida como un JSON con la siguiente estructura:
     {{
         "slides": [
@@ -38,16 +43,16 @@ def generate_presentation_content(topic, client):
     No incluyas saltos de línea o caracteres especiales dentro del JSON.
     """
     
-    response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
+    response = client.text_generation(prompt, max_new_tokens=2000, temperature=0.7)
     
     try:
         json_str = extract_and_clean_json(response)
         if json_str:
             slides_data = json.loads(json_str)
-            if 'slides' in slides_data and len(slides_data['slides']) == 5:
+            if 'slides' in slides_data and len(slides_data['slides']) == 8:
                 return slides_data['slides']
             else:
-                raise ValueError("El JSON no contiene exactamente 5 diapositivas")
+                raise ValueError("El JSON no contiene exactamente 8 diapositivas")
         else:
             raise ValueError("No se encontró un JSON válido en la respuesta")
     except json.JSONDecodeError as e:
@@ -63,17 +68,74 @@ def generate_presentation_content(topic, client):
         st.code(response)
         return None
 
-def create_powerpoint(slides):
+def apply_design(prs, design):
+    if design == "Moderno":
+        background = prs.slides[0].background
+        fill = background.fill
+        fill.solid()
+        fill.fore_color.rgb = RGBColor(240, 240, 240)
+        
+        for slide in prs.slides:
+            for shape in slide.shapes:
+                if shape.has_text_frame:
+                    tf = shape.text_frame
+                    tf.text = tf.text
+                    for paragraph in tf.paragraphs:
+                        paragraph.font.color.rgb = RGBColor(0, 0, 0)
+                        if shape.name == 'Title':
+                            paragraph.font.size = Pt(44)
+                        else:
+                            paragraph.font.size = Pt(24)
+    elif design == "Corporativo":
+        background = prs.slides[0].background
+        fill = background.fill
+        fill.solid()
+        fill.fore_color.rgb = RGBColor(255, 255, 255)
+        
+        for slide in prs.slides:
+            left = top = Inches(0.5)
+            width = Inches(1)
+            height = Inches(0.5)
+            shape = slide.shapes.add_shape(MSO_SHAPE.RECTANGLE, left, top, width, height)
+            shape.fill.solid()
+            shape.fill.fore_color.rgb = RGBColor(0, 112, 192)
+            
+            for shape in slide.shapes:
+                if shape.has_text_frame:
+                    tf = shape.text_frame
+                    tf.text = tf.text
+                    for paragraph in tf.paragraphs:
+                        paragraph.font.color.rgb = RGBColor(0, 0, 0)
+                        if shape.name == 'Title':
+                            paragraph.font.size = Pt(40)
+                            paragraph.font.color.rgb = RGBColor(0, 112, 192)
+                        else:
+                            paragraph.font.size = Pt(20)
+
+def create_powerpoint(slides, design):
     prs = Presentation()
     
-    for slide_data in slides:
+    # Diapositiva de título
+    slide = prs.slides.add_slide(prs.slide_layouts[0])
+    title = slide.shapes.title
+    subtitle = slide.placeholders[1]
+    title.text = slides[0]['title']
+    subtitle.text = slides[0]['content']
+
+    # Otras diapositivas
+    for slide_data in slides[1:]:
         slide = prs.slides.add_slide(prs.slide_layouts[1])
-        
-        title_shape = slide.shapes.title
-        content_shape = slide.placeholders[1]
-        
-        title_shape.text = slide_data['title']
-        content_shape.text = slide_data['content']
+        title = slide.shapes.title
+        content = slide.placeholders[1]
+        title.text = slide_data['title']
+        content.text = slide_data['content']
+    
+    # Diapositiva "Gracias"
+    slide = prs.slides.add_slide(prs.slide_layouts[0])
+    title = slide.shapes.title
+    title.text = "¡Gracias!"
+    
+    apply_design(prs, design)
     
     pptx_buffer = io.BytesIO()
     prs.save(pptx_buffer)
@@ -88,6 +150,11 @@ def main():
     
     topic = st.text_input("Por favor, ingrese el tema de la presentación:")
     
+    design = st.selectbox(
+        "Seleccione un diseño para la presentación:",
+        ("Simple", "Moderno", "Corporativo")
+    )
+    
     if st.button("Generar Presentación"):
         if topic:
             try:
@@ -96,7 +163,7 @@ def main():
                 
                 if slides:
                     with st.spinner("Creando archivo PowerPoint..."):
-                        pptx_buffer = create_powerpoint(slides)
+                        pptx_buffer = create_powerpoint(slides, design)
                     
                     st.success("Presentación generada con éxito!")