Update app.py

app.py

@@ -10,17 +10,10 @@ import re
 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
+def extract_json_from_text(text):
     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 json_match.group()
     return None
 
 def generate_presentation_content(topic, client):
@@ -35,28 +28,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=1500, temperature=0.7)
+    response = client.text_generation(prompt, max_new_tokens=1000, temperature=0.7)
     
     try:
-        json_str = extract_and_clean_json(response)
+        json_str = extract_json_from_text(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")
+            return slides_data['slides']
         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.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.code(json_str)
+        return None
     except ValueError as e:
         st.error(str(e))
         st.text("Respuesta del modelo:")