Update app.py

app.py

@@ -10,10 +10,17 @@ import re
 def get_inference_client():
     return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def extract_json_from_text(text):
+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:
-        return json_match.group()
+        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):
@@ -28,26 +35,27 @@ def generate_presentation_content(topic, client):
         ]
     }}
     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=1000, temperature=0.7)
+    response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
     
     try:
-        json_str = extract_json_from_text(response)
+        json_str = extract_and_clean_json(response)
         if json_str:
             slides_data = json.loads(json_str)
-            return slides_data['slides']
+            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("Respuesta del modelo:")
-        st.code(response)
-        return None
-    except KeyError:
-        st.error("El JSON no contiene la clave 'slides'")
-        st.text("JSON recibido:")
+        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))