Update app.py

app.py

@@ -4,12 +4,18 @@ from pptx.util import Inches, Pt
 import io
 from huggingface_hub import InferenceClient
 import json
+import re
 
-# Initialize the Hugging Face Inference Client
 @st.cache_resource
 def get_inference_client():
     return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
+def extract_json_from_text(text):
+    json_match = re.search(r'\{.*\}', text, re.DOTALL)
+    if json_match:
+        return json_match.group()
+    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.
@@ -21,15 +27,32 @@ 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.
     """
     
     response = client.text_generation(prompt, max_new_tokens=1000, temperature=0.7)
     
     try:
-        slides_data = json.loads(response)
-        return slides_data['slides']
-    except json.JSONDecodeError:
-        st.error("Error al procesar la respuesta del modelo. Por favor, intente nuevamente.")
+        json_str = extract_json_from_text(response)
+        if json_str:
+            slides_data = json.loads(json_str)
+            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("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:")
+        st.code(response)
         return None
 
 def create_powerpoint(slides):