Update app.py

app.py

@@ -21,63 +21,49 @@ def extract_and_clean_json(text):
         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.
-    Formatea la salida como un JSON con la siguiente estructura:
+def generate_presentation_content(topic, client, max_retries=3):
+    prompt = f"""Genera una presentación de PowerPoint sobre el tema: "{topic}".
+    Debes crear exactamente 5 diapositivas. Cada diapositiva debe tener un título y contenido.
+    Es CRUCIAL que tu respuesta sea ÚNICAMENTE un JSON válido con la siguiente estructura exacta, sin texto adicional antes o después:
+
     {{
         "slides": [
             {{"title": "Título de la diapositiva 1", "content": "Contenido de la diapositiva 1"}},
             {{"title": "Título de la diapositiva 2", "content": "Contenido de la diapositiva 2"}},
-            ...
+            {{"title": "Título de la diapositiva 3", "content": "Contenido de la diapositiva 3"}},
+            {{"title": "Título de la diapositiva 4", "content": "Contenido de la diapositiva 4"}},
+            {{"title": "Título de la diapositiva 5", "content": "Contenido de la diapositiva 5"}}
         ]
     }}
-    Asegúrate de que la salida sea un JSON válido sin ningún texto adicional antes o después.
-    No incluyas saltos de línea o caracteres especiales dentro del JSON.
-    """
-    
-    response = client.text_generation(prompt, max_new_tokens=1500, 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:
-                return slides_data['slides']
-            else:
-                raise ValueError("El JSON no contiene exactamente 5 diapositivas")
-        else:
-            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, template_path):
-    prs = Presentation(template_path)
+    No incluyas ningún otro texto, explicación o saludo. Solo el JSON.
+    """
     
-    for slide_data in slides:
-        slide = prs.slides.add_slide(prs.slide_layouts[1])
+    for attempt in range(max_retries):
+        response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
         
-        title_shape = slide.shapes.title
-        content_shape = slide.placeholders[1]
+        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']
+            
+            # Si llegamos aquí, el JSON no era válido o no tenía la estructura correcta
+            raise ValueError("JSON inválido o estructura incorrecta")
         
-        title_shape.text = slide_data['title']
-        content_shape.text = slide_data['content']
-    
-    pptx_buffer = io.BytesIO()
-    prs.save(pptx_buffer)
-    pptx_buffer.seek(0)
+        except (json.JSONDecodeError, ValueError) as e:
+            if attempt == max_retries - 1:  # Último intento
+                st.error(f"Error al generar el contenido después de {max_retries} intentos: {str(e)}")
+                st.text("Última respuesta del modelo:")
+                st.code(response)
+                return None
+            else:
+                st.warning(f"Intento {attempt + 1} fallido. Reintentando...")
     
-    return pptx_buffer
+    return None
+
+# El resto del código (create_powerpoint y main) permanece igual
 
 def main():
     st.title("Generador de presentaciones PowerPoint con IA")