Update app.py

app.py

@@ -1,8 +1,6 @@
 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
@@ -10,47 +8,24 @@ import re
 
 @st.cache_resource
 def get_inference_client():
-    return InferenceClient(model="mistralai/Mixtral-8x7B-Instruct-v0.1")
+    return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def extract_and_clean_json(text):
-    try:
-        # Extracción del JSON de la respuesta
-        json_match = re.search(r'\{.*?\}', text, re.DOTALL)
-        if json_match:
-            json_str = json_match.group()
-            # Limpieza básica
-            json_str = re.sub(r'[\n\t\r]', '', json_str)
-            json_str = re.sub(r',\s*}', '}', json_str)
-            json_str = re.sub(r',\s*]', ']', json_str)
-            
-            # Verificación y análisis del JSON
-            try:
-                data = json.loads(json_str)
-                return data
-            except json.JSONDecodeError as e:
-                st.error(f"Error al decodificar JSON limpio: {str(e)}")
-                st.text("JSON procesado (después de limpieza):")
-                st.code(json_str)
-                return None
-        else:
-            st.error("No se encontró un JSON válido en la respuesta")
-            st.text("Respuesta original del modelo:")
-            st.code(text)
-            return None
-    except Exception as e:
-        st.error(f"Error inesperado al procesar el JSON: {str(e)}")
-        st.text("Respuesta original del modelo:")
-        st.code(text)
-        return None
+    # 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
+        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 9 diapositivas con la siguiente estructura:
-    1. Título de la presentación
-    2. Introducción
-    3. Índice
-    4-9. Contenido principal (6 diapositivas)
-    
+    Genera exactamente 5 diapositivas. Cada diapositiva debe tener un título y contenido.
     Formatea la salida como un JSON con la siguiente estructura:
     {{
         "slides": [
@@ -65,87 +40,40 @@ def generate_presentation_content(topic, client):
     
     response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
     
-    # Limpieza y procesamiento del JSON
-    slides_data = extract_and_clean_json(response)
-    
-    if slides_data and 'slides' in slides_data:
-        if len(slides_data['slides']) == 9:
-            return slides_data['slides']
+    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:
+                return slides_data['slides']
+            else:
+                raise ValueError("El JSON no contiene exactamente 5 diapositivas")
         else:
-            st.error(f"El JSON no contiene exactamente 9 diapositivas. Contiene {len(slides_data['slides'])} diapositivas.")
-            st.text("JSON procesado:")
-            st.code(json.dumps(slides_data, indent=4))
-            return None
-    return None
-
-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)
+            raise ValueError("No se encontró un JSON válido en la respuesta")
+    except json.JSONDecodeError as e:
+        st.error(f"Error al decodificar JSON: {str(e)}")
+        st.text("JSON procesado (después de limpieza):")
+        st.code(json_str)
+        st.text("Respuesta original del modelo:")
+        st.code(response)
+        return None
+    except ValueError as e:
+        st.error(str(e))
+        st.text("Respuesta del modelo:")
+        st.code(response)
+        return None
 
-def create_powerpoint(slides, design):
+def create_powerpoint(slides):
     prs = Presentation()
     
-    # 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:]:
+    for slide_data in slides:
         slide = prs.slides.add_slide(prs.slide_layouts[1])
-        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)
+        
+        title_shape = slide.shapes.title
+        content_shape = slide.placeholders[1]
+        
+        title_shape.text = slide_data['title']
+        content_shape.text = slide_data['content']
     
     pptx_buffer = io.BytesIO()
     prs.save(pptx_buffer)
@@ -160,11 +88,6 @@ 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:
@@ -173,7 +96,7 @@ def main():
                 
                 if slides:
                     with st.spinner("Creando archivo PowerPoint..."):
-                        pptx_buffer = create_powerpoint(slides, design)
+                        pptx_buffer = create_powerpoint(slides)
                     
                     st.success("Presentación generada con éxito!")
                     
@@ -183,12 +106,10 @@ def main():
                         file_name=f"{topic.replace(' ', '_')}_presentacion.pptx",
                         mime="application/vnd.openxmlformats-officedocument.presentationml.presentation"
                     )
-                else:
-                    st.error("No se generó contenido de presentación válido.")
             except Exception as e:
                 st.error(f"Ocurrió un error al generar la presentación: {str(e)}")
         else:
             st.warning("Por favor, ingrese un tema para la presentación.")
 
 if __name__ == "__main__":
-    main()
+    main()