Update app.py

app.py

@@ -23,29 +23,33 @@ def extract_and_clean_json(text):
             json_str = re.sub(r',\s*}', '}', json_str)
             json_str = re.sub(r',\s*]', ']', json_str)
             
-            # Verificar si es un JSON válido
-            data = json.loads(json_str)
-            return data
+            # 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 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)
+    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
 
 def generate_presentation_content(topic, client):
-    prompt_part1 = f"""Crea una presentación de PowerPoint sobre el tema: {topic}.
-    Genera exactamente 5 diapositivas con la siguiente estructura:
+    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. Contenido principal (2 diapositivas)
+    4-9. Contenido principal (6 diapositivas)
     
     Formatea la salida como un JSON con la siguiente estructura:
     {{
@@ -59,38 +63,19 @@ def generate_presentation_content(topic, client):
     No incluyas saltos de línea o caracteres especiales dentro del JSON.
     """
     
-    prompt_part2 = f"""Crea las siguientes 4 diapositivas de la presentación de PowerPoint sobre el tema: {topic}.
-    Genera exactamente 4 diapositivas con la siguiente estructura:
-    5-8. Contenido principal (4 diapositivas)
-    
-    Formatea la salida como un JSON con la siguiente estructura:
-    {{
-        "slides": [
-            {{"title": "Título de la diapositiva 5", "content": "Contenido de la diapositiva 5"}},
-            {{"title": "Título de la diapositiva 6", "content": "Contenido de la diapositiva 6"}},
-            ...
-        ]
-    }}
-    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_part1 = client.text_generation(prompt_part1, max_new_tokens=1500, temperature=0.7)
-    response_part2 = client.text_generation(prompt_part2, max_new_tokens=1500, temperature=0.7)
+    response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
     
-    slides_data_part1 = extract_and_clean_json(response_part1)
-    slides_data_part2 = extract_and_clean_json(response_part2)
+    # Limpieza y procesamiento del JSON
+    slides_data = extract_and_clean_json(response)
     
-    if slides_data_part1 and slides_data_part2:
-        if 'slides' in slides_data_part1 and 'slides' in slides_data_part2:
-            slides = slides_data_part1['slides'] + slides_data_part2['slides']
-            if len(slides) == 9:
-                return slides
-            else:
-                st.error(f"El JSON no contiene exactamente 9 diapositivas. Contiene {len(slides)} diapositivas.")
-                st.text("JSON procesado:")
-                st.code(json.dumps(slides, indent=4))
-                return None
+    if slides_data and 'slides' in slides_data:
+        if len(slides_data['slides']) == 9:
+            return slides_data['slides']
+        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):